Update.

diff --git a/main.py b/main.py
index b372f12..02259b2 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -88,7 +88,7 @@ parser.add_argument("--proba_understands", type=float, default=0.9)
 
 parser.add_argument("--proba_not_understands", type=float, default=0.5)
 
-parser.add_argument("--generation_temperature", type=float, default=1.0)
+parser.add_argument("--generation_temperature", type=float, default=2)
 
 parser.add_argument("--dirty_debug", action="store_true", default=False)
 

diff --git a/quiz_machine.py b/quiz_machine.py
index 70daa0b..aedb72f 100755 (executable)
--- a/quiz_machine.py
+++ b/quiz_machine.py
@@ -590,7 +590,7 @@ class QuizMachine:
             input=c_quizzes,
             ar_mask=self.make_ar_mask(c_quizzes),
             seq_logproba=seq_logproba,
-            temperature=1 / temperature,
+            temperature=1.0,
             deterministic_synthesis=False,
             device=self.device,
         )
@@ -606,7 +606,7 @@ class QuizMachine:
             input=c_quizzes,
             ar_mask=self.make_ar_mask(c_quizzes),
             seq_logproba=seq_logproba,
-            temperature=1 / temperature,
+            temperature=1,
             deterministic_synthesis=False,
             device=self.device,
         )