File tree 1 file changed +7
-6
lines changed
recognition/arcface_torch
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
9
9
More details see [ eval.md] ( docs/install.md ) in docs.
10
10
11
11
## 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:
17
13
``` shell
18
14
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
19
15
```
20
- ### 3 . Multiple nodes, each node 8 GPUs:
16
+ ### 2 . Multiple nodes, each node 8 GPUs:
21
17
Node 0:
22
18
``` shell
23
19
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:
27
23
python -m torch.distributed.launch --nproc_per_node=8 --nnodes=2 --node_rank=1 --master_addr=" ip1" --master_port=1234 train.py
28
24
```
29
25
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
+
30
31
## Speed Benchmark
31
32
![ Image text] ( https://github.com/nttstar/insightface-resources/blob/master/images/arcface_speed.png )
32
33
You can’t perform that action at this time.
0 commit comments