From b0e0cd904e1634da22c366ad4f33e6f330316da3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A9=20Knaebel?= Date: Fri, 15 Sep 2017 11:12:42 +0200 Subject: [PATCH] add start parameter for run script --- run_model.sh | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/run_model.sh b/run_model.sh index 45fce6b..7559534 100644 --- a/run_model.sh +++ b/run_model.sh @@ -1,28 +1,28 @@ #!/usr/bin/env bash -N=$1 -OUTPUT=$2 -DEPTH=$3 -TYPE=$4 -RESDIR=$5 +N1=$1 +N2=$2 +OUTPUT=$3 +DEPTH=$4 +TYPE=$5 +RESDIR=$6 mkdir -p /tmp/rk/${RESDIR} -DATADIR=$6 +DATADIR=$7 EPOCHS=100 -for i in {1..$N} +for ((i = ${N1}; i <= ${N2}; i++)) do python main.py --mode train \ --train ${DATADIR}/currentData.csv \ - --model ${RESDIR}/${OUTPUT}_${TYPE}_$i \ - --epochs $EPOCHS \ + --model ${RESDIR}/${OUTPUT}_${TYPE}_${i} \ + --epochs ${EPOCHS} \ --embd 128 \ --filter_embd 256 --kernel_embd 8 --dense_embd 128 \ --domain_embd 32 \ --filter_main 32 --kernel_main 8 --dense_main 1024 \ --batch 128 \ -# --balanced_weights \ --model_output ${OUTPUT} \ --type ${TYPE} \ --depth ${DEPTH}