File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed
recognition/arcface_torch Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -9,15 +9,11 @@ torch >= 1.6.0
99More details see [ eval.md] ( docs/install.md ) in docs.
1010
1111## Training
12- ### 1. Single node, 1 GPUs:
13- ``` shell
14- 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
15- ```
16- ### 2. Single node, 8 GPUs:
12+ ### 1. Single node, 8 GPUs:
1713``` shell
1814python -m torch.distributed.launch --nproc_per_node=8 --nnodes=1 --node_rank=0 --master_addr=" 127.0.0.1" --master_port=1234 train.py
1915```
20- ### 3 . Multiple nodes, each node 8 GPUs:
16+ ### 2 . Multiple nodes, each node 8 GPUs:
2117Node 0:
2218``` shell
2319python -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:
2723python -m torch.distributed.launch --nproc_per_node=8 --nnodes=2 --node_rank=1 --master_addr=" ip1" --master_port=1234 train.py
2824```
2925
26+ ### 3.Training resnet2060 with 8 GPUs:
27+ ``` shell
28+ 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
29+ ```
30+
3031## Speed Benchmark
3132![ Image text] ( https://github.com/nttstar/insightface-resources/blob/master/images/arcface_speed.png )
3233
You can’t perform that action at this time.
0 commit comments