vla_config.json

{
  "schema_version": "task23_self_contained_policy_v1",
  "model_type": "cnn_conditioned_action_chunk",
  "variant": "small_cnn",
  "chunk": 1,
  "text_dim": 128,
  "proprio_dim": 32,
  "task_count": 14,
  "difficulty_count": 5,
  "proprio_mean": [
    -0.013340266421437263,
    0.7383145093917847,
    0.011660933494567871,
    -1.9436376094818115,
    0.019467052072286606,
    2.826350212097168,
    0.7630648612976074,
    0.0225653275847435,
    0.052658289670944214,
    0.009153264574706554,
    0.07257223129272461,
    -0.0217689611017704,
    -0.014813058078289032,
    0.045801397413015366,
    0.07692108303308487,
    -0.06413360685110092,
    0.9298723936080933,
    0.9961475133895874,
    -0.004522569477558136,
    0.07633855938911438,
    -0.005218956153839827,
    0.03139699995517731,
    -0.030553586781024933,
    0.0001240656420122832,
    5.485760084411595e-06,
    0.49797627329826355,
    0.0,
    0.0,
    0.0,
    0.0,
    0.0,
    0.0
  ],
  "proprio_std": [
    0.18346060812473297,
    0.2751201093196869,
    0.08362191170454025,
    0.4013335406780243,
    0.1880636066198349,
    0.22030314803123474,
    0.36815357208251953,
    0.09236042201519012,
    0.24793101847171783,
    0.07140849530696869,
    0.27341264486312866,
    0.12188053876161575,
    0.17399784922599792,
    0.20990528166294098,
    0.09051761776208878,
    0.1660534143447876,
    0.07342937588691711,
    0.0047762468457221985,
    0.021420147269964218,
    0.03227067366242409,
    0.016867103055119514,
    0.008550216443836689,
    0.009234834462404251,
    0.015089103952050209,
    0.012418420054018497,
    0.2886728048324585,
    1.0,
    1.0,
    1.0,
    1.0,
    1.0,
    1.0
  ],
  "task_to_id": {
    "lift_cube": 0,
    "pick_place_can": 1,
    "pick_place_milk": 2,
    "pick_place_bread": 3,
    "pick_place_cereal": 4,
    "pick_place_clutter": 5,
    "stack_blocks": 6,
    "open_door": 7,
    "nut_assembly": 8,
    "nut_assembly_square": 9,
    "nut_assembly_round": 10,
    "wipe_table": 11,
    "tool_hang": 12
  },
  "difficulty_to_id": {
    "low": 0,
    "medium": 1,
    "hard": 2,
    "very_high": 3
  },
  "adapter_inputs": [
    "image",
    "instruction",
    "proprio",
    "task",
    "step",
    "difficulty",
    "horizon"
  ],
  "forbidden_adapter_inputs": [
    "raw_obs",
    "object_state",
    "sim_state",
    "mujoco_data"
  ],
  "training_data": "data/hf",
  "trainval": false
}