Browse Source

Fix: #2674 (#2683)

* Set resume flag to false

* Check existance of val dataset
5.0
Ayush Chaurasia GitHub 3 years ago
parent
commit
514ebcdf33
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      utils/wandb_logging/wandb_utils.py

+ 1
- 1
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()

Loading…
Cancel
Save