dreamer
PyTorch Lightning implementation of the DreamerV2
This repository contains PyTorch Lightning implementation of the world model in DreamerV2. For more technical details, check out the papers listed in References.
Requirements
References
- World Models, David Ha, Jürgen Schmidhuber, 2018
- Mastering Atari with Discrete World Models, Danijar Hafner, Timothy Lillicrap, Mohammad Norouzi, Jimmy Ba, 2021