Skip to content

Commit edc37be

Browse files
committed
Remove default 1000 classes from onnx export to use model default
1 parent f14f650 commit edc37be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnx_export.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
help='Override mean pixel value of dataset')
5050
parser.add_argument('--std', type=float, nargs='+', default=None, metavar='STD',
5151
help='Override std deviation of of dataset')
52-
parser.add_argument('--num-classes', type=int, default=1000,
52+
parser.add_argument('--num-classes', type=int, default=None,
5353
help='Number classes in dataset')
5454
parser.add_argument('--checkpoint', default='', type=str, metavar='PATH',
5555
help='path to checkpoint (default: none)')

0 commit comments

Comments
 (0)