update train script

Author: hsvgbkhgbv

train_case141.sh

@@ -1,3 +1,9 @@
+if [ ! -d $3 ]
+then
+  mkdir $3
+fi
+
+
 export CUDA_VISIBLE_DEVICES=1
 nohup python train.py --alg coma --alias $1 --mode distributed --scenario case141_3min_final --voltage-barrier-type $2 --save-path $3 > coma_141_$1_$2.out &
 export CUDA_VISIBLE_DEVICES=1

train_case322.sh

@@ -1,3 +1,9 @@
+if [ ! -d $3 ]
+then
+  mkdir $3
+fi
+
+
 export CUDA_VISIBLE_DEVICES=0
 nohup python train.py --alg coma --alias $1 --mode distributed --scenario case322_3min_final --voltage-barrier-type $2 --save-path $3 > coma_322_$1_$2.out &
 export CUDA_VISIBLE_DEVICES=0

train_case33.sh

@@ -1,3 +1,9 @@
+if [ ! -d $3 ]
+then
+  mkdir $3
+fi
+
+
 export CUDA_VISIBLE_DEVICES=1
 nohup python train.py --alg coma --alias $1 --mode distributed --scenario case33_3min_final --voltage-barrier-type $2 --save-path $3 > coma_33_$1_$2.out &
 export CUDA_VISIBLE_DEVICES=1