set default epochs to converge

wasertech · Jun 4, 2022 · 65aef10 · 65aef10
1 parent 682a125
commit 65aef10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/STT/Dockerfile.train b/STT/Dockerfile.train
@@ -25,7 +25,7 @@ ENV TEST_BATCH_SIZE=$test_batch_size
 ARG n_hidden=2048
 ENV N_HIDDEN=$n_hidden
 
-ARG epochs=30
+ARG epochs=45
 ENV EPOCHS=$epochs
 
 ARG learning_rate=0.0001