diff --git a/STT/Dockerfile.train b/STT/Dockerfile.train index 450fb9ff..d92c6e90 100644 --- 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=45 +ARG epochs=40 ENV EPOCHS=$epochs ARG learning_rate=0.0001