ant/weights_tensorflow_signatures/config.json

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

   "models": [
    {
      "name": "encoder",
      "framework": null,
      "format": "module",
      "source": {
        "object_ref": "encoder_saved_model"
      },
      "signature": {
        "inputs": ["obs"]
      }
    },
    {
      "name": "dynamics",
      "framework": null,
      "format": "saved_model",
      "source": {
        "path": "transition_positional.pb"
      },
      "signature": {
        "inputs": ["z", "a"],
        "call_mode": "positional"
      }
    },
    {
      "name": "reward",
      "framework": null,
      "format": "module",
      "source": {
        "object_ref": "reward_concat_saved_model"
      },
      "signature": {
        "inputs": ["z", "a"]
      }
    }
  ]
}