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

Change set -> dict w/ None keys for dataset split synonym search, so always consistent if more than 1 exists. Fix #1224

Browse Source
This commit is contained in:
Ross Wightman 2022-07-07 15:33:53 -07:00
parent 4670d375c6
commit 9be0c84715
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/data/dataset_factory.py
View File

@ -26,8 +26,8 @@ _TORCH_BASIC_DS = dict(
kmnist=KMNIST,
fashion_mnist=FashionMNIST,
)
_TRAIN_SYNONYM = {'train', 'training'}
_EVAL_SYNONYM = {'val', 'valid', 'validation', 'eval', 'evaluation'}
_TRAIN_SYNONYM = dict(train=None, training=None)
_EVAL_SYNONYM = dict(val=None, valid=None, validation=None, eval=None, evaluation=None)
def _search_split(root, split):