Dreamer_PyTorch

Unofficial re-implementation of "Dream to Control: Learning Behaviors by Latent Imagination" (https://arxiv.org/abs/1912.01603 ). Work in progress

Instructions

For training, install the requirements (see below) and run

python3 train.py

Requirements

• Python3
• Mujoco (for DeepMind Control Suite)

and see requirements.txt for required python library

References

• Dream to Control: Learning Behaviors by Latent Imagination
• Official Implementation
• Official Implementation2
• My Implementation of PlaNet

TODO

• Speed up training
• Add comments and Improve readability
• Add results of experiments and video prediction
• Generalize code for other environments