run.sh 80 Bytes
Newer Older
dongchy920's avatar
arcface  
dongchy920 committed
1
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nproc_per_node=8 train_v2.py $@