Add Blockwise-OAT baseline artifacts (benchmarks/verify_dataset.json)

benchmarks/verify_dataset.json

@@ -0,0 +1,83 @@
+{
+  "action_max": 1.0,
+  "action_min": -1.0,
+  "action_shapes_seen": [
+    [
+      32,
+      7
+    ]
+  ],
+  "checkpoint": "output/baselines/original_oat/hf/policy_ep-0250_sr-0.596.ckpt",
+  "dataset_len": 124087,
+  "dataset_zarr_path": "/workspace/OAT-RoboMimic-Fine-tune/BLT-OAT/data/libero/libero10_N500.zarr",
+  "obs_dtypes_seen": {
+    "agentview_rgb": [
+      "torch.uint8"
+    ],
+    "robot0_eef_pos": [
+      "torch.float32"
+    ],
+    "robot0_eef_quat": [
+      "torch.float32"
+    ],
+    "robot0_eye_in_hand_rgb": [
+      "torch.uint8"
+    ],
+    "robot0_gripper_qpos": [
+      "torch.float32"
+    ],
+    "task_uid": [
+      "torch.int64"
+    ]
+  },
+  "obs_shapes_seen": {
+    "agentview_rgb": [
+      [
+        2,
+        128,
+        128,
+        3
+      ]
+    ],
+    "robot0_eef_pos": [
+      [
+        2,
+        3
+      ]
+    ],
+    "robot0_eef_quat": [
+      [
+        2,
+        4
+      ]
+    ],
+    "robot0_eye_in_hand_rgb": [
+      [
+        2,
+        128,
+        128,
+        3
+      ]
+    ],
+    "robot0_gripper_qpos": [
+      [
+        2,
+        2
+      ]
+    ],
+    "task_uid": [
+      [
+        2,
+        1
+      ]
+    ]
+  },
+  "passed": true,
+  "prefix_errors": [],
+  "prefix_samples_checked": 100,
+  "prefix_samples_ok": 100,
+  "schema_errors": [],
+  "schema_samples_checked": 64,
+  "smoke_subset": null,
+  "tokenizer_checkpoint": "output/baselines/original_oat/hf/tokenizer_ep-0950_mse-0.002.ckpt"
+}