From 5f69573b6481bcc1fe4359ab595e129411f781ae Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sun, 28 Jul 2024 08:22:43 +0200 Subject: [PATCH] Update. --- main.py | 8 ++++---- quiz_machine.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 47952af..83fd8b8 100755 --- a/main.py +++ b/main.py @@ -97,9 +97,9 @@ parser.add_argument("--proba_not_understands", type=float, default=0.5) parser.add_argument("--temperature_hot", type=float, default=1.25) -parser.add_argument("--temperature_cold", type=float, default=1.25) +parser.add_argument("--temperature_cold", type=float, default=1) -parser.add_argument("--nb_rounds", type=int, default=2) +parser.add_argument("--nb_rounds", type=int, default=1) parser.add_argument("--c_quiz_validation_mode", type=str, default="predict") @@ -454,13 +454,13 @@ def one_epoch(model, quiz_machine, local_device=main_device): lt_noisy = lambda s, logits: logits / args.temperature_hot lt_clean = lambda s, logits: logits / args.temperature_cold -c_quizzes_procedure_ = [ +c_quizzes_procedure = [ (("f_B", "f_A", "A", "B"), (1, 0, 0, 0), lt_noisy), (("f_B", "f_A", "A", "B"), (0, 1, 1, 1), lt_clean), (("A", "f_A", "B", "f_B"), (0, 0, 0, 1), lt_clean), ] -c_quizzes_procedure = [ +c_quizzes_procedure_ = [ (("A", "f_A", "B", "f_B"), (1, 1, 0, 0), lt_noisy), (("A", "f_A", "B", "f_B"), (0, 0, 1, 1), lt_clean), ] diff --git a/quiz_machine.py b/quiz_machine.py index 18c0828..ba3387c 100755 --- a/quiz_machine.py +++ b/quiz_machine.py @@ -459,7 +459,7 @@ class QuizMachine: c_quizzes[: to_recycle.size(0)] = to_recycle to_recycle = None - c_quizzes = self.problem.reconfigure(c_quizzes, ("A", "f_A", "B", "f_B")) + c_quizzes = self.problem.reconfigure(c_quizzes, ("A", "f_A", "B", "f_B")) return c_quizzes.to("cpu") -- 2.39.5