Fix: ultralytics#2674 (ultralytics#2683)

* Set resume flag to false * Check existance of val dataset
KMint1819 · Apr 2, 2021 · 45a98db · 45a98db
1 parent b45a6f9
commit 45a98db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/wandb_logging/wandb_utils.py b/utils/wandb_logging/wandb_utils.py
@@ -158,7 +158,7 @@ def setup_training(self, opt, data_dict):
         return data_dict
 
     def download_dataset_artifact(self, path, alias):
-        if path.startswith(WANDB_ARTIFACT_PREFIX):
+        if path and path.startswith(WANDB_ARTIFACT_PREFIX):
             dataset_artifact = wandb.use_artifact(remove_prefix(path, WANDB_ARTIFACT_PREFIX) + ":" + alias)
             assert dataset_artifact is not None, "'Error: W&B dataset artifact doesn\'t exist'"
             datadir = dataset_artifact.download()