unify the train script

train.sh

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