| { | |
| "obs_dim": 20, | |
| "action_dim": 8, | |
| "latent_dim": 128, | |
| "hidden_dim": 512, | |
| "encoder_hidden": 256, | |
| "decoder_hidden": 256, | |
| "dropout": 0.1, | |
| "ema_momentum": 0.996, | |
| "rollout_horizon": 5, | |
| "rollout_discount": 0.95, | |
| "rollout_weight": 0.5, | |
| "epoch": 93, | |
| "param_count": 2802838, | |
| "metrics": { | |
| "total_loss": 0.3123664617538452, | |
| "recon_loss": 0.13891788125038146, | |
| "kl_loss": 0.010396031755954027, | |
| "reward_loss": 0.08199895620346069, | |
| "done_loss": 0.06397444047033787, | |
| "rollout_loss": 0.32944561541080475 | |
| } | |
| } |