Merge pull request #6 from uzh-rpg/resume-issue

Fix missing Key in metadata when resuming training
uzh-rpg · May 16, 2023 · 5774d5c · 5774d5c
2 parents 4250ee4 + a368f2f
commit 5774d5c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/loggers/wandb_logger.py b/loggers/wandb_logger.py
@@ -318,7 +318,10 @@ def _scan_and_log_checkpoints(self, checkpoint_callback: "ReferenceType[ModelChe
                 print(f'num_new_cktps = {num_new_cktps}')
                 break
 
-        self._rm_but_top_k(checkpoint_callback.save_top_k)
+        try:
+            self._rm_but_top_k(checkpoint_callback.save_top_k)
+        except KeyError:
+            pass
 
     def _rm_but_top_k(self, top_k: int):
         # top_k == -1: save all models