Update.

diff --git a/main.py b/main.py
index 47952af..83fd8b8 100755 (executable)
--- 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 (executable)
--- 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")