From f27e43ee3ebf9e1b5587eccb4edd0a492cbbfa0b Mon Sep 17 00:00:00 2001
From: Claudio Scheer <claudioscheer@protonmail.com>
Date: Fri, 5 Jun 2020 13:14:03 -0300
Subject: [PATCH] Increase training epochs
---
src/seq2seq/main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/seq2seq/main.py b/src/seq2seq/main.py
index 1b9040c..ed33eae 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,
--
GitLab