zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix: #2674 (#2683) 

* Set resume flag to false

* Check existance of val dataset
This commit is contained in:
Ayush Chaurasia 2021-04-02 15:24:50 +05:30 committed by GitHub
parent 1148e2ea63
commit 514ebcdf33
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/wandb_logging/wandb_utils.py
View File

@ -158,7 +158,7 @@ class WandbLogger():
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()