dreamer_v2_dmc_world_model(pretrained: bool = False, overrides: object = {})Construct the Control Suite configuration with all three objectives.
Model Size
Parameters
pretrainedbool= FalseNo weights are published for this family; passing
True raises.**overridesobject= {}Optional
DreamerV2Config field overrides.Returns
DreamerV2ForWorldModelingThe Control Suite trunk plus the objectives.
Notes
Reference: Hafner, Lillicrap, Norouzi, and Ba, "Mastering Atari with Discrete World Models", ICLR, 2021 (arXiv:2010.02193).
Action repeat of 2 belongs to this configuration and is not a model
field — pass it to lucid.utils.rollout.rollout.
Examples
>>> from lucid.models import dreamer_v2_dmc_world_model
>>> model = dreamer_v2_dmc_world_model(action_dim=6).eval()
>>> model.config.actor_entropy
0.0001