File size: 1,429 Bytes
5b77d1c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
{
  "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"
}