No default img size for inference script. Fixes huggingface#109

willxian · Apr 9, 2020 · 0ebcd2e · 0ebcd2e
1 parent 1a8f590
commit 0ebcd2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference.py b/inference.py
@@ -29,7 +29,7 @@
                     help='number of data loading workers (default: 2)')
 parser.add_argument('-b', '--batch-size', default=256, type=int,
                     metavar='N', help='mini-batch size (default: 256)')
-parser.add_argument('--img-size', default=224, type=int,
+parser.add_argument('--img-size', default=None, type=int,
                     metavar='N', help='Input image dimension')
 parser.add_argument('--mean', type=float, nargs='+', default=None, metavar='MEAN',
                     help='Override mean pixel value of dataset')