From a996f0ee67c2cbc09f2df70e96cb6218d2e8d874 Mon Sep 17 00:00:00 2001 From: hsvgbkhgbv Date: Mon, 1 Nov 2021 13:18:49 +0000 Subject: [PATCH] update train script --- train_case141.sh | 6 ++++++ train_case322.sh | 6 ++++++ train_case33.sh | 6 ++++++ 3 files changed, 18 insertions(+) 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