Distributed Arcface Training in Pytorch
This is a deep learning library that makes face recognition efficient, and effective, which can train tens of millionsidentity on a single server. Requirements How to Training To train a model, run train.py with the path to the configs: 1. Single node, 8 GPUs: 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 configs/ms1mv3_r50 2. Multiple nodes, each node 8 GPUs: Node 0: python -m torch.distributed.launch –nproc_per_node=8 –nnodes=2 –node_rank=0 –master_addr=
Read more