mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 08:02:27 +00:00
15 lines
408 B
Bash
15 lines
408 B
Bash
ip_list=("ip1" "ip2" "ip3" "ip4" "ip5")
|
|
config=wf42m_pfc02_vit_h.py
|
|
|
|
for((node_rank=0;node_rank<${#ip_list[*]};node_rank++));
|
|
do
|
|
ssh root@${ip_list[node_rank]} "cd `pwd`;PATH=$PATH \
|
|
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
|
|
torchrun \
|
|
--nproc_per_node=8 \
|
|
--nnodes=${#ip_list[*]} \
|
|
--node_rank=$node_rank \
|
|
--master_addr=${ip_list[0]} \
|
|
--master_port=22345 train_v2.py configs/$config" &
|
|
done
|