From 6ea00d23d66fd491bbf19317b4677bf6f396d26f Mon Sep 17 00:00:00 2001
From: Haodong Duan <dhd.efz@gmail.com>
Date: Sat, 2 Apr 2022 17:39:22 +0800
Subject: [PATCH] [Fix] Fix local rank (#1558)

* update

* update
---
 docs/getting_started.md               | 4 ++--
 docs_zh_CN/getting_started.md         | 4 ++--
 tools/misc/clip_feature_extraction.py | 2 +-
 tools/test.py                         | 2 +-
 tools/train.py                        | 2 +-
 5 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/docs/getting_started.md b/docs/getting_started.md
index 66a7e46663..fef4b05e4a 100644
--- a/docs/getting_started.md
+++ b/docs/getting_started.md
@@ -72,12 +72,12 @@ You can use the following commands to test a dataset.
 # single-gpu testing
 python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] \
     [--gpu-collect] [--tmpdir ${TMPDIR}] [--options ${OPTIONS}] [--average-clips ${AVG_TYPE}] \
-    [--launcher ${JOB_LAUNCHER}] [--local-rank ${LOCAL_RANK}] [--onnx] [--tensorrt]
+    [--launcher ${JOB_LAUNCHER}] [--local_rank ${LOCAL_RANK}] [--onnx] [--tensorrt]
 
 # multi-gpu testing
 ./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] \
     [--gpu-collect] [--tmpdir ${TMPDIR}] [--options ${OPTIONS}] [--average-clips ${AVG_TYPE}] \
-    [--launcher ${JOB_LAUNCHER}] [--local-rank ${LOCAL_RANK}]
+    [--launcher ${JOB_LAUNCHER}] [--local_rank ${LOCAL_RANK}]
 ```
 
 Optional arguments:
diff --git a/docs_zh_CN/getting_started.md b/docs_zh_CN/getting_started.md
index 577e0cad88..9da0aa8065 100644
--- a/docs_zh_CN/getting_started.md
+++ b/docs_zh_CN/getting_started.md
@@ -71,12 +71,12 @@ MMAction2 支持仅使用 CPU 进行测试。然而，这样做的速度**非常
 # 单 GPU 测试
 python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] \
     [--gpu-collect] [--tmpdir ${TMPDIR}] [--options ${OPTIONS}] [--average-clips ${AVG_TYPE}] \
-    [--launcher ${JOB_LAUNCHER}] [--local-rank ${LOCAL_RANK}] [--onnx] [--tensorrt]
+    [--launcher ${JOB_LAUNCHER}] [--local_rank ${LOCAL_RANK}] [--onnx] [--tensorrt]
 
 # 多 GPU 测试
 ./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] \
     [--gpu-collect] [--tmpdir ${TMPDIR}] [--options ${OPTIONS}] [--average-clips ${AVG_TYPE}] \
-    [--launcher ${JOB_LAUNCHER}] [--local-rank ${LOCAL_RANK}]
+    [--launcher ${JOB_LAUNCHER}] [--local_rank ${LOCAL_RANK}]
 ```
 
 可选参数:
diff --git a/tools/misc/clip_feature_extraction.py b/tools/misc/clip_feature_extraction.py
index a6e37b3c3a..1829bf9b5c 100644
--- a/tools/misc/clip_feature_extraction.py
+++ b/tools/misc/clip_feature_extraction.py
@@ -59,7 +59,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 9152c90917..31514498cf 100644
--- a/tools/test.py
+++ b/tools/test.py
@@ -89,7 +89,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)
     parser.add_argument(
         '--onnx',
         action='store_true',
diff --git a/tools/train.py b/tools/train.py
index 6461e7030a..d404980464 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -74,7 +74,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)