From 8c9501a5550e5f63649a9eabb57233d47f19b0a8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 1 Sep 2024 22:32:38 +0200 Subject: [PATCH] Update. --- main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/main.py b/main.py index 120e19c..c34eef6 100755 --- a/main.py +++ b/main.py @@ -1535,6 +1535,8 @@ for n_epoch in range(current_epoch, args.nb_epochs): c_quizzes = generate_ae_c_quizzes(models, local_device=main_device) time_c_quizzes = time.perf_counter() - start_time time_train = 0 + for m in models: + m.test_accuracy = 0 if c_quizzes is None: log_string("no_c_quiz") -- 2.39.5