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