diff --git a/train_case141.sh b/train_case141.sh
index ecf10ad..989a822 100644
--- a/train_case141.sh
+++ b/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
diff --git a/train_case322.sh b/train_case322.sh
index 401a6b6..5b2680d 100644
--- a/train_case322.sh
+++ b/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
diff --git a/train_case33.sh b/train_case33.sh
index 8049c3c..519e25a 100644
--- a/train_case33.sh
+++ b/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