rtx.yaml

data_type: "mix"
lmdb_dir: "/group/40007/public_datasets/open-x-embodiment/lmdb_datasets/" 
video_dir: "/group/40007/public_datasets/open-x-embodiment/video_datasets/"
rgb_shape: [224, 224]
skip_frame: 3
weighted: false
sub_data_configs: [
  {
    data_type: "oxe",
    data_name: "fractal20220817_data",
    weight: 150
  },
  {
    data_type: "oxe",
    data_name: "bridge",
    weight: 50
  },
  {
    data_type: "oxe",
    data_name: "taco_play",
    weight: 5
  },
  {
    data_type: "oxe",
    data_name: "jaco_play",
    weight: 20
  },
  {
    data_type: "oxe",
    data_name: "berkeley_cable_routing",
    weight: 20
  },
  {
    data_type: "oxe",
    data_name: "roboturk",
    weight: 10
  },
  {
    data_type: "oxe",
    data_name: "nyu_door_opening_surprising_effectiveness",
    weight: 5
  },
  {
    data_type: "oxe",
    data_name: "viola",
    weight: 3
  },
  {
    data_type: "oxe",
    data_name: "berkeley_autolab_ur5",
    weight: 5
  },
  {
    data_type: "oxe",
    data_name: "toto",
    weight: 5
  }
]