Преглед изворни кода

Explicitly convert artifact path to posix_path (#3067)

* Explicitly convert artifact path to posix_path

* Remove redudant str() casting
modifyDataloader
Ayush Chaurasia GitHub пре 3 година
родитељ
комит
d2a17289c9
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 измењених фајлова са 2 додато и 1 уклоњено
  1. +2
    -1
      utils/wandb_logging/wandb_utils.py

+ 2
- 1
utils/wandb_logging/wandb_utils.py Прегледај датотеку

@@ -158,7 +158,8 @@ class WandbLogger():

def download_dataset_artifact(self, path, alias):
if isinstance(path, str) and path.startswith(WANDB_ARTIFACT_PREFIX):
dataset_artifact = wandb.use_artifact(remove_prefix(path, WANDB_ARTIFACT_PREFIX) + ":" + alias)
artifact_path = Path(remove_prefix(path, WANDB_ARTIFACT_PREFIX) + ":" + alias)
dataset_artifact = wandb.use_artifact(artifact_path.as_posix())
assert dataset_artifact is not None, "'Error: W&B dataset artifact doesn\'t exist'"
datadir = dataset_artifact.download()
return datadir, dataset_artifact

Loading…
Откажи
Сачувај