#!/usr/bin/env bash


N1=$1
N2=$2
OUTPUT=$3
DEPTH=$4
TYPE=$5
RESDIR=$6
mkdir -p /tmp/rk/${RESDIR}
DATADIR=$7

EPOCHS=10

for ((i = ${N1}; i <= ${N2}; i++))
do
    python main.py --mode train \
                --train ${DATADIR} \
                --model ${RESDIR}/${OUTPUT}_${TYPE}_${i} \
                --epochs ${EPOCHS} \
                --embd 64 \
                --filter_embd 128 --kernel_embd 5 --dense_embd 64 \
                --domain_embd 16 \
                --filter_main 32  --kernel_main 5 --dense_main 256 \
                --batch 128 \
                --model_output ${OUTPUT} \
                --type ${TYPE} \
                --depth ${DEPTH} \
                --gpu
done