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

rwightman · rwightman · commit 9be0c847154b · 2022-07-07T15:33:53.000-07:00
diff --git a/timm/data/dataset_factory.py b/timm/data/dataset_factory.py
@@ -26,8 +26,8 @@
     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):

Original file line number	Diff line number	Diff line change
`@@ -26,8 +26,8 @@`
`26`	`26`	`kmnist=KMNIST,`
`27`	`27`	`fashion_mnist=FashionMNIST,`
`28`	`28`	`)`
`29`		`-_TRAIN_SYNONYM = {'train', 'training'}`
`30`		`-_EVAL_SYNONYM = {'val', 'valid', 'validation', 'eval', 'evaluation'}`
	`29`	`+_TRAIN_SYNONYM = dict(train=None, training=None)`
	`30`	`+_EVAL_SYNONYM = dict(val=None, valid=None, validation=None, eval=None, evaluation=None)`
`31`	`31`
`32`	`32`
`33`	`33`	`def _search_split(root, split):`