add start parameter for run script

This commit is contained in:
René Knaebel 2017-09-15 11:12:42 +02:00
parent 7f49021a63
commit b0e0cd904e

View File

@ -1,28 +1,28 @@
#!/usr/bin/env bash #!/usr/bin/env bash
N=$1 N1=$1
OUTPUT=$2 N2=$2
DEPTH=$3 OUTPUT=$3
TYPE=$4 DEPTH=$4
RESDIR=$5 TYPE=$5
RESDIR=$6
mkdir -p /tmp/rk/${RESDIR} mkdir -p /tmp/rk/${RESDIR}
DATADIR=$6 DATADIR=$7
EPOCHS=100 EPOCHS=100
for i in {1..$N} for ((i = ${N1}; i <= ${N2}; i++))
do do
python main.py --mode train \ python main.py --mode train \
--train ${DATADIR}/currentData.csv \ --train ${DATADIR}/currentData.csv \
--model ${RESDIR}/${OUTPUT}_${TYPE}_$i \ --model ${RESDIR}/${OUTPUT}_${TYPE}_${i} \
--epochs $EPOCHS \ --epochs ${EPOCHS} \
--embd 128 \ --embd 128 \
--filter_embd 256 --kernel_embd 8 --dense_embd 128 \ --filter_embd 256 --kernel_embd 8 --dense_embd 128 \
--domain_embd 32 \ --domain_embd 32 \
--filter_main 32 --kernel_main 8 --dense_main 1024 \ --filter_main 32 --kernel_main 8 --dense_main 1024 \
--batch 128 \ --batch 128 \
# --balanced_weights \
--model_output ${OUTPUT} \ --model_output ${OUTPUT} \
--type ${TYPE} \ --type ${TYPE} \
--depth ${DEPTH} --depth ${DEPTH}