RSSM

RSSMs for imitation learning.

API

from rssm import RSSMV1

reference = "wandb reference"
rssm = RSSMV1.load_from_wandb(reference=reference)

References

Paper

• Dream to Control: Learning Behaviors by Latent Imagination [Hafner+ 2019]
• Mastering Atari with Discrete World Models [Hafner+ 2021]
• Hierarchical Latent Dynamics Model with Multiple Timescales for Learning Long-Horizon Tasks [Fujii+ 2023]

Code

• google-research/dreamer
• danijar/dreamerv2
• julisforest/dreamer-pytorch
• pytorch/rl/dreamer