From c8b0cf3af0bfadbc8dfa6ecb6dd8a339b6113e88 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
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(-)
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