From 653bdc7105098763cb672e344f6d716021f9c627 Mon Sep 17 00:00:00 2001 From: hongxin xiang Date: Fri, 25 Nov 2022 09:52:52 +0800 Subject: [PATCH] Fix comment: https://github.com/rwightman/pytorch-image-models/pull/1564#issuecomment-1326743424 --- timm/data/dataset_factory.py | 5 +++++ 1 file changed, 5 insertions(+) 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)