From 67454ab7aa1243ab8e1aa656fbf89605d42e2b0f Mon Sep 17 00:00:00 2001
From: Claudio Scheer <claudioscheer@protonmail.com>
Date: Tue, 2 Jun 2020 21:34:04 -0300
Subject: [PATCH] Add early_stopping_metric

---
 src/seq2seq/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/seq2seq/main.py b/src/seq2seq/main.py
index 1fa24d5..8202af9 100644
--- a/src/seq2seq/main.py
+++ b/src/seq2seq/main.py
@@ -49,6 +49,7 @@ model_args = {
     "num_beams": 3,
     "early_stopping": False,
     "learning_rate": 1e-4,
+    "early_stopping_metric": True,
 }
 
 model = Seq2SeqModel("bert", "bert-base-cased", "bert-base-cased", args=model_args)
-- 
GitLab