Upload folder using huggingface_hub

config.json

@@ -0,0 +1,120 @@
+{
+  "action_dim": 32,
+  "action_head_cfg": {
+    "_convert_": "object",
+    "_target_": "groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHead",
+    "config": {
+      "_recursive_": false,
+      "_target_": "groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHeadConfig",
+      "action_dim": 32,
+      "action_horizon": 16,
+      "action_loss_embodiment_ids": [
+        26,
+        17,
+        32
+      ],
+      "add_pos_embed": true,
+      "backbone_embedding_dim": 0,
+      "backbone_features_projector_cfg": null,
+      "decouple_video_action_noise": false,
+      "diffusion_model_cfg": {
+        "_convert_": "object",
+        "_target_": "groot.vla.model.dreamzero.modules.wan_video_dit_action_casual_chunk.CausalWanModel",
+        "concat_first_frame_latent": false,
+        "diffusion_model_pretrained_path": "/mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B",
+        "dim": 3072,
+        "eps": 1e-06,
+        "ffn_dim": 14336,
+        "frame_seqlen": 50,
+        "freq_dim": 256,
+        "in_dim": 48,
+        "max_chunk_size": 4,
+        "model_type": "ti2v",
+        "num_action_per_block": 16,
+        "num_frame_per_block": 2,
+        "num_heads": 24,
+        "num_layers": 30,
+        "num_state_per_block": 1,
+        "out_dim": 48
+      },
+      "expand_batch": null,
+      "freeze_decode_layer": false,
+      "hidden_size": 64,
+      "image_encoder_cfg": {
+        "_convert_": "object",
+        "_target_": "groot.vla.model.dreamzero.modules.wan_video_image_encoder.WanImageEncoder",
+        "image_encoder_pretrained_path": "/mnt/project_rlinf_hs/yuanhuining/models/Wan2.1-I2V-14B-480P/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth"
+      },
+      "init_lora_weights": "kaiming",
+      "input_embedding_dim": 1536,
+      "lambda_tgm_action": 0.0,
+      "lambda_tgm_video": 0.0,
+      "load_pretrained_det_decode_layer_path": null,
+      "lora_alpha": 4,
+      "lora_rank": 4,
+      "lora_target_modules": "q,k,v,o,ffn.0,ffn.2",
+      "max_action_dim": 32,
+      "max_state_dim": 64,
+      "model_dtype": "float32",
+      "noise_beta_alpha": 1.5,
+      "noise_beta_beta": 1.0,
+      "noise_s": 0.999,
+      "num_frame_per_block": 2,
+      "num_frames": 33,
+      "num_inference_timesteps": 4,
+      "num_timestep_buckets": 1000,
+      "repa_coeff": 1.0,
+      "repa_layer": 8,
+      "target_video_height": 160,
+      "target_video_width": 320,
+      "text_encoder_cfg": {
+        "_convert_": "object",
+        "_target_": "groot.vla.model.dreamzero.modules.wan_video_text_encoder.WanTextEncoder",
+        "text_encoder_pretrained_path": "/mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/models_t5_umt5-xxl-enc-bf16.pth"
+      },
+      "tgm_couple_prob": 0.5,
+      "tgm_sigma": 0.5,
+      "tile_size_height": 34,
+      "tile_size_width": 34,
+      "tile_stride_height": 18,
+      "tile_stride_width": 16,
+      "tiled": false,
+      "train_architecture": "full",
+      "tune_diffusion_model": true,
+      "tune_projector": true,
+      "use_gradient_checkpointing": true,
+      "use_vlln": true,
+      "vae_cfg": {
+        "_convert_": "object",
+        "_target_": "groot.vla.model.dreamzero.modules.wan_video_vae.WanVideoVAE38",
+        "dim": 160,
+        "vae_pretrained_path": "/mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/Wan2.2_VAE.pth",
+        "z_dim": 48
+      },
+      "video_noise_beta_alpha": 3.0,
+      "video_noise_beta_beta": 1.0,
+      "vl_self_attention_cfg": {
+        "_target_": "groot.vla.model.n1_5.modules.cross_attention_dit.SelfAttentionTransformer",
+        "attention_head_dim": 64,
+        "dropout": 0.2,
+        "final_dropout": true,
+        "num_attention_heads": 24,
+        "num_layers": 4,
+        "positional_embeddings": null
+      }
+    }
+  },
+  "action_horizon": 16,
+  "architectures": [
+    "VLA"
+  ],
+  "backbone_cfg": {
+    "_target_": "groot.vla.model.dreamzero.backbone.identity.IdentityBackbone"
+  },
+  "hidden_size": 0,
+  "model_dtype": "float32",
+  "model_type": "vla",
+  "resume_path": "/mnt/project_rlinf/yuanhuining/repos/dreamzero/runs/train/2026-05-07T16-36-40Z/checkpoints",
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3"
+}

experiment_cfg/conf.yaml

