documents-restoration / start_train.sh
mohamed12ahmed's picture
Update start_train.sh
4c2b086 verified
raw
history blame contribute delete
137 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
python train.py