Przeglądaj źródła

Explicitly convert artifact path to posix_path (#3067)

* Explicitly convert artifact path to posix_path

* Remove redudant str() casting
modifyDataloader
Ayush Chaurasia GitHub 3 lat temu
rodzic
commit
d2a17289c9
Nie znaleziono w bazie danych klucza dla tego podpisu ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 2 dodań i 1 usunięć
  1. +2
    -1
      utils/wandb_logging/wandb_utils.py

+ 2
- 1
utils/wandb_logging/wandb_utils.py Wyświetl plik

@@ -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

Ładowanie…
Anuluj
Zapisz