@@ -0,0 +1,3131 @@
+model:
+  _target_: groot.vla.model.dreamzero.base_vla.VLA
+  _convert_: object
+  config:
+    _target_: groot.vla.model.dreamzero.base_vla.VLAConfig
+    _recursive_: false
+    model_dtype: float32
+    hidden_size: 0
+    action_horizon: 16
+    action_dim: 32
+    backbone_cfg:
+      _target_: groot.vla.model.dreamzero.backbone.identity.IdentityBackbone
+    action_head_cfg:
+      config:
+        backbone_features_projector_cfg: null
+        _target_: groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHeadConfig
+        _recursive_: false
+        tiled: false
+        tile_size_height: 34
+        tile_size_width: 34
+        tile_stride_height: 18
+        tile_stride_width: 16
+        lora_rank: 4
+        lora_alpha: 4
+        num_frames: 33
+        num_frame_per_block: 2
+        lora_target_modules: q,k,v,o,ffn.0,ffn.2
+        init_lora_weights: kaiming
+        train_architecture: full
+        use_gradient_checkpointing: true
+        add_pos_embed: true
+        model_dtype: float32
+        max_state_dim: 64
+        max_action_dim: 32
+        action_loss_embodiment_ids:
+        - 26
+        - 17
+        - 32
+        hidden_size: 64
+        input_embedding_dim: 1536
+        backbone_embedding_dim: 0
+        repa_layer: 8
+        repa_coeff: 1.0
+        load_pretrained_det_decode_layer_path: null
+        freeze_decode_layer: false
+        expand_batch: null
+        use_vlln: true
+        vl_self_attention_cfg:
+          _target_: groot.vla.model.n1_5.modules.cross_attention_dit.SelfAttentionTransformer
+          positional_embeddings: null
+          num_layers: 4
+          num_attention_heads: 24
+          attention_head_dim: 64
+          dropout: 0.2
+          final_dropout: true
+        diffusion_model_cfg:
+          _target_: groot.vla.model.dreamzero.modules.wan_video_dit_action_casual_chunk.CausalWanModel
+          _convert_: object
+          diffusion_model_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B
+          model_type: ti2v
+          frame_seqlen: 50
+          dim: 3072
+          in_dim: 48
+          ffn_dim: 14336
+          out_dim: 48
+          freq_dim: 256
+          eps: 1.0e-06
+          num_heads: 24
+          num_layers: 30
+          max_chunk_size: 4
+          num_frame_per_block: 2
+          num_action_per_block: 16
+          num_state_per_block: 1
+          concat_first_frame_latent: false
+        text_encoder_cfg:
+          _target_: groot.vla.model.dreamzero.modules.wan_video_text_encoder.WanTextEncoder
+          _convert_: object
+          text_encoder_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/models_t5_umt5-xxl-enc-bf16.pth
+        image_encoder_cfg:
+          _target_: groot.vla.model.dreamzero.modules.wan_video_image_encoder.WanImageEncoder
+          _convert_: object
+          image_encoder_pretrained_path: /mnt/project_rlinf_hs/yuanhuining/models/Wan2.1-I2V-14B-480P/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth
+        vae_cfg:
+          _target_: groot.vla.model.dreamzero.modules.wan_video_vae.WanVideoVAE38
+          _convert_: object
+          vae_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/Wan2.2_VAE.pth
+          z_dim: 48
+          dim: 160
+        action_dim: 32
+        action_horizon: 16
+        num_inference_timesteps: 4
+        noise_beta_alpha: 1.5
+        noise_beta_beta: 1.0
+        noise_s: 0.999
+        num_timestep_buckets: 1000
+        decouple_video_action_noise: false
+        video_noise_beta_alpha: 3.0
+        video_noise_beta_beta: 1.0
+        lambda_tgm_video: 0.0
+        lambda_tgm_action: 0.0
+        tgm_sigma: 0.5
+        tgm_couple_prob: 0.5
+        tune_projector: true
+        tune_diffusion_model: true
+        target_video_height: 160
+        target_video_width: 320
+      _target_: groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHead
+      _convert_: object
+train_dataset:
+  _target_: groot.vla.data.dataset.lerobot_sharded.ShardedLeRobotMixtureDataset.from_mixture_spec
+  _convert_: object
+  mixture_spec:
+  - dataset_path:
+      libero_sim:
+      - /mnt/project_rlinf_hs/yuanhuining/datasets/libero
+    dataset_weight: 1.0
+    distribute_weights: true
+  dataset_class: groot.vla.data.dataset.lerobot_sharded.ShardedLeRobotSubLangSingleActionChunkDatasetLIBERO
+  all_modality_configs:
+    oxe_droid:
+      video:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        - 24
+        eval_delta_indices:
+        - 0
+        modality_keys:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+      state:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - state.joint_position
+        - state.gripper_position
+      action:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        modality_keys:
+        - action.joint_position
+        - action.gripper_position
+      language:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.language.language_instruction
+        - annotation.language.language_instruction_2
+        - annotation.language.language_instruction_3
+      lapa_action:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - lapa_action
+    agibot:
+      video:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        - 24
+        eval_delta_indices:
+        - -3
+        - -2
+        - -1
+        - 0
+        modality_keys:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+      state:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - state.left_arm_joint_position
+        - state.right_arm_joint_position
+        - state.left_effector_position
+        - state.right_effector_position
+        - state.head_position
+        - state.waist_position
+      action:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        modality_keys:
+        - action.left_arm_joint_position
+        - action.right_arm_joint_position
+        - action.left_effector_position
+        - action.right_effector_position
+        - action.head_position
+        - action.waist_position
+        - action.robot_velocity
+      language:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.language.action_text
+    yam:
+      video:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        - 24
+        eval_delta_indices:
+        - 0
+        modality_keys:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+      state:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - state.left_joint_pos
+        - state.left_gripper_pos
+        - state.right_joint_pos
+        - state.right_gripper_pos
+      action:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        modality_keys:
+        - action.left_joint_pos
+        - action.left_gripper_pos
+        - action.right_joint_pos
+        - action.right_gripper_pos
+      language:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.task
+    libero_sim:
+      video:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        - 24
+        eval_delta_indices:
+        - 0
+        modality_keys:
+        - video.image
+        - video.wrist_image
+      state:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - state.state
+      action:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        - 1
+        - 2
+        - 3
+        - 4
+        - 5
+        - 6
+        - 7
+        - 8
+        - 9
+        - 10
+        - 11
+        - 12
+        - 13
+        - 14
+        - 15
+        - 16
+        - 17
+        - 18
+        - 19
+        - 20
+        - 21
+        - 22
+        - 23
+        modality_keys:
+        - action.actions
+      language:
+        _target_: groot.vla.data.dataset.ModalityConfig
+        delta_indices:
+        - 0
+        modality_keys:
+        - annotation.task
+  all_transforms:
+    oxe_droid:
+      _target_: groot.vla.data.transform.ComposedModalityTransform
+      transforms:
+      - _target_: groot.vla.data.transform.VideoToTensor
+        apply_to:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+      - _target_: groot.vla.data.transform.VideoCrop
+        apply_to:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+        scale: 0.95
+        mode: random
+      - _target_: groot.vla.data.transform.VideoResize
+        apply_to:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+        height: 256
+        width: 256
+        interpolation: linear
+      - _target_: groot.vla.data.transform.VideoColorJitter
+        apply_to:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+        brightness: 0.3
+        contrast: 0.4
+        saturation: 0.5
+        hue: 0.08
+      - _target_: groot.vla.data.transform.VideoToNumpy
+        apply_to:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - state.joint_position
+        - state.gripper_position
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - state.joint_position
+        - state.gripper_position
+        normalization_modes:
+          state.joint_position: q99
+          state.gripper_position: q99
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - action.joint_position
+        - action.gripper_position
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - action.joint_position
+        - action.gripper_position
+        normalization_modes:
+          action.joint_position: q99
+          action.gripper_position: q99
+      - _target_: groot.vla.data.transform.ConcatTransform
+        video_concat_order:
+        - video.exterior_image_1_left
+        - video.exterior_image_2_left
+        - video.wrist_image_left
+        state_concat_order:
+        - state.joint_position
+        - state.gripper_position
+        action_concat_order:
+        - action.joint_position
+        - action.gripper_position
+      - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+        default_instruction: Perform the default behavior.
+        language_dropout_prob: 0.0
+        always_use_default_instruction: false
+        max_state_dim: 64
+        max_action_dim: 32
+        max_length: 512
+        state_horizon: 1
+        action_horizon: 16
+        embodiment_tag_mapping:
+          real_gr1_arms_only: 0
+          real_gr1_arms_only_annotated: 1
+          real_gr1_arms_waist: 2
+          real_gr1_arms_waist_annotated: 3
+          dexmg_gr1_arms_only_inspire: 4
+          dexmg_gr1_arms_only_fourier: 5
+          dexmg_gr1_arms_waist_fourier: 6
+          robocasa_single_arm: 7
+          onex_eve_gripper: 8
+          robocasa_gr1_arms_only_inspire_hands: 9
+          robocasa_gr1_arms_only_fourier_hands: 10
+          robocasa_gr1_fixed_lower_body_inspire_hands: 11
+          robocasa_gr1_fixed_lower_body_fourier_hands: 12
+          robocasa_panda_omron: 13
+          robocasa_bimanual_panda_parallel_gripper: 15
+          robocasa_bimanual_panda_inspire_hand: 16
+          oxe_droid: 17
+          oxe_fractal: 18
+          oxe_language_table: 19
+          oxe_bridge: 20
+          real_panda_single_arm: 21
+          hot3d_hands_only: 23
+          gr1_unified: 24
+          robocasa_gr1_arms_waist_fourier_hands: 25
+          agibot: 26
+          lapa: 27
+          oxe_mutex: 28
+          oxe_roboset: 29
+          oxe_plex: 30
+          dream: 31
+          yam: 32
+          libero_sim: 21
+          xdof: 22
+          gr1_unified_segmentation: 14
+          language_table_sim: 7
+          gr1_isaac: 0
+          sim_behavior_r1_pro: 31
+          mecka_hands: 27
+          real_r1_pro_sharpa: 28
+        tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+    agibot:
+      _target_: groot.vla.data.transform.ComposedModalityTransform
+      transforms:
+      - _target_: groot.vla.data.transform.VideoToTensor
+        apply_to:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+      - _target_: groot.vla.data.transform.VideoCrop
+        apply_to:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+        scale: 0.95
+        mode: random
+      - _target_: groot.vla.data.transform.VideoResize
+        apply_to:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+        height: 256
+        width: 256
+        interpolation: linear
+      - _target_: groot.vla.data.transform.VideoColorJitter
+        apply_to:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+        brightness: 0.3
+        contrast: 0.4
+        saturation: 0.5
+        hue: 0.08
+      - _target_: groot.vla.data.transform.VideoToNumpy
+        apply_to:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - state.left_arm_joint_position
+        - state.right_arm_joint_position
+        - state.left_effector_position
+        - state.right_effector_position
+        - state.head_position
+        - state.waist_position
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - state.left_arm_joint_position
+        - state.right_arm_joint_position
+        - state.left_effector_position
+        - state.right_effector_position
+        - state.head_position
+        - state.waist_position
+        normalization_modes:
+          state.left_arm_joint_position: q99
+          state.right_arm_joint_position: q99
+          state.left_effector_position: q99
+          state.right_effector_position: q99
+          state.head_position: q99
+          state.waist_position: q99
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - action.left_arm_joint_position
+        - action.right_arm_joint_position
+        - action.left_effector_position
+        - action.right_effector_position
+        - action.head_position
+        - action.waist_position
+        - action.robot_velocity
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - action.left_arm_joint_position
+        - action.right_arm_joint_position
+        - action.left_effector_position
+        - action.right_effector_position
+        - action.head_position
+        - action.waist_position
+        - action.robot_velocity
+        normalization_modes:
+          action.left_arm_joint_position: q99
+          action.right_arm_joint_position: q99
+          action.left_effector_position: q99
+          action.right_effector_position: q99
+          action.head_position: q99
+          action.waist_position: q99
+          action.robot_velocity: q99
+      - _target_: groot.vla.data.transform.ConcatTransform
+        video_concat_order:
+        - video.top_head
+        - video.hand_left
+        - video.hand_right
+        state_concat_order:
+        - state.left_arm_joint_position
+        - state.right_arm_joint_position
+        - state.left_effector_position
+        - state.right_effector_position
+        - state.head_position
+        - state.waist_position
+        action_concat_order:
+        - action.left_arm_joint_position
+        - action.right_arm_joint_position
+        - action.left_effector_position
+        - action.right_effector_position
+        - action.head_position
+        - action.waist_position
+        - action.robot_velocity
+      - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+        default_instruction: Perform the default behavior.
+        language_dropout_prob: 0.0
+        always_use_default_instruction: false
+        max_state_dim: 64
+        max_action_dim: 32
+        max_length: 512
+        state_horizon: 1
+        action_horizon: 16
+        embodiment_tag_mapping:
+          real_gr1_arms_only: 0
+          real_gr1_arms_only_annotated: 1
+          real_gr1_arms_waist: 2
+          real_gr1_arms_waist_annotated: 3
+          dexmg_gr1_arms_only_inspire: 4
+          dexmg_gr1_arms_only_fourier: 5
+          dexmg_gr1_arms_waist_fourier: 6
+          robocasa_single_arm: 7
+          onex_eve_gripper: 8
+          robocasa_gr1_arms_only_inspire_hands: 9
+          robocasa_gr1_arms_only_fourier_hands: 10
+          robocasa_gr1_fixed_lower_body_inspire_hands: 11
+          robocasa_gr1_fixed_lower_body_fourier_hands: 12
+          robocasa_panda_omron: 13
+          robocasa_bimanual_panda_parallel_gripper: 15
+          robocasa_bimanual_panda_inspire_hand: 16
+          oxe_droid: 17
+          oxe_fractal: 18
+          oxe_language_table: 19
+          oxe_bridge: 20
+          real_panda_single_arm: 21
+          hot3d_hands_only: 23
+          gr1_unified: 24
+          robocasa_gr1_arms_waist_fourier_hands: 25
+          agibot: 26
+          lapa: 27
+          oxe_mutex: 28
+          oxe_roboset: 29
+          oxe_plex: 30
+          dream: 31
+          yam: 32
+          libero_sim: 21
+          xdof: 22
+          gr1_unified_segmentation: 14
+          language_table_sim: 7
+          gr1_isaac: 0
+          sim_behavior_r1_pro: 31
+          mecka_hands: 27
+          real_r1_pro_sharpa: 28
+        tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+    yam:
+      _target_: groot.vla.data.transform.ComposedModalityTransform
+      transforms:
+      - _target_: groot.vla.data.transform.VideoToTensor
+        apply_to:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+      - _target_: groot.vla.data.transform.VideoCrop
+        apply_to:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+        scale: 0.95
+        mode: random
+      - _target_: groot.vla.data.transform.VideoResize
+        apply_to:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+        height: 256
+        width: 256
+        interpolation: linear
+      - _target_: groot.vla.data.transform.VideoColorJitter
+        apply_to:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+        brightness: 0.3
+        contrast: 0.4
+        saturation: 0.5
+        hue: 0.08
+      - _target_: groot.vla.data.transform.VideoToNumpy
+        apply_to:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - state.left_joint_pos
+        - state.left_gripper_pos
+        - state.right_joint_pos
+        - state.right_gripper_pos
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - state.left_joint_pos
+        - state.left_gripper_pos
+        - state.right_joint_pos
+        - state.right_gripper_pos
+        normalization_modes:
+          state.left_joint_pos: q99
+          state.left_gripper_pos: q99
+          state.right_joint_pos: q99
+          state.right_gripper_pos: q99
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - action.left_joint_pos
+        - action.left_gripper_pos
+        - action.right_joint_pos
+        - action.right_gripper_pos
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - action.left_joint_pos
+        - action.left_gripper_pos
+        - action.right_joint_pos
+        - action.right_gripper_pos
+        normalization_modes:
+          action.left_joint_pos: q99
+          action.left_gripper_pos: q99
+          action.right_joint_pos: q99
+          action.right_gripper_pos: q99
+      - _target_: groot.vla.data.transform.ConcatTransform
+        video_concat_order:
+        - video.top_camera-images-rgb
+        - video.left_camera-images-rgb
+        - video.right_camera-images-rgb
+        state_concat_order:
+        - state.left_joint_pos
+        - state.left_gripper_pos
+        - state.right_joint_pos
+        - state.right_gripper_pos
+        action_concat_order:
+        - action.left_joint_pos
+        - action.left_gripper_pos
+        - action.right_joint_pos
+        - action.right_gripper_pos
+      - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+        default_instruction: Perform the default behavior.
+        language_dropout_prob: 0.0
+        always_use_default_instruction: false
+        max_state_dim: 64
+        max_action_dim: 32
+        max_length: 512
+        state_horizon: 1
+        action_horizon: 16
+        embodiment_tag_mapping:
+          real_gr1_arms_only: 0
+          real_gr1_arms_only_annotated: 1
+          real_gr1_arms_waist: 2
+          real_gr1_arms_waist_annotated: 3
+          dexmg_gr1_arms_only_inspire: 4
+          dexmg_gr1_arms_only_fourier: 5
+          dexmg_gr1_arms_waist_fourier: 6
+          robocasa_single_arm: 7
+          onex_eve_gripper: 8
+          robocasa_gr1_arms_only_inspire_hands: 9
+          robocasa_gr1_arms_only_fourier_hands: 10
+          robocasa_gr1_fixed_lower_body_inspire_hands: 11
+          robocasa_gr1_fixed_lower_body_fourier_hands: 12
+          robocasa_panda_omron: 13
+          robocasa_bimanual_panda_parallel_gripper: 15
+          robocasa_bimanual_panda_inspire_hand: 16
+          oxe_droid: 17
+          oxe_fractal: 18
+          oxe_language_table: 19
+          oxe_bridge: 20
+          real_panda_single_arm: 21
+          hot3d_hands_only: 23
+          gr1_unified: 24
+          robocasa_gr1_arms_waist_fourier_hands: 25
+          agibot: 26
+          lapa: 27
+          oxe_mutex: 28
+          oxe_roboset: 29
+          oxe_plex: 30
+          dream: 31
+          yam: 32
+          libero_sim: 21
+          xdof: 22
+          gr1_unified_segmentation: 14
+          language_table_sim: 7
+          gr1_isaac: 0
+          sim_behavior_r1_pro: 31
+          mecka_hands: 27
+          real_r1_pro_sharpa: 28
+        tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+    libero_sim:
+      _target_: groot.vla.data.transform.ComposedModalityTransform
+      transforms:
+      - _target_: groot.vla.data.transform.VideoToTensor
+        apply_to:
+        - video.image
+        - video.wrist_image
+      - _target_: groot.vla.data.transform.VideoCrop
+        apply_to:
+        - video.image
+        - video.wrist_image
+        scale: 0.95
+        mode: random
+      - _target_: groot.vla.data.transform.VideoResize
+        apply_to:
+        - video.image
+        - video.wrist_image
+        height: 256
+        width: 256
+        interpolation: linear
+      - _target_: groot.vla.data.transform.VideoColorJitter
+        apply_to:
+        - video.image
+        - video.wrist_image
+        brightness: 0.3
+        contrast: 0.4
+        saturation: 0.5
+        hue: 0.08
+      - _target_: groot.vla.data.transform.VideoToNumpy
+        apply_to:
+        - video.image
+        - video.wrist_image
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - state.state
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - state.state
+        normalization_modes:
+          state.state: q99
+      - _target_: groot.vla.data.transform.StateActionToTensor
+        apply_to:
+        - action.actions
+      - _target_: groot.vla.data.transform.StateActionTransform
+        apply_to:
+        - action.actions
+        normalization_modes:
+          action.actions: q99
+      - _target_: groot.vla.data.transform.ConcatTransform
+        video_concat_order:
+        - video.image
+        - video.wrist_image
+        state_concat_order:
+        - state.state
+        action_concat_order:
+        - action.actions
+      - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+        default_instruction: Perform the default behavior.
+        language_dropout_prob: 0.0
+        always_use_default_instruction: false
+        max_state_dim: 64
+        max_action_dim: 32
+        max_length: 512
+        state_horizon: 1
+        action_horizon: 16
+        embodiment_tag_mapping:
+          real_gr1_arms_only: 0
+          real_gr1_arms_only_annotated: 1
+          real_gr1_arms_waist: 2
+          real_gr1_arms_waist_annotated: 3
+          dexmg_gr1_arms_only_inspire: 4
+          dexmg_gr1_arms_only_fourier: 5
+          dexmg_gr1_arms_waist_fourier: 6
+          robocasa_single_arm: 7
+          onex_eve_gripper: 8
+          robocasa_gr1_arms_only_inspire_hands: 9
+          robocasa_gr1_arms_only_fourier_hands: 10
+          robocasa_gr1_fixed_lower_body_inspire_hands: 11
+          robocasa_gr1_fixed_lower_body_fourier_hands: 12
+          robocasa_panda_omron: 13
+          robocasa_bimanual_panda_parallel_gripper: 15
+          robocasa_bimanual_panda_inspire_hand: 16
+          oxe_droid: 17
+          oxe_fractal: 18
+          oxe_language_table: 19
+          oxe_bridge: 20
+          real_panda_single_arm: 21
+          hot3d_hands_only: 23
+          gr1_unified: 24
+          robocasa_gr1_arms_waist_fourier_hands: 25
+          agibot: 26
+          lapa: 27
+          oxe_mutex: 28
+          oxe_roboset: 29
+          oxe_plex: 30
+          dream: 31
+          yam: 32
+          libero_sim: 21
+          xdof: 22
+          gr1_unified_segmentation: 14
+          language_table_sim: 7
+          gr1_isaac: 0
+          sim_behavior_r1_pro: 31
+          mecka_hands: 27
+          real_r1_pro_sharpa: 28
+        tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+  metadata_versions:
+    oxe_droid: '0221'
+    agibot: '0221'
+    yam: '0221'
+    libero_sim: '0221'
+  fps:
+    yam: 30
+    libero_sim: 10
+  dataset_kwargs:
+    video_backend: decord
+    use_global_metadata: false
+    max_chunk_size: 4
+    relative_action: false
+    relative_action_per_horizon: false
+  mixture_kwargs:
+    training: true
+    balance_dataset_weights: false
+    seed: 42
+    shard_sampling_rate: 1.0
+trainer:
+  _target_: groot.vla.experiment.VLATrainer
+  _partial_: true
+  _recursive_: false
+  callbacks: null
+  model: ???
+  train_dataset: ???
+  compute_dtype: ???
+  benchmark_time: false
+  enable_profiling: false
+  profiling_steps: 5
+  enable_prof_callback: false
+  profile_start_step: 50
+  profile_warmup_steps: 1
+  profile_active_steps: 3
+  profile_record_shapes: false
+  profile_with_stack: false
+  profile_memory: false
+wandb_project: dreamzero
+output_dir: /mnt/project_rlinf/yuanhuining/repos/dreamzero/runs/train/2026-05-07T16-36-40Z/checkpoints
+load_from_yaml: null
+gear_credentials: null
+upload_checkpoints: false
+upload_every: 1000
+upload_last_n_checkpoints: 5
+remove_unused_columns: false
+bf16: true
+tf32: true
+global_batch_size: null
+raise_error_if_global_batch_size_not_set: false
+per_device_train_batch_size: 32
+per_device_eval_batch_size: 64
+gradient_accumulation_steps: 1
+dataloader_num_workers: 1
+dataloader_pin_memory: false
+dataloader_persistent_workers: true
+optim: adamw_torch
+learning_rate: 0.0001
+adam_beta1: 0.95
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+weight_decay: 1.0e-05
+lr_scheduler_type: cosine
+warmup_ratio: 0.05
+logging_steps: 10.0
+num_train_epochs: 1000
+max_steps: 50000
+save_strategy: steps
+save_steps: 2000
+eval_strategy: 'no'
+save_total_limit: 10
+report_to: tensorboard
+seed: 42
+do_eval: false
+gradient_checkpointing: false
+ddp_find_unused_parameters: false
+ddp_bucket_cap_mb: 100
+ray_num_workers: ???
+eval_bf16: true
+torch_compile_mode: null
+pretrained_model_path: null
+only_tune_projectors: false
+save_deepspeed_checkpoint: false
+save_llm: false
+save_lora_only: false
+save_value_model: false
+save_q_model: false
+download_cache: false
+training_args:
+  _target_: transformers.TrainingArguments
+  output_dir: /mnt/project_rlinf/yuanhuining/repos/dreamzero/runs/train/2026-05-07T16-36-40Z/checkpoints
+  run_name: checkpoints
+  remove_unused_columns: false
+  deepspeed: groot/vla/configs/deepspeed/zero2_offload.json
+  gradient_checkpointing: false
+  bf16: true
+  tf32: true
+  per_device_train_batch_size: 32
+  per_device_eval_batch_size: 64
+  gradient_accumulation_steps: 1
+  dataloader_num_workers: 1
+  dataloader_pin_memory: false
+  dataloader_persistent_workers: true
+  optim: adamw_torch
+  adam_beta1: 0.95
+  adam_beta2: 0.999
+  adam_epsilon: 1.0e-08
+  learning_rate: 1.0e-05
+  weight_decay: 1.0e-05
+  warmup_ratio: 0.05
+  lr_scheduler_type: cosine
+  logging_steps: 10.0
+  logging_dir: /mnt/project_rlinf/yuanhuining/repos/dreamzero/runs/train/2026-05-07T16-36-40Z/tensorboard
+  num_train_epochs: 1000
+  max_steps: 50000
+  save_strategy: steps
+  save_steps: 2000
+  save_total_limit: 10
+  report_to: tensorboard
+  seed: 42
+  do_eval: false
+  ddp_find_unused_parameters: false
+  ddp_bucket_cap_mb: 100
+  torch_compile_mode: null
+profile_dir: null
+backbone_hidden_size: 0
+backbone_cfg:
+  _target_: groot.vla.model.dreamzero.backbone.identity.IdentityBackbone
+action_head_cfg:
+  config:
+    backbone_features_projector_cfg: null
+    _target_: groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHeadConfig
+    _recursive_: false
+    tiled: false
+    tile_size_height: 34
+    tile_size_width: 34
+    tile_stride_height: 18
+    tile_stride_width: 16
+    lora_rank: 4
+    lora_alpha: 4
+    num_frames: 33
+    num_frame_per_block: 2
+    lora_target_modules: q,k,v,o,ffn.0,ffn.2
+    init_lora_weights: kaiming
+    train_architecture: full
+    use_gradient_checkpointing: true
+    add_pos_embed: true
+    model_dtype: float32
+    max_state_dim: 64
+    max_action_dim: 32
+    action_loss_embodiment_ids:
+    - 26
+    - 17
+    - 32
+    hidden_size: 64
+    input_embedding_dim: 1536
+    backbone_embedding_dim: 0
+    repa_layer: 8
+    repa_coeff: 1.0
+    load_pretrained_det_decode_layer_path: null
+    freeze_decode_layer: false
+    expand_batch: null
+    use_vlln: true
+    vl_self_attention_cfg:
+      _target_: groot.vla.model.n1_5.modules.cross_attention_dit.SelfAttentionTransformer
+      positional_embeddings: null
+      num_layers: 4
+      num_attention_heads: 24
+      attention_head_dim: 64
+      dropout: 0.2
+      final_dropout: true
+    diffusion_model_cfg:
+      _target_: groot.vla.model.dreamzero.modules.wan_video_dit_action_casual_chunk.CausalWanModel
+      _convert_: object
+      diffusion_model_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B
+      model_type: ti2v
+      frame_seqlen: 50
+      dim: 3072
+      in_dim: 48
+      ffn_dim: 14336
+      out_dim: 48
+      freq_dim: 256
+      eps: 1.0e-06
+      num_heads: 24
+      num_layers: 30
+      max_chunk_size: 4
+      num_frame_per_block: 2
+      num_action_per_block: 16
+      num_state_per_block: 1
+      concat_first_frame_latent: false
+    text_encoder_cfg:
+      _target_: groot.vla.model.dreamzero.modules.wan_video_text_encoder.WanTextEncoder
+      _convert_: object
+      text_encoder_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/models_t5_umt5-xxl-enc-bf16.pth
+    image_encoder_cfg:
+      _target_: groot.vla.model.dreamzero.modules.wan_video_image_encoder.WanImageEncoder
+      _convert_: object
+      image_encoder_pretrained_path: /mnt/project_rlinf_hs/yuanhuining/models/Wan2.1-I2V-14B-480P/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth
+    vae_cfg:
+      _target_: groot.vla.model.dreamzero.modules.wan_video_vae.WanVideoVAE38
+      _convert_: object
+      vae_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/Wan2.2_VAE.pth
+      z_dim: 48
+      dim: 160
+    action_dim: 32
+    action_horizon: 16
+    num_inference_timesteps: 4
+    noise_beta_alpha: 1.5
+    noise_beta_beta: 1.0
+    noise_s: 0.999
+    num_timestep_buckets: 1000
+    decouple_video_action_noise: false
+    video_noise_beta_alpha: 3.0
+    video_noise_beta_beta: 1.0
+    lambda_tgm_video: 0.0
+    lambda_tgm_action: 0.0
+    tgm_sigma: 0.5
+    tgm_couple_prob: 0.5
+    tune_projector: true
+    tune_diffusion_model: true
+    target_video_height: 160
+    target_video_width: 320
+  _target_: groot.vla.model.dreamzero.action_head.wan_flow_matching_action_tf.WANPolicyHead
+  _convert_: object
+add_pos_embed: true
+hidden_size: 64
+attn_dropout: 0.2
+repa_layer: 8
+repa_coeff: 1.0
+load_pretrained_det_decode_layer_path: null
+expand_batch: null
+dit_version: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B
+text_encoder_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/models_t5_umt5-xxl-enc-bf16.pth
+image_encoder_pretrained_path: /mnt/project_rlinf_hs/yuanhuining/models/Wan2.1-I2V-14B-480P/models_clip_open-clip-xlm-roberta-large-vit-huge-14.pth
+vae_pretrained_path: /mnt/project_rlinf/hf_models/Wan2.2-TI2V-5B/Wan2.2_VAE.pth
+train_architecture: full
+num_frame_per_block: 2
+num_action_per_block: 16
+num_state_per_block: 1
+frame_seqlen: 50
+embodiment_tag_to_projector_index:
+  real_gr1_arms_only: 0
+  real_gr1_arms_only_annotated: 1
+  real_gr1_arms_waist: 2
+  real_gr1_arms_waist_annotated: 3
+  dexmg_gr1_arms_only_inspire: 4
+  dexmg_gr1_arms_only_fourier: 5
+  dexmg_gr1_arms_waist_fourier: 6
+  robocasa_single_arm: 7
+  onex_eve_gripper: 8
+  robocasa_gr1_arms_only_inspire_hands: 9
+  robocasa_gr1_arms_only_fourier_hands: 10
+  robocasa_gr1_fixed_lower_body_inspire_hands: 11
+  robocasa_gr1_fixed_lower_body_fourier_hands: 12
+  robocasa_panda_omron: 13
+  robocasa_bimanual_panda_parallel_gripper: 15
+  robocasa_bimanual_panda_inspire_hand: 16
+  oxe_droid: 17
+  oxe_fractal: 18
+  oxe_language_table: 19
+  oxe_bridge: 20
+  real_panda_single_arm: 21
+  hot3d_hands_only: 23
+  gr1_unified: 24
+  robocasa_gr1_arms_waist_fourier_hands: 25
+  agibot: 26
+  lapa: 27
+  oxe_mutex: 28
+  oxe_roboset: 29
+  oxe_plex: 30
+  dream: 31
+  yam: 32
+  libero_sim: 21
+  xdof: 22
+  gr1_unified_segmentation: 14
+  language_table_sim: 7
+  gr1_isaac: 0
+  sim_behavior_r1_pro: 31
+  mecka_hands: 27
+  real_r1_pro_sharpa: 28
+max_length: 512
+num_views: 2
+tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+data_collator:
+  _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DefaultDataCollator
+  tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+  max_length: 512
+  num_views: 2
+  embodiment_tag_mapping:
+    real_gr1_arms_only: 0
+    real_gr1_arms_only_annotated: 1
+    real_gr1_arms_waist: 2
+    real_gr1_arms_waist_annotated: 3
+    dexmg_gr1_arms_only_inspire: 4
+    dexmg_gr1_arms_only_fourier: 5
+    dexmg_gr1_arms_waist_fourier: 6
+    robocasa_single_arm: 7
+    onex_eve_gripper: 8
+    robocasa_gr1_arms_only_inspire_hands: 9
+    robocasa_gr1_arms_only_fourier_hands: 10
+    robocasa_gr1_fixed_lower_body_inspire_hands: 11
+    robocasa_gr1_fixed_lower_body_fourier_hands: 12
+    robocasa_panda_omron: 13
+    robocasa_bimanual_panda_parallel_gripper: 15
+    robocasa_bimanual_panda_inspire_hand: 16
+    oxe_droid: 17
+    oxe_fractal: 18
+    oxe_language_table: 19
+    oxe_bridge: 20
+    real_panda_single_arm: 21
+    hot3d_hands_only: 23
+    gr1_unified: 24
+    robocasa_gr1_arms_waist_fourier_hands: 25
+    agibot: 26
+    lapa: 27
+    oxe_mutex: 28
+    oxe_roboset: 29
+    oxe_plex: 30
+    dream: 31
+    yam: 32
+    libero_sim: 21
+    xdof: 22
+    gr1_unified_segmentation: 14
+    language_table_sim: 7
+    gr1_isaac: 0
+    sim_behavior_r1_pro: 31
+    mecka_hands: 27
+    real_r1_pro_sharpa: 28
+num_visual_tokens_per_frame: 16
+max_state_dim: 64
+max_action_dim: 32
+language_dropout_prob: 0.0
+model_specific_transform:
+  _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+  default_instruction: Perform the default behavior.
+  language_dropout_prob: 0.0
+  always_use_default_instruction: false
+  max_state_dim: 64
+  max_action_dim: 32
+  max_length: 512
+  state_horizon: 1
+  action_horizon: 16
+  embodiment_tag_mapping:
+    real_gr1_arms_only: 0
+    real_gr1_arms_only_annotated: 1
+    real_gr1_arms_waist: 2
+    real_gr1_arms_waist_annotated: 3
+    dexmg_gr1_arms_only_inspire: 4
+    dexmg_gr1_arms_only_fourier: 5
+    dexmg_gr1_arms_waist_fourier: 6
+    robocasa_single_arm: 7
+    onex_eve_gripper: 8
+    robocasa_gr1_arms_only_inspire_hands: 9
+    robocasa_gr1_arms_only_fourier_hands: 10
+    robocasa_gr1_fixed_lower_body_inspire_hands: 11
+    robocasa_gr1_fixed_lower_body_fourier_hands: 12
+    robocasa_panda_omron: 13
+    robocasa_bimanual_panda_parallel_gripper: 15
+    robocasa_bimanual_panda_inspire_hand: 16
+    oxe_droid: 17
+    oxe_fractal: 18
+    oxe_language_table: 19
+    oxe_bridge: 20
+    real_panda_single_arm: 21
+    hot3d_hands_only: 23
+    gr1_unified: 24
+    robocasa_gr1_arms_waist_fourier_hands: 25
+    agibot: 26
+    lapa: 27
+    oxe_mutex: 28
+    oxe_roboset: 29
+    oxe_plex: 30
+    dream: 31
+    yam: 32
+    libero_sim: 21
+    xdof: 22
+    gr1_unified_segmentation: 14
+    language_table_sim: 7
+    gr1_isaac: 0
+    sim_behavior_r1_pro: 31
+    mecka_hands: 27
+    real_r1_pro_sharpa: 28
+  tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+use_global_metadata: false
+num_frames: 33
+action_horizon: 16
+state_horizon: 1
+image_resolution_width: 256
+image_resolution_height: 256
+image_resolution_width_single_frame: 256
+image_resolution_height_single_frame: 256
+totensor_cfg:
+  _target_: groot.vla.data.transform.VideoToTensor
+  apply_to: ???
+crop_cfg:
+  _target_: groot.vla.data.transform.VideoCrop
+  apply_to: ???
+  scale: 0.95
+  mode: random
+resize_cfg:
+  _target_: groot.vla.data.transform.VideoResize
+  apply_to: ???
+  height: 256
+  width: 256
+  interpolation: linear
+resize_cfg_single_frame:
+  _target_: groot.vla.data.transform.VideoResize
+  apply_to: ???
+  height: 256
+  width: 256
+  interpolation: linear
+color_jitter_cfg:
+  _target_: groot.vla.data.transform.VideoColorJitter
+  apply_to: ???
+  brightness: 0.3
+  contrast: 0.4
+  saturation: 0.5
+  hue: 0.08
+random_grayscale_cfg:
+  _target_: groot.vla.data.transform.VideoRandomGrayscale
+  apply_to: ???
+  p: 0.1
+random_posterize_cfg:
+  _target_: groot.vla.data.transform.VideoRandomPosterize
+  apply_to: ???
+  bits: 4
+  p: 0.1
+normalize_cfg:
+  _target_: groot.vla.data.transform.VideoNormalize
+  apply_to: ???
+  mean:
+  - 0.5
+  - 0.5
+  - 0.5
+  std:
+  - 0.5
+  - 0.5
+  - 0.5
+to_numpy_cfg:
+  _target_: groot.vla.data.transform.VideoToNumpy
+  apply_to: ???
+modality_config_oxe_droid:
+  video:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    - 24
+    eval_delta_indices:
+    - 0
+    modality_keys:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+  state:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - state.joint_position
+    - state.gripper_position
+  action:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    modality_keys:
+    - action.joint_position
+    - action.gripper_position
+  language:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - annotation.language.language_instruction
+    - annotation.language.language_instruction_2
+    - annotation.language.language_instruction_3
+  lapa_action:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - lapa_action
+transform_oxe_droid:
+  _target_: groot.vla.data.transform.ComposedModalityTransform
+  transforms:
+  - _target_: groot.vla.data.transform.VideoToTensor
+    apply_to:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+  - _target_: groot.vla.data.transform.VideoCrop
+    apply_to:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+    scale: 0.95
+    mode: random
+  - _target_: groot.vla.data.transform.VideoResize
+    apply_to:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+    height: 256
+    width: 256
+    interpolation: linear
+  - _target_: groot.vla.data.transform.VideoColorJitter
+    apply_to:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  - _target_: groot.vla.data.transform.VideoToNumpy
+    apply_to:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - state.joint_position
+    - state.gripper_position
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - state.joint_position
+    - state.gripper_position
+    normalization_modes:
+      state.joint_position: q99
+      state.gripper_position: q99
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - action.joint_position
+    - action.gripper_position
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - action.joint_position
+    - action.gripper_position
+    normalization_modes:
+      action.joint_position: q99
+      action.gripper_position: q99
+  - _target_: groot.vla.data.transform.ConcatTransform
+    video_concat_order:
+    - video.exterior_image_1_left
+    - video.exterior_image_2_left
+    - video.wrist_image_left
+    state_concat_order:
+    - state.joint_position
+    - state.gripper_position
+    action_concat_order:
+    - action.joint_position
+    - action.gripper_position
+  - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+    default_instruction: Perform the default behavior.
+    language_dropout_prob: 0.0
+    always_use_default_instruction: false
+    max_state_dim: 64
+    max_action_dim: 32
+    max_length: 512
+    state_horizon: 1
+    action_horizon: 16
+    embodiment_tag_mapping:
+      real_gr1_arms_only: 0
+      real_gr1_arms_only_annotated: 1
+      real_gr1_arms_waist: 2
+      real_gr1_arms_waist_annotated: 3
+      dexmg_gr1_arms_only_inspire: 4
+      dexmg_gr1_arms_only_fourier: 5
+      dexmg_gr1_arms_waist_fourier: 6
+      robocasa_single_arm: 7
+      onex_eve_gripper: 8
+      robocasa_gr1_arms_only_inspire_hands: 9
+      robocasa_gr1_arms_only_fourier_hands: 10
+      robocasa_gr1_fixed_lower_body_inspire_hands: 11
+      robocasa_gr1_fixed_lower_body_fourier_hands: 12
+      robocasa_panda_omron: 13
+      robocasa_bimanual_panda_parallel_gripper: 15
+      robocasa_bimanual_panda_inspire_hand: 16
+      oxe_droid: 17
+      oxe_fractal: 18
+      oxe_language_table: 19
+      oxe_bridge: 20
+      real_panda_single_arm: 21
+      hot3d_hands_only: 23
+      gr1_unified: 24
+      robocasa_gr1_arms_waist_fourier_hands: 25
+      agibot: 26
+      lapa: 27
+      oxe_mutex: 28
+      oxe_roboset: 29
+      oxe_plex: 30
+      dream: 31
+      yam: 32
+      libero_sim: 21
+      xdof: 22
+      gr1_unified_segmentation: 14
+      language_table_sim: 7
+      gr1_isaac: 0
+      sim_behavior_r1_pro: 31
+      mecka_hands: 27
+      real_r1_pro_sharpa: 28
+    tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+modality_config_agibot:
+  video:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    - 24
+    eval_delta_indices:
+    - -3
+    - -2
+    - -1
+    - 0
+    modality_keys:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+  state:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - state.left_arm_joint_position
+    - state.right_arm_joint_position
+    - state.left_effector_position
+    - state.right_effector_position
+    - state.head_position
+    - state.waist_position
+  action:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    modality_keys:
+    - action.left_arm_joint_position
+    - action.right_arm_joint_position
+    - action.left_effector_position
+    - action.right_effector_position
+    - action.head_position
+    - action.waist_position
+    - action.robot_velocity
+  language:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - annotation.language.action_text
+transform_agibot:
+  _target_: groot.vla.data.transform.ComposedModalityTransform
+  transforms:
+  - _target_: groot.vla.data.transform.VideoToTensor
+    apply_to:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+  - _target_: groot.vla.data.transform.VideoCrop
+    apply_to:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+    scale: 0.95
+    mode: random
+  - _target_: groot.vla.data.transform.VideoResize
+    apply_to:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+    height: 256
+    width: 256
+    interpolation: linear
+  - _target_: groot.vla.data.transform.VideoColorJitter
+    apply_to:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  - _target_: groot.vla.data.transform.VideoToNumpy
+    apply_to:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - state.left_arm_joint_position
+    - state.right_arm_joint_position
+    - state.left_effector_position
+    - state.right_effector_position
+    - state.head_position
+    - state.waist_position
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - state.left_arm_joint_position
+    - state.right_arm_joint_position
+    - state.left_effector_position
+    - state.right_effector_position
+    - state.head_position
+    - state.waist_position
+    normalization_modes:
+      state.left_arm_joint_position: q99
+      state.right_arm_joint_position: q99
+      state.left_effector_position: q99
+      state.right_effector_position: q99
+      state.head_position: q99
+      state.waist_position: q99
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - action.left_arm_joint_position
+    - action.right_arm_joint_position
+    - action.left_effector_position
+    - action.right_effector_position
+    - action.head_position
+    - action.waist_position
+    - action.robot_velocity
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - action.left_arm_joint_position
+    - action.right_arm_joint_position
+    - action.left_effector_position
+    - action.right_effector_position
+    - action.head_position
+    - action.waist_position
+    - action.robot_velocity
+    normalization_modes:
+      action.left_arm_joint_position: q99
+      action.right_arm_joint_position: q99
+      action.left_effector_position: q99
+      action.right_effector_position: q99
+      action.head_position: q99
+      action.waist_position: q99
+      action.robot_velocity: q99
+  - _target_: groot.vla.data.transform.ConcatTransform
+    video_concat_order:
+    - video.top_head
+    - video.hand_left
+    - video.hand_right
+    state_concat_order:
+    - state.left_arm_joint_position
+    - state.right_arm_joint_position
+    - state.left_effector_position
+    - state.right_effector_position
+    - state.head_position
+    - state.waist_position
+    action_concat_order:
+    - action.left_arm_joint_position
+    - action.right_arm_joint_position
+    - action.left_effector_position
+    - action.right_effector_position
+    - action.head_position
+    - action.waist_position
+    - action.robot_velocity
+  - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+    default_instruction: Perform the default behavior.
+    language_dropout_prob: 0.0
+    always_use_default_instruction: false
+    max_state_dim: 64
+    max_action_dim: 32
+    max_length: 512
+    state_horizon: 1
+    action_horizon: 16
+    embodiment_tag_mapping:
+      real_gr1_arms_only: 0
+      real_gr1_arms_only_annotated: 1
+      real_gr1_arms_waist: 2
+      real_gr1_arms_waist_annotated: 3
+      dexmg_gr1_arms_only_inspire: 4
+      dexmg_gr1_arms_only_fourier: 5
+      dexmg_gr1_arms_waist_fourier: 6
+      robocasa_single_arm: 7
+      onex_eve_gripper: 8
+      robocasa_gr1_arms_only_inspire_hands: 9
+      robocasa_gr1_arms_only_fourier_hands: 10
+      robocasa_gr1_fixed_lower_body_inspire_hands: 11
+      robocasa_gr1_fixed_lower_body_fourier_hands: 12
+      robocasa_panda_omron: 13
+      robocasa_bimanual_panda_parallel_gripper: 15
+      robocasa_bimanual_panda_inspire_hand: 16
+      oxe_droid: 17
+      oxe_fractal: 18
+      oxe_language_table: 19
+      oxe_bridge: 20
+      real_panda_single_arm: 21
+      hot3d_hands_only: 23
+      gr1_unified: 24
+      robocasa_gr1_arms_waist_fourier_hands: 25
+      agibot: 26
+      lapa: 27
+      oxe_mutex: 28
+      oxe_roboset: 29
+      oxe_plex: 30
+      dream: 31
+      yam: 32
+      libero_sim: 21
+      xdof: 22
+      gr1_unified_segmentation: 14
+      language_table_sim: 7
+      gr1_isaac: 0
+      sim_behavior_r1_pro: 31
+      mecka_hands: 27
+      real_r1_pro_sharpa: 28
+    tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+modality_config_yam:
+  video:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    - 24
+    eval_delta_indices:
+    - 0
+    modality_keys:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+  state:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - state.left_joint_pos
+    - state.left_gripper_pos
+    - state.right_joint_pos
+    - state.right_gripper_pos
+  action:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    modality_keys:
+    - action.left_joint_pos
+    - action.left_gripper_pos
+    - action.right_joint_pos
+    - action.right_gripper_pos
+  language:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - annotation.task
+transform_yam:
+  _target_: groot.vla.data.transform.ComposedModalityTransform
+  transforms:
+  - _target_: groot.vla.data.transform.VideoToTensor
+    apply_to:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+  - _target_: groot.vla.data.transform.VideoCrop
+    apply_to:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+    scale: 0.95
+    mode: random
+  - _target_: groot.vla.data.transform.VideoResize
+    apply_to:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+    height: 256
+    width: 256
+    interpolation: linear
+  - _target_: groot.vla.data.transform.VideoColorJitter
+    apply_to:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  - _target_: groot.vla.data.transform.VideoToNumpy
+    apply_to:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - state.left_joint_pos
+    - state.left_gripper_pos
+    - state.right_joint_pos
+    - state.right_gripper_pos
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - state.left_joint_pos
+    - state.left_gripper_pos
+    - state.right_joint_pos
+    - state.right_gripper_pos
+    normalization_modes:
+      state.left_joint_pos: q99
+      state.left_gripper_pos: q99
+      state.right_joint_pos: q99
+      state.right_gripper_pos: q99
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - action.left_joint_pos
+    - action.left_gripper_pos
+    - action.right_joint_pos
+    - action.right_gripper_pos
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - action.left_joint_pos
+    - action.left_gripper_pos
+    - action.right_joint_pos
+    - action.right_gripper_pos
+    normalization_modes:
+      action.left_joint_pos: q99
+      action.left_gripper_pos: q99
+      action.right_joint_pos: q99
+      action.right_gripper_pos: q99
+  - _target_: groot.vla.data.transform.ConcatTransform
+    video_concat_order:
+    - video.top_camera-images-rgb
+    - video.left_camera-images-rgb
+    - video.right_camera-images-rgb
+    state_concat_order:
+    - state.left_joint_pos
+    - state.left_gripper_pos
+    - state.right_joint_pos
+    - state.right_gripper_pos
+    action_concat_order:
+    - action.left_joint_pos
+    - action.left_gripper_pos
+    - action.right_joint_pos
+    - action.right_gripper_pos
+  - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+    default_instruction: Perform the default behavior.
+    language_dropout_prob: 0.0
+    always_use_default_instruction: false
+    max_state_dim: 64
+    max_action_dim: 32
+    max_length: 512
+    state_horizon: 1
+    action_horizon: 16
+    embodiment_tag_mapping:
+      real_gr1_arms_only: 0
+      real_gr1_arms_only_annotated: 1
+      real_gr1_arms_waist: 2
+      real_gr1_arms_waist_annotated: 3
+      dexmg_gr1_arms_only_inspire: 4
+      dexmg_gr1_arms_only_fourier: 5
+      dexmg_gr1_arms_waist_fourier: 6
+      robocasa_single_arm: 7
+      onex_eve_gripper: 8
+      robocasa_gr1_arms_only_inspire_hands: 9
+      robocasa_gr1_arms_only_fourier_hands: 10
+      robocasa_gr1_fixed_lower_body_inspire_hands: 11
+      robocasa_gr1_fixed_lower_body_fourier_hands: 12
+      robocasa_panda_omron: 13
+      robocasa_bimanual_panda_parallel_gripper: 15
+      robocasa_bimanual_panda_inspire_hand: 16
+      oxe_droid: 17
+      oxe_fractal: 18
+      oxe_language_table: 19
+      oxe_bridge: 20
+      real_panda_single_arm: 21
+      hot3d_hands_only: 23
+      gr1_unified: 24
+      robocasa_gr1_arms_waist_fourier_hands: 25
+      agibot: 26
+      lapa: 27
+      oxe_mutex: 28
+      oxe_roboset: 29
+      oxe_plex: 30
+      dream: 31
+      yam: 32
+      libero_sim: 21
+      xdof: 22
+      gr1_unified_segmentation: 14
+      language_table_sim: 7
+      gr1_isaac: 0
+      sim_behavior_r1_pro: 31
+      mecka_hands: 27
+      real_r1_pro_sharpa: 28
+    tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+modality_config_libero:
+  video:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    - 24
+    eval_delta_indices:
+    - 0
+    modality_keys:
+    - video.image
+    - video.wrist_image
+  state:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - state.state
+  action:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+    - 17
+    - 18
+    - 19
+    - 20
+    - 21
+    - 22
+    - 23
+    modality_keys:
+    - action.actions
+  language:
+    _target_: groot.vla.data.dataset.ModalityConfig
+    delta_indices:
+    - 0
+    modality_keys:
+    - annotation.task
+transform_libero:
+  _target_: groot.vla.data.transform.ComposedModalityTransform
+  transforms:
+  - _target_: groot.vla.data.transform.VideoToTensor
+    apply_to:
+    - video.image
+    - video.wrist_image
+  - _target_: groot.vla.data.transform.VideoCrop
+    apply_to:
+    - video.image
+    - video.wrist_image
+    scale: 0.95
+    mode: random
+  - _target_: groot.vla.data.transform.VideoResize
+    apply_to:
+    - video.image
+    - video.wrist_image
+    height: 256
+    width: 256
+    interpolation: linear
+  - _target_: groot.vla.data.transform.VideoColorJitter
+    apply_to:
+    - video.image
+    - video.wrist_image
+    brightness: 0.3
+    contrast: 0.4
+    saturation: 0.5
+    hue: 0.08
+  - _target_: groot.vla.data.transform.VideoToNumpy
+    apply_to:
+    - video.image
+    - video.wrist_image
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - state.state
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - state.state
+    normalization_modes:
+      state.state: q99
+  - _target_: groot.vla.data.transform.StateActionToTensor
+    apply_to:
+    - action.actions
+  - _target_: groot.vla.data.transform.StateActionTransform
+    apply_to:
+    - action.actions
+    normalization_modes:
+      action.actions: q99
+  - _target_: groot.vla.data.transform.ConcatTransform
+    video_concat_order:
+    - video.image
+    - video.wrist_image
+    state_concat_order:
+    - state.state
+    action_concat_order:
+    - action.actions
+  - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+    default_instruction: Perform the default behavior.
+    language_dropout_prob: 0.0
+    always_use_default_instruction: false
+    max_state_dim: 64
+    max_action_dim: 32
+    max_length: 512
+    state_horizon: 1
+    action_horizon: 16
+    embodiment_tag_mapping:
+      real_gr1_arms_only: 0
+      real_gr1_arms_only_annotated: 1
+      real_gr1_arms_waist: 2
+      real_gr1_arms_waist_annotated: 3
+      dexmg_gr1_arms_only_inspire: 4
+      dexmg_gr1_arms_only_fourier: 5
+      dexmg_gr1_arms_waist_fourier: 6
+      robocasa_single_arm: 7
+      onex_eve_gripper: 8
+      robocasa_gr1_arms_only_inspire_hands: 9
+      robocasa_gr1_arms_only_fourier_hands: 10
+      robocasa_gr1_fixed_lower_body_inspire_hands: 11
+      robocasa_gr1_fixed_lower_body_fourier_hands: 12
+      robocasa_panda_omron: 13
+      robocasa_bimanual_panda_parallel_gripper: 15
+      robocasa_bimanual_panda_inspire_hand: 16
+      oxe_droid: 17
+      oxe_fractal: 18
+      oxe_language_table: 19
+      oxe_bridge: 20
+      real_panda_single_arm: 21
+      hot3d_hands_only: 23
+      gr1_unified: 24
+      robocasa_gr1_arms_waist_fourier_hands: 25
+      agibot: 26
+      lapa: 27
+      oxe_mutex: 28
+      oxe_roboset: 29
+      oxe_plex: 30
+      dream: 31
+      yam: 32
+      libero_sim: 21
+      xdof: 22
+      gr1_unified_segmentation: 14
+      language_table_sim: 7
+      gr1_isaac: 0
+      sim_behavior_r1_pro: 31
+      mecka_hands: 27
+      real_r1_pro_sharpa: 28
+    tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+modality_configs:
+  oxe_droid:
+    video:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      - 24
+      eval_delta_indices:
+      - 0
+      modality_keys:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+    state:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - state.joint_position
+      - state.gripper_position
+    action:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      modality_keys:
+      - action.joint_position
+      - action.gripper_position
+    language:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - annotation.language.language_instruction
+      - annotation.language.language_instruction_2
+      - annotation.language.language_instruction_3
+    lapa_action:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - lapa_action
+  agibot:
+    video:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      - 24
+      eval_delta_indices:
+      - -3
+      - -2
+      - -1
+      - 0
+      modality_keys:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+    state:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - state.left_arm_joint_position
+      - state.right_arm_joint_position
+      - state.left_effector_position
+      - state.right_effector_position
+      - state.head_position
+      - state.waist_position
+    action:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      modality_keys:
+      - action.left_arm_joint_position
+      - action.right_arm_joint_position
+      - action.left_effector_position
+      - action.right_effector_position
+      - action.head_position
+      - action.waist_position
+      - action.robot_velocity
+    language:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - annotation.language.action_text
+  yam:
+    video:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      - 24
+      eval_delta_indices:
+      - 0
+      modality_keys:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+    state:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - state.left_joint_pos
+      - state.left_gripper_pos
+      - state.right_joint_pos
+      - state.right_gripper_pos
+    action:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      modality_keys:
+      - action.left_joint_pos
+      - action.left_gripper_pos
+      - action.right_joint_pos
+      - action.right_gripper_pos
+    language:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - annotation.task
+  libero_sim:
+    video:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      - 24
+      eval_delta_indices:
+      - 0
+      modality_keys:
+      - video.image
+      - video.wrist_image
+    state:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - state.state
+    action:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      - 1
+      - 2
+      - 3
+      - 4
+      - 5
+      - 6
+      - 7
+      - 8
+      - 9
+      - 10
+      - 11
+      - 12
+      - 13
+      - 14
+      - 15
+      - 16
+      - 17
+      - 18
+      - 19
+      - 20
+      - 21
+      - 22
+      - 23
+      modality_keys:
+      - action.actions
+    language:
+      _target_: groot.vla.data.dataset.ModalityConfig
+      delta_indices:
+      - 0
+      modality_keys:
+      - annotation.task
+transforms:
+  oxe_droid:
+    _target_: groot.vla.data.transform.ComposedModalityTransform
+    transforms:
+    - _target_: groot.vla.data.transform.VideoToTensor
+      apply_to:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+    - _target_: groot.vla.data.transform.VideoCrop
+      apply_to:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+      scale: 0.95
+      mode: random
+    - _target_: groot.vla.data.transform.VideoResize
+      apply_to:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+      height: 256
+      width: 256
+      interpolation: linear
+    - _target_: groot.vla.data.transform.VideoColorJitter
+      apply_to:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+      brightness: 0.3
+      contrast: 0.4
+      saturation: 0.5
+      hue: 0.08
+    - _target_: groot.vla.data.transform.VideoToNumpy
+      apply_to:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - state.joint_position
+      - state.gripper_position
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - state.joint_position
+      - state.gripper_position
+      normalization_modes:
+        state.joint_position: q99
+        state.gripper_position: q99
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - action.joint_position
+      - action.gripper_position
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - action.joint_position
+      - action.gripper_position
+      normalization_modes:
+        action.joint_position: q99
+        action.gripper_position: q99
+    - _target_: groot.vla.data.transform.ConcatTransform
+      video_concat_order:
+      - video.exterior_image_1_left
+      - video.exterior_image_2_left
+      - video.wrist_image_left
+      state_concat_order:
+      - state.joint_position
+      - state.gripper_position
+      action_concat_order:
+      - action.joint_position
+      - action.gripper_position
+    - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+      default_instruction: Perform the default behavior.
+      language_dropout_prob: 0.0
+      always_use_default_instruction: false
+      max_state_dim: 64
+      max_action_dim: 32
+      max_length: 512
+      state_horizon: 1
+      action_horizon: 16
+      embodiment_tag_mapping:
+        real_gr1_arms_only: 0
+        real_gr1_arms_only_annotated: 1
+        real_gr1_arms_waist: 2
+        real_gr1_arms_waist_annotated: 3
+        dexmg_gr1_arms_only_inspire: 4
+        dexmg_gr1_arms_only_fourier: 5
+        dexmg_gr1_arms_waist_fourier: 6
+        robocasa_single_arm: 7
+        onex_eve_gripper: 8
+        robocasa_gr1_arms_only_inspire_hands: 9
+        robocasa_gr1_arms_only_fourier_hands: 10
+        robocasa_gr1_fixed_lower_body_inspire_hands: 11
+        robocasa_gr1_fixed_lower_body_fourier_hands: 12
+        robocasa_panda_omron: 13
+        robocasa_bimanual_panda_parallel_gripper: 15
+        robocasa_bimanual_panda_inspire_hand: 16
+        oxe_droid: 17
+        oxe_fractal: 18
+        oxe_language_table: 19
+        oxe_bridge: 20
+        real_panda_single_arm: 21
+        hot3d_hands_only: 23
+        gr1_unified: 24
+        robocasa_gr1_arms_waist_fourier_hands: 25
+        agibot: 26
+        lapa: 27
+        oxe_mutex: 28
+        oxe_roboset: 29
+        oxe_plex: 30
+        dream: 31
+        yam: 32
+        libero_sim: 21
+        xdof: 22
+        gr1_unified_segmentation: 14
+        language_table_sim: 7
+        gr1_isaac: 0
+        sim_behavior_r1_pro: 31
+        mecka_hands: 27
+        real_r1_pro_sharpa: 28
+      tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+  agibot:
+    _target_: groot.vla.data.transform.ComposedModalityTransform
+    transforms:
+    - _target_: groot.vla.data.transform.VideoToTensor
+      apply_to:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+    - _target_: groot.vla.data.transform.VideoCrop
+      apply_to:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+      scale: 0.95
+      mode: random
+    - _target_: groot.vla.data.transform.VideoResize
+      apply_to:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+      height: 256
+      width: 256
+      interpolation: linear
+    - _target_: groot.vla.data.transform.VideoColorJitter
+      apply_to:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+      brightness: 0.3
+      contrast: 0.4
+      saturation: 0.5
+      hue: 0.08
+    - _target_: groot.vla.data.transform.VideoToNumpy
+      apply_to:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - state.left_arm_joint_position
+      - state.right_arm_joint_position
+      - state.left_effector_position
+      - state.right_effector_position
+      - state.head_position
+      - state.waist_position
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - state.left_arm_joint_position
+      - state.right_arm_joint_position
+      - state.left_effector_position
+      - state.right_effector_position
+      - state.head_position
+      - state.waist_position
+      normalization_modes:
+        state.left_arm_joint_position: q99
+        state.right_arm_joint_position: q99
+        state.left_effector_position: q99
+        state.right_effector_position: q99
+        state.head_position: q99
+        state.waist_position: q99
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - action.left_arm_joint_position
+      - action.right_arm_joint_position
+      - action.left_effector_position
+      - action.right_effector_position
+      - action.head_position
+      - action.waist_position
+      - action.robot_velocity
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - action.left_arm_joint_position
+      - action.right_arm_joint_position
+      - action.left_effector_position
+      - action.right_effector_position
+      - action.head_position
+      - action.waist_position
+      - action.robot_velocity
+      normalization_modes:
+        action.left_arm_joint_position: q99
+        action.right_arm_joint_position: q99
+        action.left_effector_position: q99
+        action.right_effector_position: q99
+        action.head_position: q99
+        action.waist_position: q99
+        action.robot_velocity: q99
+    - _target_: groot.vla.data.transform.ConcatTransform
+      video_concat_order:
+      - video.top_head
+      - video.hand_left
+      - video.hand_right
+      state_concat_order:
+      - state.left_arm_joint_position
+      - state.right_arm_joint_position
+      - state.left_effector_position
+      - state.right_effector_position
+      - state.head_position
+      - state.waist_position
+      action_concat_order:
+      - action.left_arm_joint_position
+      - action.right_arm_joint_position
+      - action.left_effector_position
+      - action.right_effector_position
+      - action.head_position
+      - action.waist_position
+      - action.robot_velocity
+    - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+      default_instruction: Perform the default behavior.
+      language_dropout_prob: 0.0
+      always_use_default_instruction: false
+      max_state_dim: 64
+      max_action_dim: 32
+      max_length: 512
+      state_horizon: 1
+      action_horizon: 16
+      embodiment_tag_mapping:
+        real_gr1_arms_only: 0
+        real_gr1_arms_only_annotated: 1
+        real_gr1_arms_waist: 2
+        real_gr1_arms_waist_annotated: 3
+        dexmg_gr1_arms_only_inspire: 4
+        dexmg_gr1_arms_only_fourier: 5
+        dexmg_gr1_arms_waist_fourier: 6
+        robocasa_single_arm: 7
+        onex_eve_gripper: 8
+        robocasa_gr1_arms_only_inspire_hands: 9
+        robocasa_gr1_arms_only_fourier_hands: 10
+        robocasa_gr1_fixed_lower_body_inspire_hands: 11
+        robocasa_gr1_fixed_lower_body_fourier_hands: 12
+        robocasa_panda_omron: 13
+        robocasa_bimanual_panda_parallel_gripper: 15
+        robocasa_bimanual_panda_inspire_hand: 16
+        oxe_droid: 17
+        oxe_fractal: 18
+        oxe_language_table: 19
+        oxe_bridge: 20
+        real_panda_single_arm: 21
+        hot3d_hands_only: 23
+        gr1_unified: 24
+        robocasa_gr1_arms_waist_fourier_hands: 25
+        agibot: 26
+        lapa: 27
+        oxe_mutex: 28
+        oxe_roboset: 29
+        oxe_plex: 30
+        dream: 31
+        yam: 32
+        libero_sim: 21
+        xdof: 22
+        gr1_unified_segmentation: 14
+        language_table_sim: 7
+        gr1_isaac: 0
+        sim_behavior_r1_pro: 31
+        mecka_hands: 27
+        real_r1_pro_sharpa: 28
+      tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+  yam:
+    _target_: groot.vla.data.transform.ComposedModalityTransform
+    transforms:
+    - _target_: groot.vla.data.transform.VideoToTensor
+      apply_to:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+    - _target_: groot.vla.data.transform.VideoCrop
+      apply_to:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+      scale: 0.95
+      mode: random
+    - _target_: groot.vla.data.transform.VideoResize
+      apply_to:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+      height: 256
+      width: 256
+      interpolation: linear
+    - _target_: groot.vla.data.transform.VideoColorJitter
+      apply_to:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+      brightness: 0.3
+      contrast: 0.4
+      saturation: 0.5
+      hue: 0.08
+    - _target_: groot.vla.data.transform.VideoToNumpy
+      apply_to:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - state.left_joint_pos
+      - state.left_gripper_pos
+      - state.right_joint_pos
+      - state.right_gripper_pos
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - state.left_joint_pos
+      - state.left_gripper_pos
+      - state.right_joint_pos
+      - state.right_gripper_pos
+      normalization_modes:
+        state.left_joint_pos: q99
+        state.left_gripper_pos: q99
+        state.right_joint_pos: q99
+        state.right_gripper_pos: q99
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - action.left_joint_pos
+      - action.left_gripper_pos
+      - action.right_joint_pos
+      - action.right_gripper_pos
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - action.left_joint_pos
+      - action.left_gripper_pos
+      - action.right_joint_pos
+      - action.right_gripper_pos
+      normalization_modes:
+        action.left_joint_pos: q99
+        action.left_gripper_pos: q99
+        action.right_joint_pos: q99
+        action.right_gripper_pos: q99
+    - _target_: groot.vla.data.transform.ConcatTransform
+      video_concat_order:
+      - video.top_camera-images-rgb
+      - video.left_camera-images-rgb
+      - video.right_camera-images-rgb
+      state_concat_order:
+      - state.left_joint_pos
+      - state.left_gripper_pos
+      - state.right_joint_pos
+      - state.right_gripper_pos
+      action_concat_order:
+      - action.left_joint_pos
+      - action.left_gripper_pos
+      - action.right_joint_pos
+      - action.right_gripper_pos
+    - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+      default_instruction: Perform the default behavior.
+      language_dropout_prob: 0.0
+      always_use_default_instruction: false
+      max_state_dim: 64
+      max_action_dim: 32
+      max_length: 512
+      state_horizon: 1
+      action_horizon: 16
+      embodiment_tag_mapping:
+        real_gr1_arms_only: 0
+        real_gr1_arms_only_annotated: 1
+        real_gr1_arms_waist: 2
+        real_gr1_arms_waist_annotated: 3
+        dexmg_gr1_arms_only_inspire: 4
+        dexmg_gr1_arms_only_fourier: 5
+        dexmg_gr1_arms_waist_fourier: 6
+        robocasa_single_arm: 7
+        onex_eve_gripper: 8
+        robocasa_gr1_arms_only_inspire_hands: 9
+        robocasa_gr1_arms_only_fourier_hands: 10
+        robocasa_gr1_fixed_lower_body_inspire_hands: 11
+        robocasa_gr1_fixed_lower_body_fourier_hands: 12
+        robocasa_panda_omron: 13
+        robocasa_bimanual_panda_parallel_gripper: 15
+        robocasa_bimanual_panda_inspire_hand: 16
+        oxe_droid: 17
+        oxe_fractal: 18
+        oxe_language_table: 19
+        oxe_bridge: 20
+        real_panda_single_arm: 21
+        hot3d_hands_only: 23
+        gr1_unified: 24
+        robocasa_gr1_arms_waist_fourier_hands: 25
+        agibot: 26
+        lapa: 27
+        oxe_mutex: 28
+        oxe_roboset: 29
+        oxe_plex: 30
+        dream: 31
+        yam: 32
+        libero_sim: 21
+        xdof: 22
+        gr1_unified_segmentation: 14
+        language_table_sim: 7
+        gr1_isaac: 0
+        sim_behavior_r1_pro: 31
+        mecka_hands: 27
+        real_r1_pro_sharpa: 28
+      tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+  libero_sim:
+    _target_: groot.vla.data.transform.ComposedModalityTransform
+    transforms:
+    - _target_: groot.vla.data.transform.VideoToTensor
+      apply_to:
+      - video.image
+      - video.wrist_image
+    - _target_: groot.vla.data.transform.VideoCrop
+      apply_to:
+      - video.image
+      - video.wrist_image
+      scale: 0.95
+      mode: random
+    - _target_: groot.vla.data.transform.VideoResize
+      apply_to:
+      - video.image
+      - video.wrist_image
+      height: 256
+      width: 256
+      interpolation: linear
+    - _target_: groot.vla.data.transform.VideoColorJitter
+      apply_to:
+      - video.image
+      - video.wrist_image
+      brightness: 0.3
+      contrast: 0.4
+      saturation: 0.5
+      hue: 0.08
+    - _target_: groot.vla.data.transform.VideoToNumpy
+      apply_to:
+      - video.image
+      - video.wrist_image
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - state.state
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - state.state
+      normalization_modes:
+        state.state: q99
+    - _target_: groot.vla.data.transform.StateActionToTensor
+      apply_to:
+      - action.actions
+    - _target_: groot.vla.data.transform.StateActionTransform
+      apply_to:
+      - action.actions
+      normalization_modes:
+        action.actions: q99
+    - _target_: groot.vla.data.transform.ConcatTransform
+      video_concat_order:
+      - video.image
+      - video.wrist_image
+      state_concat_order:
+      - state.state
+      action_concat_order:
+      - action.actions
+    - _target_: groot.vla.model.dreamzero.transform.dreamzero_cotrain.DreamTransform
+      default_instruction: Perform the default behavior.
+      language_dropout_prob: 0.0
+      always_use_default_instruction: false
+      max_state_dim: 64
+      max_action_dim: 32
+      max_length: 512
+      state_horizon: 1
+      action_horizon: 16
+      embodiment_tag_mapping:
+        real_gr1_arms_only: 0
+        real_gr1_arms_only_annotated: 1
+        real_gr1_arms_waist: 2
+        real_gr1_arms_waist_annotated: 3
+        dexmg_gr1_arms_only_inspire: 4
+        dexmg_gr1_arms_only_fourier: 5
+        dexmg_gr1_arms_waist_fourier: 6
+        robocasa_single_arm: 7
+        onex_eve_gripper: 8
+        robocasa_gr1_arms_only_inspire_hands: 9
+        robocasa_gr1_arms_only_fourier_hands: 10
+        robocasa_gr1_fixed_lower_body_inspire_hands: 11
+        robocasa_gr1_fixed_lower_body_fourier_hands: 12
+        robocasa_panda_omron: 13
+        robocasa_bimanual_panda_parallel_gripper: 15
+        robocasa_bimanual_panda_inspire_hand: 16
+        oxe_droid: 17
+        oxe_fractal: 18
+        oxe_language_table: 19
+        oxe_bridge: 20
+        real_panda_single_arm: 21
+        hot3d_hands_only: 23
+        gr1_unified: 24
+        robocasa_gr1_arms_waist_fourier_hands: 25
+        agibot: 26
+        lapa: 27
+        oxe_mutex: 28
+        oxe_roboset: 29
+        oxe_plex: 30
+        dream: 31
+        yam: 32
+        libero_sim: 21
+        xdof: 22
+        gr1_unified_segmentation: 14
+        language_table_sim: 7
+        gr1_isaac: 0
+        sim_behavior_r1_pro: 31
+        mecka_hands: 27
+        real_r1_pro_sharpa: 28
+      tokenizer_path: /mnt/project_rlinf_hs/yuanhuining/models/umt5-xxl
+metadata_versions:
+  oxe_droid: '0221'
+  agibot: '0221'
+  yam: '0221'
+  libero_sim: '0221'
+fps:
+  yam: 30
+  libero_sim: 10
+relative_action: false
+relative_action_per_horizon: false
+max_chunk_size: 4
+dataset_shard_sampling_rate: 1.0
+mixture_dataset_cls: groot.vla.data.dataset.lerobot_sharded.ShardedLeRobotMixtureDataset.from_mixture_spec
+single_dataset_cls: groot.vla.data.dataset.lerobot_sharded.ShardedLeRobotSubLangSingleActionChunkDatasetLIBERO
+libero_data_root: /mnt/project_rlinf_hs/yuanhuining/datasets/libero
+total_training_steps: 327680000000

