Update.

diff --git a/quiz_machine.py b/quiz_machine.py
index 1168921..14e8739 100755 (executable)
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -365,7 +365,7 @@ class QuizMachine:
                 seq_logproba=seq_logproba,
                 temperature=1.0,
                 deterministic_synthesis=deterministic_synthesis,
-                progress_bar_desc=None,
+                progress_bar_desc="accuracy",
                 device=self.device,
             )
 
@@ -402,7 +402,7 @@ class QuizMachine:
             model.test_w_quizzes[:2000], log_prefix="test"
         )
 
-        n_test_forward = model.test_w_quizzes[:, 0] == self.token_forward
+        n_test_forward = model.test_w_quizzes[:2000, 0] == self.token_forward
 
         forward_test_correct = test_correct[n_test_forward]