-
Notifications
You must be signed in to change notification settings - Fork 39
Open
Description
dreamer-pytorch/dreamer/algos/dreamer_algo.py
Line 194 in 47bd509
| observation = samples.all_observation[:-1] # [t, t+batch_length+1] -> [t, t+batch_length] |
in there, observation shape is [15, 50, 3, 64, 64].
dreamer-pytorch/dreamer/algos/dreamer_algo.py
Line 145 in 47bd509
| samples_from_replay = self.replay_buffer.sample_batch(self._batch_size, self.batch_length) |
but i guess you intended to get [50, 50, 3, 64, 64]. what do you think?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels