From baa8fb5f42ee884ca858bdb91653fb67f27be9d7 Mon Sep 17 00:00:00 2001
From: till-m <36440677+till-m@users.noreply.github.com>
Date: Mon, 20 Oct 2025 16:55:41 +0200
Subject: [PATCH] fix overwriting `best.pt` every validation

---
 the_well/benchmark/trainer/training.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/the_well/benchmark/trainer/training.py b/the_well/benchmark/trainer/training.py
index 772dfa41..c7200083 100644
--- a/the_well/benchmark/trainer/training.py
+++ b/the_well/benchmark/trainer/training.py
@@ -460,6 +460,7 @@ def train(self):
                     self.save_model(
                         epoch, val_loss, os.path.join(self.checkpoint_folder, "best.pt")
                     )
+                    self.best_val_loss = val_loss
             # Check if time for expensive validation - periodic or final
             if epoch % self.rollout_val_frequency == 0 or (epoch == self.max_epoch):
                 logger.info(