add start parameter for run script
This commit is contained in:
parent
7f49021a63
commit
b0e0cd904e
20
run_model.sh
20
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}
|
||||
|
Loading…
Reference in New Issue
Block a user