Преглед на файлове

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…
Отказ
Запис