mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

remove download argument from torch_kwargs

Browse Source
This commit is contained in:
Ryan Ramos 2025-05-09 15:11:22 +09:00 committed by Ross Wightman
parent 59e59ff040
commit 81900a6bae
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/data/dataset_factory.py
View File

@ -144,6 +144,7 @@ def create_dataset(
use_train = split in _TRAIN_SYNONYM
ds = QMNIST(train=use_train, **torch_kwargs)
elif name == 'imagenet':
torch_kwargs.pop('download')
assert has_imagenet, 'Please update to a newer PyTorch and torchvision for ImageNet dataset.'
if split in _EVAL_SYNONYM:
split = 'val'