From f1e4d45bc225648deac0cf498a16f648f501b93b Mon Sep 17 00:00:00 2001 From: mzr1996 Date: Wed, 30 Mar 2022 16:58:29 +0800 Subject: [PATCH] [Enhance] Use `--a-b` instead of `--a_b` in arguments --- tools/convert_models/vgg_to_mmcls.py | 2 +- tools/kfold-cross-valid.py | 2 +- tools/test.py | 2 +- tools/train.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tools/convert_models/vgg_to_mmcls.py b/tools/convert_models/vgg_to_mmcls.py index 56f5caf401c..b5ab87f6604 100644 --- a/tools/convert_models/vgg_to_mmcls.py +++ b/tools/convert_models/vgg_to_mmcls.py @@ -104,7 +104,7 @@ def main(): parser.add_argument( '--bn', action='store_true', help='whether original vgg has BN') parser.add_argument( - '--layer_num', + '--layer-num', type=int, choices=[11, 13, 16, 19], default=11, diff --git a/tools/kfold-cross-valid.py b/tools/kfold-cross-valid.py index 2f3a70e14b6..c6d8d89d013 100644 --- a/tools/kfold-cross-valid.py +++ b/tools/kfold-cross-valid.py @@ -97,7 +97,7 @@ def parse_args(): choices=['none', 'pytorch', 'slurm', 'mpi'], default='none', help='job launcher') - parser.add_argument('--local_rank', type=int, default=0) + parser.add_argument('--local-rank', type=int, default=0) args = parser.parse_args() if 'LOCAL_RANK' not in os.environ: os.environ['LOCAL_RANK'] = str(args.local_rank) diff --git a/tools/test.py b/tools/test.py index c294c5e5678..18b52357219 100644 --- a/tools/test.py +++ b/tools/test.py @@ -93,7 +93,7 @@ def parse_args(): choices=['none', 'pytorch', 'slurm', 'mpi'], default='none', help='job launcher') - parser.add_argument('--local_rank', type=int, default=0) + parser.add_argument('--local-rank', type=int, default=0) args = parser.parse_args() if 'LOCAL_RANK' not in os.environ: os.environ['LOCAL_RANK'] = str(args.local_rank) diff --git a/tools/train.py b/tools/train.py index 75085c57c1b..cf21c93a5ee 100644 --- a/tools/train.py +++ b/tools/train.py @@ -73,7 +73,7 @@ def parse_args(): choices=['none', 'pytorch', 'slurm', 'mpi'], default='none', help='job launcher') - parser.add_argument('--local_rank', type=int, default=0) + parser.add_argument('--local-rank', type=int, default=0) args = parser.parse_args() if 'LOCAL_RANK' not in os.environ: os.environ['LOCAL_RANK'] = str(args.local_rank)