update train script

Future-Power-Networks · Nov 1, 2021 · a996f0e · a996f0e
1 parent 603fcba
commit a996f0e
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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