From a6989b841f71f000132ed0b4b9b6ed4c4d23a74e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=A4x=20K=2E?= <max.kiehnscherf@gmail.com>
Date: Tue, 27 Aug 2024 16:22:20 +0200
Subject: [PATCH] [fix] pass patience arg to EarlyStopping class (#152)

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 29094bb7..eb8a2054 100644
--- a/train.py
+++ b/train.py
@@ -407,7 +407,7 @@ def main(args):
         scheduler = None
 
     save_best_model = SaveBestModel()
-    early_stopping = EarlyStopping()
+    early_stopping = EarlyStopping(patience=args['patience'])
 
     for epoch in range(start_epochs, NUM_EPOCHS):
         train_loss_hist.reset()