From 926d9cdfdedc8b5d4bdaa7c724fd852d2091d887 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 18 Aug 2024 08:44:23 +0200 Subject: [PATCH] Update. --- main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/main.py b/main.py index 127b71b..db16214 100755 --- a/main.py +++ b/main.py @@ -1122,6 +1122,9 @@ for n_epoch in range(current_epoch, args.nb_epochs): cta = " ".join([f"{float(m.test_accuracy):.04f}" for m in models]) log_string(f"current_test_accuracies {cta}") + cta = " ".join([f"{float(m.best_test_accuracy):.04f}" for m in models]) + log_string(f"current_best_test_accuracies {cta}") + ################################################## # If all the models are good enough, generate new quizzes and # re-compute the test errors @@ -1133,6 +1136,7 @@ for n_epoch in range(current_epoch, args.nb_epochs): ) model.best_dict = copy.deepcopy(model.state_dict()) model.best_test_accuracy = model.test_accuracy + model.test_accuracy = 0.0 # we restart if total_time_generating_c_quizzes == 0: -- 2.39.5