documents-restoration / start_train.sh
qubvel-hf's picture
qubvel-hf HF staff
Init project
c509e76
raw
history blame
119 Bytes
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node 8 --master_port 26413 train.py