Skip to content

Latest commit

 

History

History
27 lines (21 loc) · 878 Bytes

README.md

File metadata and controls

27 lines (21 loc) · 878 Bytes

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

TODO

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