config.json

{
  "radio_dim": 2304,
  "nemotron_dim": 4480,
  "state_dim": 39,
  "action_dim": 4,
  "d_model": 256,
  "n_heads": 4,
  "diffusion_T": 20,
  "epoch": 79,
  "loss": 0.11484310122542694
}