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

pull/1327/head
Ross Wightman 2 years ago
parent 4670d375c6
commit 9be0c84715

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

Loading…
Cancel
Save