From: François Fleuret <francois@fleuret.org>
Date: Sun, 24 Mar 2024 22:11:55 +0000 (+0100)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=9eadd2cd6913a0de53b4b0f526157497e8d14381;p=picoclvr.git

Update.
---

diff --git a/tasks.py b/tasks.py
index 56c2b0f..fddcaff 100755
--- a/tasks.py
+++ b/tasks.py
@@ -1939,7 +1939,7 @@ class Escape(Task):
         ):
             # Put the lookahead reward to either 0 or -1 for the
             # current iteration, sample the next state
-            s = -1  # (torch.rand(result.size(0), device = result.device) < 0.2).long()
+            s = -(torch.rand(result.size(0), device=result.device) < 0.2).long()
             result[:, u - 1] = s + 1 + escape.first_lookahead_rewards_code
             ar_mask = (t >= u).long() * (t < u + state_len).long()
             ar(result, ar_mask)
@@ -1980,7 +1980,7 @@ class Escape(Task):
     def produce_results(
         self, n_epoch, model, result_dir, logger, deterministic_synthesis, nmax=1000
     ):
-        result = self.test_input[:100].clone()
+        result = self.test_input[:250].clone()
 
         # Saving the ground truth