File size: 4,185 Bytes
ef4a0a8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 | action_dim: 20
action_fields:
- robot__action__poses__left::panda__xyz_relative
- robot__action__poses__right::panda__xyz_relative
- robot__action__poses__left::panda__rot_6d_relative
- robot__action__poses__right::panda__rot_6d_relative
- robot__action__grippers__left::panda_hand
- robot__action__grippers__right::panda_hand
allow_multiple_epochs: true
augmentation:
enabled: true
image:
color_jitter:
brightness: 0.2
contrast: 0.4
enabled: true
hue:
- -0.05
- 0.05
saturation: 0.2
crop:
enabled: true
mode: random
shape:
- 224
- 224
point_cloud:
color_jitter:
brightness: 0.2
contrast: 0.4
enabled: false
hue:
- -0.05
- 0.05
saturation: 0.2
camera_names:
- scene_right_0
- scene_left_0
- wrist_left_plus
- wrist_right_minus
dataloader_in_order: false
dataset_cache:
cache_dir: null
cache_size_gb: null
cache_verbose: null
enabled: false
dataset_manifest:
- null
dataset_modality:
- robotics
dataset_statistics:
- null
dataset_weighting:
- 1.0
extrinsics_fields:
- extrinsics.scene_right_0
- extrinsics.scene_left_0
- extrinsics.wrist_left_minus
- extrinsics.wrist_left_plus
- extrinsics.wrist_right_minus
- extrinsics.wrist_right_plus
hf_fast_tokenizer_rayon_threads: 16
hf_fast_tokenizers_parallelism: true
image_indices:
- -1
- 0
image_names:
- scene_right_0_t-1
- scene_left_0_t-1
- wrist_left_plus_t-1
- wrist_right_minus_t-1
- scene_right_0_t0
- scene_left_0_t0
- wrist_left_plus_t0
- wrist_right_minus_t0
image_size: 224
img_num_tokens: 64
intrinsics_fields:
- intrinsics.scene_right_0
- intrinsics.scene_left_0
- intrinsics.wrist_left_minus
- intrinsics.wrist_left_plus
- intrinsics.wrist_right_minus
- intrinsics.wrist_right_plus
language_instruction_types:
- original
- randomized
- verbose
- alternative
lowdim_future_timesteps: 8
lowdim_past_timesteps: 1
mask_padded_images: true
max_text_seq_len: null
normalization:
centered_norm: true
enabled: true
epsilon: 0.01
field_configs:
robot__action__grippers__left::panda_hand:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
robot__action__grippers__right::panda_hand:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
robot__action__poses__left::panda__rot_6d_relative:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
robot__action__poses__left::panda__xyz_relative:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
robot__action__poses__right::panda__rot_6d_relative:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
robot__action__poses__right::panda__xyz_relative:
enabled: true
epsilon: 0.01
method: percentile_1_99
scope: per_timestep
include_fields:
- robot__action__poses__left::panda__xyz_relative
- robot__action__poses__right::panda__xyz_relative
- robot__action__poses__left::panda__rot_6d_relative
- robot__action__poses__right::panda__rot_6d_relative
- robot__action__grippers__left::panda_hand
- robot__action__grippers__right::panda_hand
lowdim_future_timesteps: 19
lowdim_past_timesteps: 5
method: percentile_1_99
scope: per_timestep
num_workers: 8
pad_missing_images: true
point_cloud_num_points: 4096
pose_groups:
- name: left_panda_action
position_key: robot__action__poses__left::panda__xyz
rotation_key: robot__action__poses__left::panda__rot_6d
- name: right_panda_action
position_key: robot__action__poses__right::panda__xyz
rotation_key: robot__action__poses__right::panda__rot_6d
prefetch_factor: 8
processor: HuggingFaceTB/SmolVLM2-256M-Video-Instruct
processor_kwargs:
max_image_size:
longest_edge: 224
size:
longest_edge: 224
proprioception_dim: 0
proprioception_fields: []
seed: 42
seq_len: 2048
shuffle: true
shuffle_buffer_size: 2000
shuffle_initial: 500
tactile_fields: []
type: robotics
use_hf_fast_tokenizer: true
use_point_cloud: false
val_dataset_manifest: []
val_dataset_statistics: []
val_dataset_weighting: []
|