Create config.json

ant/pwm_torch_seperate/config.json

@@ -0,0 +1,50 @@
+{
+  "name": "ant_pwm_world_model",
+  "env": "Ant-v5",
+  "model_type": "pwm",
+  "metadata": {
+    "latent_dim": 512
+  },
+
+  "models": [
+    {
+      "name": "encoder",
+      "framework": null,
+      "format": "jit",
+      "source": {
+        "class_path": "random_torch.py",
+        "class_name": "Encoder"
+        "weights_path": "encoder.pth"
+      },
+      "signature": {
+        "inputs": ["obs"]
+      }
+    },
+    {
+      "name": "dynamics",
+      "framework": null,
+      "format": "jit",
+      "source": {
+        "class_path": "random_torch.py",
+        "class_name": "TransitionModel"
+        "weights_path": "transition.pth"
+      },
+      "signature": {
+        "inputs": ["z", "a"]
+      }
+    },
+    {
+      "name": "reward",
+      "framework": null,
+      "format": "jit",
+      "source": {
+        "class_path": "random_torch.py",
+        "class_name": "RewardModel"
+        "weights_path": "reward.pth"
+      },
+      "signature": {
+        "inputs": ["z", "a"]
+      }
+    }
+  ]
+}