world-models-Breakout / config.json
Basem1166's picture
Initial World Models checkpoint upload
5216407 verified
raw
history blame contribute delete
456 Bytes
{
"architecture": "WorldModels",
"latent_size": 64,
"hidden_size": 256,
"action_size": 4,
"n_gaussians": 5,
"components": {
"vae": "Variational Autoencoder for visual compression",
"rnn": "MDRNN (Memory-Augmented Recurrent Neural Network)",
"controller": "CMA-ES optimized linear controller"
},
"trained_on": "Atari environment (Breakout)",
"upload_date": "2025-12-25T08:13:58.344484",
"framework": "PyTorch"
}