29 lines
697 B
Bash
29 lines
697 B
Bash
#!/usr/bin/env bash
|
|
|
|
|
|
N=$1
|
|
OUTPUT=$2
|
|
DEPTH=$3
|
|
TYPE=$4
|
|
RESDIR=$5
|
|
mkdir -p /tmp/rk/${RESDIR}
|
|
DATADIR=$6
|
|
|
|
EPOCHS=100
|
|
|
|
for i in {1..$N}
|
|
do
|
|
python main.py --mode train \
|
|
--train ${DATADIR}/currentData.csv \
|
|
--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 256 \
|
|
--balanced_weights \
|
|
--model_output ${OUTPUT} \
|
|
--type ${TYPE} \
|
|
--depth ${DEPTH}
|
|
done |