Our implementation of "Dream to Control: Learning Behaviors by Latent Imagination" in Pytorch

Key components

RSSM

Transition model, rewards model

Actor-Critic Model

TO DO

Debug error with fixed episode length that returns done and some phases done is not included into calculating the reward structure which can be weird