updated training resnet2060 command

deepinsight · Jun 23, 2021 · 5f1c233 · 5f1c233
1 parent f7007b6
commit 5f1c233
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/recognition/arcface_torch/README.md b/recognition/arcface_torch/README.md
@@ -9,15 +9,11 @@ torch >= 1.6.0
 More details see [eval.md](docs/install.md) in docs.
 
 ## Training
-### 1. Single node, 1 GPUs:
-```shell
-python -m torch.distributed.launch --nproc_per_node=1 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=1234 train.py
-```
-### 2. Single node, 8 GPUs:
+### 1. Single node, 8 GPUs:
 ```shell
 python -m torch.distributed.launch --nproc_per_node=8 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=1234 train.py
 ```
-### 3. Multiple nodes, each node 8 GPUs: 
+### 2. Multiple nodes, each node 8 GPUs: 
 Node 0: 
 ```shell
 python -m torch.distributed.launch --nproc_per_node=8 --nnodes=2 --node_rank=0 --master_addr="ip1" --master_port=1234 train.py
@@ -27,6 +23,11 @@ Node 1:
 python -m torch.distributed.launch --nproc_per_node=8 --nnodes=2 --node_rank=1 --master_addr="ip1" --master_port=1234 train.py
 ```
 
+### 3.Training resnet2060 with 8 GPUs:
+```shell
+python -m torch.distributed.launch --nproc_per_node=1 --nnodes=1 --node_rank=0 --master_addr="127.0.0.1" --master_port=1234 train.py --network r2060
+```
+
 ## Speed Benchmark
 ![Image text](https://github.com/nttstar/insightface-resources/blob/master/images/arcface_speed.png)