config.json

{
  "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
  }
}