experiment_cfg/metadata.json

@@ -0,0 +1,169 @@
+{
+    "libero_sim": {
+        "statistics": {
+            "state": {
+                "state": {
+                    "max": [
+                        0.21031762659549713,
+                        0.39128610491752625,
+                        1.3660105466842651,
+                        3.6714255809783936,
+                        3.560650587081909,
+                        1.386339545249939,
+                        0.04233968257904053,
+                        0.0013633022317662835
+                    ],
+                    "min": [
+                        -0.4828203022480011,
+                        -0.3255046010017395,
+                        0.008128180168569088,
+                        0.35277295112609863,
+                        -3.641430377960205,
+                        -1.842738389968872,
+                        -0.0013586411951109767,
+                        -0.042040832340717316
+                    ],
+                    "mean": [
+                        -0.04651879519224167,
+                        0.03440921753644943,
+                        0.7645525336265564,
+                        2.972202777862549,
+                        -0.22047005593776703,
+                        -0.1255796253681183,
+                        0.026914266869425774,
+                        -0.02719070389866829
+                    ],
+                    "std": [
+                        0.10494378954172134,
+                        0.15176637470722198,
+                        0.3785160183906555,
+                        0.34427398443222046,
+                        0.9069469571113586,
+                        0.3253920078277588,
+                        0.014175914227962494,
+                        0.014058894477784634
+                    ],
+                    "q01": [
+                        -0.4828203022480011,
+                        -0.3255046010017395,
+                        0.008128180168569088,
+                        0.35277295112609863,
+                        -3.641430377960205,
+                        -1.842738389968872,
+                        -0.0013586411951109767,
+                        -0.042040832340717316
+                    ],
+                    "q99": [
+                        0.21031762659549713,
+                        0.39128610491752625,
+                        1.3660105466842651,
+                        3.6714255809783936,
+                        3.560650587081909,
+                        1.386339545249939,
+                        0.04233968257904053,
+                        0.0013633022317662835
+                    ]
+                }
+            },
+            "action": {
+                "actions": {
+                    "max": [
+                        0.9375,
+                        0.9375,
+                        0.9375,
+                        0.3557142913341522,
+                        0.375,
+                        0.375,
+                        1.0
+                    ],
+                    "min": [
+                        -0.9375,
+                        -0.9375,
+                        -0.9375,
+                        -0.2582142949104309,
+                        -0.375,
+                        -0.3675000071525574,
+                        -1.0
+                    ],
+                    "mean": [
+                        0.06278137117624283,
+                        0.0868409126996994,
+                        -0.09037282317876816,
+                        0.0005407406715676188,
+                        0.005643361248075962,
+                        -0.005229088477790356,
+                        -0.04964079707860947
+                    ],
+                    "std": [
+                        0.3355240225791931,
+                        0.3784470558166504,
+                        0.44472837448120117,
+                        0.03924351558089256,
+                        0.06339313089847565,
+                        0.07797032594680786,
+                        0.9987710118293762
+                    ],
+                    "q01": [
+                        -0.9375,
+                        -0.9375,
+                        -0.9375,
+                        -0.2582142949104309,
+                        -0.375,
+                        -0.3675000071525574,
+                        -1.0
+                    ],
+                    "q99": [
+                        0.9375,
+                        0.9375,
+                        0.9375,
+                        0.3557142913341522,
+                        0.375,
+                        0.375,
+                        1.0
+                    ]
+                }
+            }
+        },
+        "modalities": {
+            "video": {
+                "image": {
+                    "resolution": [
+                        256,
+                        256
+                    ],
+                    "channels": 3,
+                    "fps": 10.0
+                },
+                "wrist_image": {
+                    "resolution": [
+                        256,
+                        256
+                    ],
+                    "channels": 3,
+                    "fps": 10.0
+                }
+            },
+            "state": {
+                "state": {
+                    "absolute": true,
+                    "rotation_type": null,
+                    "shape": [
+                        8
+                    ],
+                    "continuous": true
+                }
+            },
+            "action": {
+                "actions": {
+                    "absolute": true,
+                    "rotation_type": null,
+                    "shape": [
+                        7
+                    ],
+                    "continuous": true
+                }
+            }
+        },
+        "embodiment_tag": "libero_sim"
+    }
+}

model-00001-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61475262aba12c64dfd8641221a64cb56034d8cfee18e2a86241b06741d5b3e7
+size 4285869314

model-00002-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0f5123deb52f7f7b17755c77012d05b1e2570b32b89c2d798de05d258b84073b
+size 4286590904

model-00003-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a2150a9f476fa1c7293750f0e726f0e48fee1a6b0260791ffd77e807213d393
+size 4291152304

model-00004-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa3c37ddf58c30dfa43e4abcefa8d4cf096d7b9848324bb09810634113593bf3
+size 4278429536

model-00005-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e1f4f32b2936656486fa65c44bb9aea553d1d6fd442a500924bafa6f08d9e18
+size 4278429480

model-00006-of-00006.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f70a434372bbaa3b776add27b11c2af591ea1fe37f18934010fe94d126664002
+size 3828500456