Hugging Face's logo

Spaces:
mohamed12ahmed
/
documents-restoration
Sleeping

App Files Community
mohamed12ahmed commited on
Commit
4c2b086
·
verified ·
1 Parent(s): 0ea9dda

Update start_train.sh

Browse files
Files changed (1) hide show
  1. start_train.sh +2 -1
start_train.sh CHANGED
@@ -1 +1,2 @@
1
- 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
 
 
1
+ #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
2
+ python train.py