diff --git a/src/seq2seq/main.py b/src/seq2seq/main.py
index 1b9040ce019825bec9b3fd46040030549679a115..ed33eaeac259213efeccec135495cc76deefab81 100644
--- a/src/seq2seq/main.py
+++ b/src/seq2seq/main.py
@@ -31,7 +31,7 @@ model_args = {
     "max_seq_length": 128,
     "train_batch_size": 4,
     "eval_batch_size": 1,
-    "num_train_epochs": 1,
+    "num_train_epochs": 16,
     "max_length": 128,
     "num_beams": 3,
     "early_stopping": False,