diff --git a/run.sh b/run.sh index 4c502f5..cd25fd6 100644 --- a/run.sh +++ b/run.sh @@ -34,7 +34,7 @@ for depth in small do python main.py --mode train \ --train ${DATADIR}/currentData.csv \ - --model ${RESDIR}/both_${depth}_inter \ + --model ${RESDIR}/both_${depth}_staggered \ --epochs 50 \ --embd 128 \ --filter_embd 256 --kernel_embd 8 --dense_embd 128 \ @@ -45,4 +45,4 @@ do --model_output both \ --type staggered \ --depth ${depth} -done \ No newline at end of file +done