fix argument inconsistency

magic-research · Jan 6, 2024 · 24a2a77 · 24a2a77
1 parent f2d7744
commit 24a2a77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_image_models/train.py b/pytorch_image_models/train.py
@@ -96,7 +96,7 @@
                     help='Allow download of dataset for torch/ and tfds/ datasets that support it.')
 group.add_argument('--class-map', default='', type=str, metavar='FILENAME',
                     help='path to class to idx mapping file (default: "")')
-group.add_argument('--select_indices', default=[], type=str, nargs='+',
+group.add_argument('--select-indices', default=[], type=str, nargs='+',
                     help='pre-defined subset indices')
 
 # Model parameters