From c807a75f34cf3fddc0845d4af39d0894464ad29a Mon Sep 17 00:00:00 2001
From: Eren G??lge <egolge@coqui.ai>
Date: Mon, 28 Aug 2023 12:50:22 +0200
Subject: [PATCH] Fixup saving best model

---
 trainer/io.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trainer/io.py b/trainer/io.py
index 6e08aea..dfda581 100644
--- a/trainer/io.py
+++ b/trainer/io.py
@@ -180,7 +180,7 @@ def save_best_model(
     save_func=None,
     **kwargs,
 ):
-    if current_loss < best_loss:
+    if current_loss < best_loss and current_step > keep_after:
         best_model_name = f"best_model_{current_step}.pth"
         checkpoint_path = os.path.join(out_path, best_model_name)
         logger.info(" > BEST MODEL : %s", checkpoint_path)