Skip to content

Commit 634b68a

Browse files
JosuaRiederrwightman
authored andcommitted
Fix metavar for --input-size
1 parent 5d535d7 commit 634b68a

File tree

4 files changed

+7
-8
lines changed

4 files changed

+7
-8
lines changed

benchmark.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,8 @@
8888
metavar='N', help='mini-batch size (default: 256)')
8989
parser.add_argument('--img-size', default=None, type=int,
9090
metavar='N', help='Input image dimension, uses model default if empty')
91-
parser.add_argument('--input-size', default=None, nargs=3, type=int,
92-
metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
91+
parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N',
92+
help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
9393
parser.add_argument('--use-train-size', action='store_true', default=False,
9494
help='Run inference at train size, not test-input-size if it exists.')
9595
parser.add_argument('--num-classes', type=int, default=None,

inference.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,8 @@
6969
metavar='N', help='Input image dimension, uses model default if empty')
7070
parser.add_argument('--in-chans', type=int, default=None, metavar='N',
7171
help='Image input channels (default: None => 3)')
72-
parser.add_argument('--input-size', default=None, nargs=3, type=int,
73-
metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
72+
parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N',
73+
help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
7474
parser.add_argument('--use-train-size', action='store_true', default=False,
7575
help='force use of train input size, even when test size is specified in pretrained cfg')
7676
parser.add_argument('--crop-pct', default=None, type=float,

train.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,7 @@
128128
help='Image size (default: None => model default)')
129129
group.add_argument('--in-chans', type=int, default=None, metavar='N',
130130
help='Image input channels (default: None => 3)')
131-
group.add_argument('--input-size', default=None, nargs=3, type=int,
132-
metavar='N N N',
131+
group.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N',
133132
help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
134133
group.add_argument('--crop-pct', default=None, type=float,
135134
metavar='N', help='Input image center crop percent (for validation only)')

validate.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,8 @@
8181
metavar='N', help='Input image dimension, uses model default if empty')
8282
parser.add_argument('--in-chans', type=int, default=None, metavar='N',
8383
help='Image input channels (default: None => 3)')
84-
parser.add_argument('--input-size', default=None, nargs=3, type=int,
85-
metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
84+
parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N',
85+
help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty')
8686
parser.add_argument('--use-train-size', action='store_true', default=False,
8787
help='force use of train input size, even when test size is specified in pretrained cfg')
8888
parser.add_argument('--crop-pct', default=None, type=float,

0 commit comments

Comments
 (0)