use 4 gpus

train.sh

@@ -52,7 +52,7 @@ else
 fi
 
 # -------------------------- Train Pipeline --------------------------
-WORLD_SIZE=8
+WORLD_SIZE=4
 if [ $WORLD_SIZE == 1 ]; then
     python train.py \
             --cuda \