From b41d98a66bdeb0282a8bf2268d658ab60081a819 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Wed, 17 Jul 2024 05:53:27 +0200 Subject: [PATCH] Update. --- quiz_machine.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quiz_machine.py b/quiz_machine.py index 1168921..14e8739 100755 --- 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] -- 2.20.1