diff --git a/timm/data/dataset_factory.py b/timm/data/dataset_factory.py index 92529357..757c2e5d 100644 --- a/timm/data/dataset_factory.py +++ b/timm/data/dataset_factory.py @@ -131,7 +131,12 @@ def create_dataset( elif split in _EVAL_SYNONYM: split = 'val' ds = Places365(split=split, **torch_kwargs) + elif name == 'qmnist': + assert has_qmnist, 'Please update to a newer PyTorch and torchvision for QMNIST dataset.' + use_train = split in _TRAIN_SYNONYM + ds = QMNIST(train=use_train, **torch_kwargs) elif name == 'imagenet': + assert has_imagenet, 'Please update to a newer PyTorch and torchvision for ImageNet dataset.' if split in _EVAL_SYNONYM: split = 'val' ds = ImageNet(split=split, **torch_kwargs)