[Enhance] Use --a-b instead of --a_b in arguments. (open-mmlab#754)

mzr1996 · Mar 30, 2022 · 6496030 · 6496030
1 parent 1afb608
commit 6496030
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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)