fixes bug causing finished runs to resume

ultralytics · May 20, 2021 · 4d79628 · 4d79628
1 parent ede690f
commit 4d79628
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/wandb_logging/wandb_utils.py b/utils/wandb_logging/wandb_utils.py
@@ -187,8 +187,8 @@ def download_model_artifact(self, opt):
             modeldir = model_artifact.download()
             epochs_trained = model_artifact.metadata.get('epochs_trained')
             total_epochs = model_artifact.metadata.get('total_epochs')
-            assert epochs_trained < total_epochs, 'training to %g epochs is finished, nothing to resume.' % (
-                total_epochs)
+            is_finished = total_epochs is None
+            assert not is_finished, 'training is finished, can only resume incomplete runs.'
             return modeldir, model_artifact
         return None, None