issues
search
world-modelz
/
dreamax
A scalable Dreamer implementation in JAX
MIT License
11
stars
2
forks
source link
Implement Dreamer V2
#12
Open
andreaskoepf
opened
2 years ago
andreaskoepf
commented
2 years ago
Identify common blocks and differences of Dreamer V1 & V2, read
Mastering Atari with Discrete World Models
paper and review the
official dreamer v2 implementation
.
Refactor code in order to use the same base infrastructure for dreamer V1 & dreamer V2 (e.g. environment, configuration, logging, training)
Implement Dreamer V2 as an optional additional algorithm (specified through the configuration system)