pick best model by loss function is run_eval is true

2026-09-11 12:00:24 +08:00 · 2019-02-05 11:56:44 +01:00
parent e12bbc2a5c
commit 0308c92c16
1 changed files with 4 additions and 1 deletions
@@ -429,7 +429,10 @@ def main(args):
            " | > Train Loss: {:.5f}   Validation Loss: {:.5f}".format(
                train_loss, val_loss),
            flush=True)
-        best_loss = save_best_model(model, optimizer, train_loss, best_loss,
+        target_loss = train_loss
+        if c.run_eval:
+            target_loss = val_loss
+        best_loss = save_best_model(model, optimizer, target_loss, best_loss,
                                    OUT_PATH, current_step, epoch)