| { | |
| "action_dim": 2, | |
| "architectures": [ | |
| "SIADWorldModel" | |
| ], | |
| "dropout": 0.1, | |
| "dtype": "float32", | |
| "ema_decay": 0.996, | |
| "ema_warmup_steps": 1000, | |
| "encoder_blocks": 8, | |
| "encoder_heads": 16, | |
| "encoder_mlp_dim": 4096, | |
| "in_channels": 8, | |
| "latent_dim": 1024, | |
| "model_type": "siad-world-model", | |
| "transformers_version": "5.2.0", | |
| "transition_blocks": 12, | |
| "transition_heads": 16, | |
| "transition_mlp_dim": 4096, | |
| "use_decoder": true | |
| } | |