search

yusukeurakami / dreamer-pytorch

pytorch-implementation of Dreamer (Model-based Image RL Algorithm)
MIT License
162 stars 34 forks source link

Input to imagine_ahead in utils.py #10

Open aravindvenu7 opened 3 years ago

aravindvenu7 commented 3 years ago

Input should contain only the initial belief and state right? Here, input contains the entire sequence of beliefs and states instead. Not sure how this works with the algorithm