From c8b0cf3af0bfadbc8dfa6ecb6dd8a339b6113e88 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 25 Jun 2022 23:59:56 +0200 Subject: Update checkpoint path in wandb callback --- training/callbacks/wandb_callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training') 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) -- cgit v1.2.3-70-g09d2