summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-25 23:59:56 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-25 23:59:56 +0200
commitc8b0cf3af0bfadbc8dfa6ecb6dd8a339b6113e88 (patch)
treec8e1b864d0f5f869e694fb507898091d116af969 /training
parentbc8184728044ef823537ca2358cd1cd9c22e1b9e (diff)
Update checkpoint path in wandb callback
Diffstat (limited to 'training')
-rw-r--r--training/callbacks/wandb_callbacks.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py
index 9e1d4d4..0598fbf 100644
--- a/training/callbacks/wandb_callbacks.py
+++ b/training/callbacks/wandb_callbacks.py
@@ -75,7 +75,7 @@ class UploadCheckpointsAsArtifact(Callback):
if self.upload_best_only:
ckpts.add_file(trainer.checkpoint_callback.best_model_path)
else:
- for ckpt in (self.ckpt_dir).rglob("**/*.ckpt"):
+ for ckpt in (self.ckpt_dir).rglob("*.ckpt"):
ckpts.add_file(ckpt)
experiment.use_artifact(ckpts)