Update.

diff --git a/main.py b/main.py
index 120e19c..c34eef6 100755 (executable)
--- 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")