We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f14f650 commit edc37beCopy full SHA for edc37be
onnx_export.py
@@ -49,7 +49,7 @@
49
help='Override mean pixel value of dataset')
50
parser.add_argument('--std', type=float, nargs='+', default=None, metavar='STD',
51
help='Override std deviation of of dataset')
52
-parser.add_argument('--num-classes', type=int, default=1000,
+parser.add_argument('--num-classes', type=int, default=None,
53
help='Number classes in dataset')
54
parser.add_argument('--checkpoint', default='', type=str, metavar='PATH',
55
help='path to checkpoint (default: none)')
0 commit comments