| """PlaNet RSSM config.""" | |
| from experiments.shared.src.models.image_world_models import ImageWorldModelConfig | |
| def default_config(): | |
| return ImageWorldModelConfig(emb_dim=96, z_dim=108, c_dim=8, hidden_dim=120, history_len=8, context_len=8) | |
| """PlaNet RSSM config.""" | |
| from experiments.shared.src.models.image_world_models import ImageWorldModelConfig | |
| def default_config(): | |
| return ImageWorldModelConfig(emb_dim=96, z_dim=108, c_dim=8, hidden_dim=120, history_len=8, context_len=8) | |