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
a4ef557
..
38c85ed
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-1885,10
+1885,10
@@
class Escape(Task):
self.width = width
states, actions, rewards = escape.generate_episodes(
self.width = width
states, actions, rewards = escape.generate_episodes(
- nb_train_samples + nb_test_samples, height, width,
3 *
T
+ nb_train_samples + nb_test_samples, height, width, T
)
seq = escape.episodes2seq(states, actions, rewards, lookahead_delta=T)
)
seq = escape.episodes2seq(states, actions, rewards, lookahead_delta=T)
- seq = seq[:, seq.size(1) // 3 : 2 * seq.size(1) // 3]
+
#
seq = seq[:, seq.size(1) // 3 : 2 * seq.size(1) // 3]
self.train_input = seq[:nb_train_samples].to(self.device)
self.test_input = seq[nb_train_samples:].to(self.device)
self.train_input = seq[:nb_train_samples].to(self.device)
self.test_input = seq[nb_train_samples:].to(self.device)