Update.
authorFrançois Fleuret <francois@fleuret.org>
Fri, 16 Aug 2024 18:45:50 +0000 (20:45 +0200)
committerFrançois Fleuret <francois@fleuret.org>
Fri, 16 Aug 2024 18:45:50 +0000 (20:45 +0200)
main.py

diff --git a/main.py b/main.py
index 133f536..3021ff8 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -1152,7 +1152,7 @@ for n_epoch in range(current_epoch, args.nb_epochs):
     # re-compute the test errors
 
     for model in models:
-        if model.test_accuracy >= model.best_test_accuracy:
+        if model.test_accuracy > model.best_test_accuracy:
             log_string(
                 f"storing_best model {model.id} accuracy {model.best_test_accuracy} -> {model.test_accuracy}"
             )