Dream to Control: Learning Behaviors by Latent Imagination
Paper: https://arxiv.org/abs/1912.01603
Project Website: https://danijar.com/project/dreamer/
TensorFlow 2 implementation: https://github.com/danijar/dreamer
TensorFlow 1 implementation: https://github.com/google-research/dreamer
- Clone the rlpyt framework with
git clone https://github.com/astooke/rlpyt.git
and copy therlpyt
sub-directory into this directory. - Or install directly from github with
pip install git+https://github.com/astooke/rlpyt.git
(you may need --user flag when installing with pip)
To run tests, install pytest: pip install pytest
Install PyTorch according to their website
Install gym with pip install gym
pip install atari_py
opencv:
- anaconda: https://anaconda.org/conda-forge/opencv
- pip:
pip install opencv-python
psutil: pip install psutil
pyprind: pip install pyprind
pip install mujoco_py
You must have a mujoco license
To run tests:
pytest tests
If you want additional code coverage information:
pytest tests --cov=dreamer
Use PEP8 style for python syntax. (ctrl-alt-l in PyCharm)
Contact [email protected] or send in a pull request.