projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
tasks.py
diff --git
a/tasks.py
b/tasks.py
index
870ab95
..
8e8faa9
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-1951,8
+1951,11
@@
class Escape(Task):
# Generate iteration after iteration
result = self.test_input[:250].clone()
# Generate iteration after iteration
result = self.test_input[:250].clone()
+ # Erase all the content but that of the first iteration
result[:, self.it_len :] = -1
result[:, self.it_len :] = -1
+ # Set the lookahead_reward of the firs to UNKNOWN
result[:, self.index_lookahead_reward] = escape.lookahead_reward2code(2)
result[:, self.index_lookahead_reward] = escape.lookahead_reward2code(2)
+
t = torch.arange(result.size(1), device=result.device)[None, :]
for u in tqdm.tqdm(
t = torch.arange(result.size(1), device=result.device)[None, :]
for u in tqdm.tqdm(