Update --cache disk deprecate *_npy/ dirs (#6876)

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Updates

* Cleanup

* Cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-03-06 16:16:17 +01:00
1 parent 8a66ebad44
commit 4728840745
3 files changed
+40 -40

No files matched your search

+1 -1
View File
@@ -403,7 +403,7 @@ class WandbLogger():
# TODO: Explore multiprocessing to slpit this loop parallely| This is essential for speeding up the the logging
artifact = wandb.Artifact(name=name, type="dataset")
img_files = tqdm([dataset.path]) if isinstance(dataset.path, str) and Path(dataset.path).is_dir() else None
img_files = tqdm(dataset.img_files) if not img_files else img_files
img_files = tqdm(dataset.im_files) if not img_files else img_files
for img_file in img_files:
if Path(img_file).is_dir():
artifact.add_dir(img_file, name='data/images')