[fix] pass patience arg to EarlyStopping class (#152)

sovit-123 · Aug 27, 2024 · a6989b8 · a6989b8
1 parent c8e2e43
commit a6989b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()