mrbear1024's picture
init project
8eb4303
raw
history blame contribute delete
80 Bytes
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nproc_per_node=8 train_v2.py $@