Skip to content

Commit 0ebcd2e

Browse files
committed
No default img size for inference script. Fixes #109
1 parent 1a8f590 commit 0ebcd2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
help='number of data loading workers (default: 2)')
3030
parser.add_argument('-b', '--batch-size', default=256, type=int,
3131
metavar='N', help='mini-batch size (default: 256)')
32-
parser.add_argument('--img-size', default=224, type=int,
32+
parser.add_argument('--img-size', default=None, type=int,
3333
metavar='N', help='Input image dimension')
3434
parser.add_argument('--mean', type=float, nargs='+', default=None, metavar='MEAN',
3535
help='Override mean pixel value of dataset')

0 commit comments

Comments
 (0)