golden/groot_n17_boundary_shapes.json

{
  "model": "5685905b32e6d6cfe8b18811ac0d95770a4f4dfc",
  "dataset": "geonmin-kim/SO101-lv4-3color-cube-mat-to-mat-release-3regions-0908-merged",
  "dataset_dir": "datasets--geonmin-kim--SO101-lv4-3color-cube-mat-to-mat-release-3regions-0908-merged",
  "frame": 0,
  "seed": 0,
  "counts": {
    "backbone": 1,
    "dit": 4
  },
  "seq_lengths_seen": [
    93
  ],
  "suite_prompt_lengths": {
    "Move the red cube from the right mat to the left mat.": 93,
    "Move the red cube from the left mat to the right mat.": 93,
    "Move the green cube from the left mat to the right mat.": 93,
    "Move the green cube from the right mat to the left mat.": 93,
    "Move the blue cube from the left mat to the right mat.": 93,
    "Move the blue cube from the right mat to the left mat.": 93
  },
  "shapes": {
    "backbone_in.arg0.action_mask": [
      1,
      32
    ],
    "backbone_in.arg0.attention_mask": [
      1,
      93
    ],
    "backbone_in.arg0.embodiment_id": [
      1
    ],
    "backbone_in.arg0.image_grid_thw": [
      2,
      3
    ],
    "backbone_in.arg0.input_ids": [
      1,
      93
    ],
    "backbone_in.arg0.mm_token_type_ids": [
      1,
      93
    ],
    "backbone_in.arg0.pixel_values": [
      288,
      1536
    ],
    "backbone_in.arg0.state": [
      1,
      1,
      132
    ],
    "backbone_out.backbone_attention_mask": [
      1,
      93
    ],
    "backbone_out.backbone_features": [
      1,
      93,
      2048
    ],
    "backbone_out.image_mask": [
      1,
      93
    ],
    "dit_in.backbone_attention_mask": [
      1,
      93
    ],
    "dit_in.encoder_hidden_states": [
      1,
      93,
      2048
    ],
    "dit_in.hidden_states": [
      1,
      41,
      1536
    ],
    "dit_in.image_mask": [
      1,
      93
    ],
    "dit_in.timestep": [
      1
    ],
    "dit_out": [
      1,
      41,
      1024
    ]
  }
}