data_config.json

{
  "batch_size": 64,
  "num_workers": 4,
  "shuffle_buffer_size": 1000,
  "normalization_mapping": {
    "ACTION": "MEAN_STD",
    "ENV": "MEAN_STD",
    "STATE": "MEAN_STD",
    "VISUAL": "IDENTITY"
  },
  "features": {
    "action": {
      "type": "ACTION",
      "shape": "(20,)"
    },
    "observation.image.0": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.image.1": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.image.2": {
      "type": "VISUAL",
      "shape": "(3, 224, 224)"
    },
    "observation.state": {
      "type": "STATE",
      "shape": "(20,)"
    }
  },
  "clip_values": null,
  "chunk_size": null,
  "action_time_horizon_s": null,
  "min_action_chunk_size": 8,
  "actionchunk_actions_are_absolute": false,
  "observation_mapping": {
    "action.ee_6d_pos": "action",
    "observation.ee_6d_pos": "observation.state",
    "observation.images.head": "observation.image.0",
    "observation.images.left_wrist": "observation.image.1",
    "observation.images.right_wrist": "observation.image.2"
  },
  "observation_whitelist": null,
  "observation_blacklist": null,
  "transform_mapping": {
    "observation.image.0": [
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ],
    "observation.image.1": [
      {
        "type": "resize_with_padding",
        "height": 224,
        "width": 224,
        "mode": "bilinear",
        "align_corners": false,
        "input_type": "Tensor"
      },
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ],
    "observation.image.2": [
      {
        "type": "resize_with_padding",
        "height": 224,
        "width": 224,
        "mode": "bilinear",
        "align_corners": false,
        "input_type": "Tensor"
      },
      {
        "type": "color_jitter",
        "brightness": 0.2,
        "contrast": [
          0.8,
          1.2
        ],
        "saturation": [
          0.8,
          1.2
        ],
        "hue": 0.05,
        "input_type": "Tensor"
      }
    ]
  },
  "language_instruction": null,
  "training_mode": "ROBOT_FLOWMATCH",
  "action_type": "EE_6D_POS",
  "serialize_stats": true,
  "video_decoder_cache_size": null,
  "repo_id": "roboeval_lift_pot",
  "video_root_dir": null,
  "streaming": false,
  "tolerance_s": 0.1,
  "use_imagenet_stats": false,
  "split_name": "train",
  "prefetch_factor": 2,
  "max_episodes": null
}