ant/weights_keras/config.json

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

  "models": [
    {
      "name": "encoder",
      "framework": null,
      "format": "saved_model",
      "source": {
        "path": "encoder.h5"
      },
      "signature": {
        "inputs": ["obs"]
      }
    },
    {
      "name": "dynamics",
      "framework": null,
      "format": "saved_model",
      "source": {
        "path": "transition_kwargs.h5"
      },
      "signature": {
        "inputs": ["z", "a"],
        "call_mode": "kwargs"
      }
    },
    {
      "name": "reward",
      "framework": null,
      "format": "saved_model",
      "source": {
        "path": "reward_kwargs.h5"
      },
      "signature": {
        "inputs": ["z", "a"],
        "call_mode": "kwargs"
      }
    }
  ]
}