Update delta-iris/crafter/config.json

delta-iris/crafter/config.json

@@ -0,0 +1,45 @@
+{
+  "name": "crafter_delta_iris_world_model",
+  "env": "CrafterReward-v1",
+  "model_type": "delta_iris",
+  "metadata": {
+  },
+
+  "models": [
+    {
+      "name": "tokenizer",
+      "framework": null,
+      "format": "state_dict",
+      "source": {
+        "weights_path": "last.pt",
+        "class_path": "../src/world_model.py",
+        "class_name": "WorldModel"
+      },
+      "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"]
+      }
+    }
+  ]
+}