diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-25 23:59:56 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-25 23:59:56 +0200 |
| commit | c8b0cf3af0bfadbc8dfa6ecb6dd8a339b6113e88 (patch) | |
| tree | c8e1b864d0f5f869e694fb507898091d116af969 /training/callbacks | |
| parent | bc8184728044ef823537ca2358cd1cd9c22e1b9e (diff) | |
Update checkpoint path in wandb callback
Diffstat (limited to 'training/callbacks')
| -rw-r--r-- | training/callbacks/wandb_callbacks.py | 2 |
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) |