ant/pwm/config.json

{
  "name": "ant_pwm_world_model",
  "env": "Ant-v5",
  "model_type": "pwm",
  "metadata": {
    "latent_dim": 512
  },

  "models": [
    {
      "name": "encoder",
      "framework": null,
      "format": "jit",
      "source": {
        "path": "encoder.pt"
      },
      "signature": {
        "inputs": ["obs"]
      }
    },
    {
      "name": "dynamics",
      "framework": null,
      "format": "jit",
      "source": {
        "path": "transition.pt"
      },
      "signature": {
        "inputs": ["z", "a"]
      }
    },
    {
      "name": "reward",
      "framework": null,
      "format": "jit",
      "source": {
        "path": "reward.pt"
      },
      "signature": {
        "inputs": ["z", "a"]
      }
    }
  ]
}