deo / main.sh
jinyin_chen
test
e8b0040
raw
history blame
114 Bytes
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 --use_env main_train.py