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