From f4a69e4efb503764b2ec977917f8a1b0e28b1e82 Mon Sep 17 00:00:00 2001 From: Josua Rieder Date: Mon, 20 Jan 2025 01:11:06 +0100 Subject: [PATCH] Fix metavar for --input-size --- benchmark.py | 4 ++-- inference.py | 4 ++-- train.py | 3 +-- validate.py | 4 ++-- 4 files changed, 7 insertions(+), 8 deletions(-) diff --git a/benchmark.py b/benchmark.py index d31395b83e..56c75f6c1e 100755 --- a/benchmark.py +++ b/benchmark.py @@ -88,8 +88,8 @@ metavar='N', help='mini-batch size (default: 256)') parser.add_argument('--img-size', default=None, type=int, metavar='N', help='Input image dimension, uses model default if empty') -parser.add_argument('--input-size', default=None, nargs=3, type=int, - metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') +parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N', + help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') parser.add_argument('--use-train-size', action='store_true', default=False, help='Run inference at train size, not test-input-size if it exists.') parser.add_argument('--num-classes', type=int, default=None, diff --git a/inference.py b/inference.py index 2272f4e618..46232534d1 100755 --- a/inference.py +++ b/inference.py @@ -69,8 +69,8 @@ metavar='N', help='Input image dimension, uses model default if empty') parser.add_argument('--in-chans', type=int, default=None, metavar='N', help='Image input channels (default: None => 3)') -parser.add_argument('--input-size', default=None, nargs=3, type=int, - metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') +parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N', + help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') parser.add_argument('--use-train-size', action='store_true', default=False, help='force use of train input size, even when test size is specified in pretrained cfg') parser.add_argument('--crop-pct', default=None, type=float, diff --git a/train.py b/train.py index c6c1fcb1a9..6e7057bc00 100755 --- a/train.py +++ b/train.py @@ -128,8 +128,7 @@ help='Image size (default: None => model default)') group.add_argument('--in-chans', type=int, default=None, metavar='N', help='Image input channels (default: None => 3)') -group.add_argument('--input-size', default=None, nargs=3, type=int, - metavar='N N N', +group.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') group.add_argument('--crop-pct', default=None, type=float, metavar='N', help='Input image center crop percent (for validation only)') diff --git a/validate.py b/validate.py index a414f36f6c..2e416e650b 100755 --- a/validate.py +++ b/validate.py @@ -81,8 +81,8 @@ metavar='N', help='Input image dimension, uses model default if empty') parser.add_argument('--in-chans', type=int, default=None, metavar='N', help='Image input channels (default: None => 3)') -parser.add_argument('--input-size', default=None, nargs=3, type=int, - metavar='N N N', help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') +parser.add_argument('--input-size', default=None, nargs=3, type=int, metavar='N', + help='Input all image dimensions (d h w, e.g. --input-size 3 224 224), uses model default if empty') parser.add_argument('--use-train-size', action='store_true', default=False, help='force use of train input size, even when test size is specified in pretrained cfg') parser.add_argument('--crop-pct', default=None, type=float,