From: François Fleuret <francois@fleuret.org>
Date: Thu, 4 Jul 2024 04:37:17 +0000 (+0300)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=dfa00f17ae94d481ca0f8fd6ce96dbcaa4bbe06e;p=culture.git

Update.
---

diff --git a/reasoning.py b/reasoning.py
index 09e5362..2874adc 100755
--- a/reasoning.py
+++ b/reasoning.py
@@ -467,8 +467,8 @@ if __name__ == "__main__":
     delay = time.perf_counter() - start_time
     print(f"{prompts.size(0)/delay:02f} seq/s")
 
-    # predicted_prompts = torch.rand(prompts.size(0)) < 0.5
-    # predicted_answers = torch.logical_not(predicted_prompts)
+    predicted_prompts = torch.rand(prompts.size(0)) < 0.5
+    predicted_answers = torch.logical_not(predicted_prompts)
 
     reasoning.save_quizzes(
         "/tmp",
@@ -476,5 +476,6 @@ if __name__ == "__main__":
         prompts[:64],
         answers[:64],
         # You can add a bool to put a frame around the predicted parts
-        # predicted_prompts, predicted_answers
+        predicted_prompts[:64],
+        predicted_answers[:64],
     )