Files
insightface/recognition/arcface_torch/run.sh
XiangAn bf32ec2d3f Resolved torchrun Bug: Fixed issue #2163
Updated torch.distributed.launch to torchrun.
2023-02-08 23:15:46 +08:00

2 lines
80 B
Bash

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nproc_per_node=8 train_v2.py $@