RSSMs for imitation learning.
from rssm import RSSMV1
reference = "wandb reference"
rssm = RSSMV1.load_from_wandb(reference=reference)
- 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]