Update train_DiffIRS1.sh

IFF-0303 · Sep 19, 2023 · f779387 · f779387
1 parent 88b7d44
commit f779387
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/DiffIR-RealSR/train_DiffIRS1.sh b/DiffIR-RealSR/train_DiffIRS1.sh
@@ -1,6 +1,9 @@
 
+CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
+python3 -m torch.distributed.launch --nproc_per_node=8 --master_port=7310 DiffIR/train.py -opt options/train_DiffIRS1_x4.yml --launcher pytorch 
 
+# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
+# python3 -m torch.distributed.launch --nproc_per_node=8 --master_port=7310 DiffIR/train.py -opt options/train_DiffIRS1_x2.yml --launcher pytorch 
 
-
-CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
-python3 -m torch.distributed.launch --nproc_per_node=8 --master_port=7310 DiffIR/train.py -opt options/train_DiffIRS1_x4.yml --launcher pytorch 
+# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
+# python3 -m torch.distributed.launch --nproc_per_node=8 --master_port=7310 DiffIR/train.py -opt options/train_DiffIRS1_x1.yml --launcher pytorch