zhenyuzhao commited on
Commit
35699cc
·
verified ·
1 Parent(s): b3798c6

Add GR00T real checkpoint for Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water

Browse files
Files changed (33) hide show
  1. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/config.json +70 -0
  2. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/embodiment_id.json +14 -0
  3. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/conf.yaml +253 -0
  4. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/config.yaml +328 -0
  5. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/dataset_statistics.json +641 -0
  6. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/final_model_config.json +53 -0
  7. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/final_processor_config.json +0 -0
  8. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model-00001-of-00002.safetensors +3 -0
  9. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model-00002-of-00002.safetensors +3 -0
  10. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model.safetensors.index.json +0 -0
  11. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/optimizer.pt +3 -0
  12. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/processor_config.json +650 -0
  13. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/rng_state.pth +3 -0
  14. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/scheduler.pt +3 -0
  15. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/statistics.json +0 -0
  16. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/trainer_state.json +0 -0
  17. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/training_args.bin +3 -0
  18. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/wandb_config.json +1 -0
  19. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/config.json +70 -0
  20. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/conf.yaml +253 -0
  21. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/config.yaml +328 -0
  22. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/dataset_statistics.json +641 -0
  23. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/final_model_config.json +53 -0
  24. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/final_processor_config.json +0 -0
  25. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model-00001-of-00002.safetensors +3 -0
  26. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model-00002-of-00002.safetensors +3 -0
  27. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model.safetensors.index.json +0 -0
  28. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/openloop_eval_summary.json +82 -0
  29. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/embodiment_id.json +14 -0
  30. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/processor_config.json +650 -0
  31. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/statistics.json +0 -0
  32. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/training_args.bin +3 -0
  33. gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/wandb_config.json +1 -0
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_horizon": 50,
3
+ "add_pos_embed": true,
4
+ "apply_sincos_state_encoding": true,
5
+ "architectures": [
6
+ "Gr00tN1d6"
7
+ ],
8
+ "attn_dropout": 0.2,
9
+ "attn_implementation": null,
10
+ "backbone_embedding_dim": 2048,
11
+ "backbone_model_type": "eagle",
12
+ "backbone_trainable_params_fp32": true,
13
+ "collator_overwrite_image_inputs": false,
14
+ "color_jitter_params": {
15
+ "brightness": 0.1,
16
+ "contrast": 0.1,
17
+ "hue": 0.1,
18
+ "saturation": 0.1
19
+ },
20
+ "crop_fraction": 0.95,
21
+ "diffusion_model_cfg": {
22
+ "attention_head_dim": 48,
23
+ "dropout": 0.2,
24
+ "final_dropout": true,
25
+ "interleave_self_attention": true,
26
+ "norm_type": "ada_norm",
27
+ "num_attention_heads": 32,
28
+ "num_layers": 32,
29
+ "output_dim": 1024,
30
+ "positional_embeddings": null
31
+ },
32
+ "eagle_collator": true,
33
+ "formalize_language": true,
34
+ "gemma_collator": false,
35
+ "hidden_size": 1024,
36
+ "image_crop_size": null,
37
+ "image_target_size": null,
38
+ "input_embedding_dim": 1536,
39
+ "load_bf16": true,
40
+ "max_action_dim": 128,
41
+ "max_num_embodiments": 32,
42
+ "max_seq_len": 1024,
43
+ "max_state_dim": 128,
44
+ "model_dtype": "bfloat16",
45
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
46
+ "model_type": "Gr00tN1d6",
47
+ "noise_beta_alpha": 1.5,
48
+ "noise_beta_beta": 1.0,
49
+ "noise_s": 0.999,
50
+ "num_inference_timesteps": 4,
51
+ "num_timestep_buckets": 1000,
52
+ "random_rotation_angle": null,
53
+ "reproject_vision": false,
54
+ "select_layer": 16,
55
+ "shortest_image_edge": 256,
56
+ "state_dropout_prob": 0.0,
57
+ "torch_dtype": "bfloat16",
58
+ "transformers_version": "4.51.3",
59
+ "tune_diffusion_model": true,
60
+ "tune_llm": false,
61
+ "tune_projector": true,
62
+ "tune_top_llm_layers": 4,
63
+ "tune_visual": false,
64
+ "tune_vlln": true,
65
+ "use_albumentations_transforms": true,
66
+ "use_alternate_vl_dit": true,
67
+ "use_flash_attention": true,
68
+ "use_relative_action": true,
69
+ "use_vlln": true
70
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/embodiment_id.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "robocasa_panda_omron": 13,
3
+ "gr1": 20,
4
+ "behavior_r1_pro": 24,
5
+ "unitree_g1": 8,
6
+ "oxe_google": 0,
7
+ "oxe_widowx": 1,
8
+ "libero_panda": 2,
9
+ "new_embodiment": 10,
10
+ "g1_ee_a16": 10,
11
+ "h1_ee_a16": 11,
12
+ "g1_upper_a16": 10,
13
+ "g1_loco_downstream": 10
14
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/conf.yaml ADDED
@@ -0,0 +1,253 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ load_config_path: null
2
+ model:
3
+ model_type: Gr00tN1d6
4
+ model_dtype: bfloat16
5
+ model_name: nvidia/Eagle-Block2A-2B-v2
6
+ backbone_model_type: eagle
7
+ model_revision: null
8
+ tune_top_llm_layers: 4
9
+ backbone_embedding_dim: 2048
10
+ tune_llm: false
11
+ tune_visual: false
12
+ select_layer: 16
13
+ reproject_vision: false
14
+ use_flash_attention: true
15
+ load_bf16: false
16
+ collator_overwrite_image_inputs: false
17
+ eagle_collator: true
18
+ backbone_trainable_params_fp32: true
19
+ image_crop_size: null
20
+ image_target_size: null
21
+ shortest_image_edge: 256
22
+ crop_fraction: 0.95
23
+ random_rotation_angle: null
24
+ color_jitter_params:
25
+ brightness: 0.3
26
+ contrast: 0.4
27
+ saturation: 0.5
28
+ hue: 0.08
29
+ use_albumentations_transforms: true
30
+ formalize_language: true
31
+ apply_sincos_state_encoding: true
32
+ use_relative_action: true
33
+ max_state_dim: 128
34
+ max_action_dim: 128
35
+ action_horizon: 50
36
+ hidden_size: 1024
37
+ input_embedding_dim: 1536
38
+ add_pos_embed: true
39
+ attn_dropout: 0.2
40
+ use_vlln: true
41
+ max_seq_len: 1024
42
+ use_alternate_vl_dit: true
43
+ attend_text_every_n_blocks: 2
44
+ diffusion_model_cfg:
45
+ attention_head_dim: 48
46
+ dropout: 0.2
47
+ final_dropout: true
48
+ interleave_self_attention: true
49
+ norm_type: ada_norm
50
+ num_attention_heads: 32
51
+ num_layers: 32
52
+ output_dim: 1024
53
+ positional_embeddings: null
54
+ num_inference_timesteps: 4
55
+ noise_beta_alpha: 1.5
56
+ noise_beta_beta: 1.0
57
+ noise_s: 0.999
58
+ num_timestep_buckets: 1000
59
+ tune_projector: true
60
+ tune_diffusion_model: true
61
+ tune_vlln: true
62
+ state_dropout_prob: 0.0
63
+ state_additive_noise_scale: 0.0
64
+ max_num_embodiments: 32
65
+ data:
66
+ datasets:
67
+ - dataset_paths:
68
+ - /hfm/data/real_teleop_g1/lerobot/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
69
+ embodiment_tag: g1_loco_downstream
70
+ mix_ratio: 1.0
71
+ dataset_type: physical_embodiment
72
+ val_dataset_path: null
73
+ modality_configs:
74
+ g1_loco_downstream:
75
+ video:
76
+ delta_indices:
77
+ - 0
78
+ modality_keys:
79
+ - rs_view
80
+ sin_cos_embedding_keys: null
81
+ mean_std_embedding_keys: null
82
+ action_configs: null
83
+ state:
84
+ delta_indices:
85
+ - 0
86
+ modality_keys:
87
+ - left_hand
88
+ - right_hand
89
+ - left_arm
90
+ - right_arm
91
+ - rpy
92
+ - height
93
+ sin_cos_embedding_keys: null
94
+ mean_std_embedding_keys: null
95
+ action_configs: null
96
+ action:
97
+ delta_indices:
98
+ - 0
99
+ - 1
100
+ - 2
101
+ - 3
102
+ - 4
103
+ - 5
104
+ - 6
105
+ - 7
106
+ - 8
107
+ - 9
108
+ - 10
109
+ - 11
110
+ - 12
111
+ - 13
112
+ - 14
113
+ - 15
114
+ modality_keys:
115
+ - left_hand
116
+ - right_hand
117
+ - left_arm
118
+ - right_arm
119
+ - rpy
120
+ - height
121
+ - torso_vx
122
+ - torso_vy
123
+ - torso_vyaw
124
+ - target_yaw
125
+ sin_cos_embedding_keys: null
126
+ mean_std_embedding_keys: null
127
+ action_configs:
128
+ - rep: ABSOLUTE
129
+ type: NON_EEF
130
+ format: DEFAULT
131
+ state_key: null
132
+ - rep: ABSOLUTE
133
+ type: NON_EEF
134
+ format: DEFAULT
135
+ state_key: null
136
+ - rep: ABSOLUTE
137
+ type: NON_EEF
138
+ format: DEFAULT
139
+ state_key: null
140
+ - rep: ABSOLUTE
141
+ type: NON_EEF
142
+ format: DEFAULT
143
+ state_key: null
144
+ - rep: ABSOLUTE
145
+ type: NON_EEF
146
+ format: DEFAULT
147
+ state_key: null
148
+ - rep: ABSOLUTE
149
+ type: NON_EEF
150
+ format: DEFAULT
151
+ state_key: null
152
+ - rep: ABSOLUTE
153
+ type: NON_EEF
154
+ format: DEFAULT
155
+ state_key: null
156
+ - rep: ABSOLUTE
157
+ type: NON_EEF
158
+ format: DEFAULT
159
+ state_key: null
160
+ - rep: ABSOLUTE
161
+ type: NON_EEF
162
+ format: DEFAULT
163
+ state_key: null
164
+ - rep: ABSOLUTE
165
+ type: NON_EEF
166
+ format: DEFAULT
167
+ state_key: null
168
+ language:
169
+ delta_indices:
170
+ - 0
171
+ modality_keys:
172
+ - annotation.human.task_description
173
+ sin_cos_embedding_keys: null
174
+ mean_std_embedding_keys: null
175
+ action_configs: null
176
+ download_cache: false
177
+ shard_size: 1024
178
+ episode_sampling_rate: 0.1
179
+ num_shards_per_epoch: 100000
180
+ override_pretraining_statistics: false
181
+ mode: single_turn
182
+ random_chop: 0.0
183
+ mock_dataset_mode: false
184
+ shuffle: true
185
+ seed: 42
186
+ multiprocessing_context: fork
187
+ allow_padding: false
188
+ subsample_ratio: 1.0
189
+ image_crop_size:
190
+ - 244
191
+ - 244
192
+ image_target_size:
193
+ - 224
194
+ - 224
195
+ video_backend: torchcodec
196
+ training:
197
+ output_dir: /hfm/zhenyu/Psi0/checkpoints/gr00t-real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
198
+ experiment_name: null
199
+ max_steps: 40000
200
+ global_batch_size: 12
201
+ batch_size: null
202
+ gradient_accumulation_steps: 1
203
+ learning_rate: 0.0001
204
+ lr_scheduler_type: cosine
205
+ weight_decay: 1.0e-05
206
+ warmup_ratio: 0.05
207
+ warmup_steps: 0
208
+ max_grad_norm: 1.0
209
+ optim: adamw_torch
210
+ start_from_checkpoint: /hfm/zhenyu/psi/checkpoints/G1WholebodyBendPick-v0-psi0/checkpoint-50000
211
+ reinit_action_head: false
212
+ tf32: true
213
+ fp16: false
214
+ bf16: true
215
+ eval_bf16: true
216
+ logging_steps: 10
217
+ save_steps: 40000
218
+ save_total_limit: 4
219
+ save_vl_model: false
220
+ upload_checkpoints: false
221
+ upload_every: 1000
222
+ upload_last_n_checkpoints: 5
223
+ max_concurrent_uploads: 2
224
+ eval_strategy: 'no'
225
+ eval_steps: 500
226
+ eval_set_split_ratio: 0.1
227
+ eval_batch_size: 2
228
+ save_best_eval_metric_name: ''
229
+ save_best_eval_metric_greater_is_better: true
230
+ deepspeed_stage: 2
231
+ gradient_checkpointing: false
232
+ transformers_trust_remote_code: true
233
+ transformers_local_files_only: false
234
+ transformers_cache_dir: null
235
+ transformers_access_token: null
236
+ use_ddp: false
237
+ ddp_bucket_cap_mb: 100
238
+ num_gpus: 1
239
+ dataloader_num_workers: 4
240
+ remove_unused_columns: false
241
+ use_wandb: false
242
+ wandb_project: finetune-gr00t-n1d6
243
+ enable_profiling: false
244
+ max_retries: 3
245
+ assert_loss_less_than: null
246
+ add_rl_callback: false
247
+ enable_open_loop_eval: false
248
+ open_loop_eval_traj_ids:
249
+ - 0
250
+ open_loop_eval_steps_per_traj: 100
251
+ open_loop_eval_plot_indices: null
252
+ max_steps: 40000
253
+ save_steps: 40000
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/config.yaml ADDED
@@ -0,0 +1,328 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object:gr00t.configs.base_config.Config
2
+ data: !!python/object:gr00t.configs.data.data_config.DataConfig
3
+ allow_padding: false
4
+ datasets:
5
+ - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
6
+ dataset_paths:
7
+ - /hfm/data/real_teleop_g1/lerobot/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
8
+ dataset_type: physical_embodiment
9
+ embodiment_tag: g1_loco_downstream
10
+ mix_ratio: 1.0
11
+ val_dataset_path: null
12
+ download_cache: false
13
+ episode_sampling_rate: 0.1
14
+ image_crop_size:
15
+ - 244
16
+ - 244
17
+ image_target_size:
18
+ - 224
19
+ - 224
20
+ mock_dataset_mode: false
21
+ modality_configs:
22
+ g1_loco_downstream:
23
+ action: !!python/object:gr00t.data.types.ModalityConfig
24
+ action_configs:
25
+ - !!python/object:gr00t.data.types.ActionConfig
26
+ format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
27
+ - default
28
+ rep: &id002 !!python/object/apply:gr00t.data.types.ActionRepresentation
29
+ - absolute
30
+ state_key: null
31
+ type: &id003 !!python/object/apply:gr00t.data.types.ActionType
32
+ - non_eef
33
+ - !!python/object:gr00t.data.types.ActionConfig
34
+ format: *id001
35
+ rep: *id002
36
+ state_key: null
37
+ type: *id003
38
+ - !!python/object:gr00t.data.types.ActionConfig
39
+ format: *id001
40
+ rep: *id002
41
+ state_key: null
42
+ type: *id003
43
+ - !!python/object:gr00t.data.types.ActionConfig
44
+ format: *id001
45
+ rep: *id002
46
+ state_key: null
47
+ type: *id003
48
+ - !!python/object:gr00t.data.types.ActionConfig
49
+ format: *id001
50
+ rep: *id002
51
+ state_key: null
52
+ type: *id003
53
+ - !!python/object:gr00t.data.types.ActionConfig
54
+ format: *id001
55
+ rep: *id002
56
+ state_key: null
57
+ type: *id003
58
+ - !!python/object:gr00t.data.types.ActionConfig
59
+ format: *id001
60
+ rep: *id002
61
+ state_key: null
62
+ type: *id003
63
+ - !!python/object:gr00t.data.types.ActionConfig
64
+ format: *id001
65
+ rep: *id002
66
+ state_key: null
67
+ type: *id003
68
+ - !!python/object:gr00t.data.types.ActionConfig
69
+ format: *id001
70
+ rep: *id002
71
+ state_key: null
72
+ type: *id003
73
+ - !!python/object:gr00t.data.types.ActionConfig
74
+ format: *id001
75
+ rep: *id002
76
+ state_key: null
77
+ type: *id003
78
+ delta_indices:
79
+ - 0
80
+ - 1
81
+ - 2
82
+ - 3
83
+ - 4
84
+ - 5
85
+ - 6
86
+ - 7
87
+ - 8
88
+ - 9
89
+ - 10
90
+ - 11
91
+ - 12
92
+ - 13
93
+ - 14
94
+ - 15
95
+ mean_std_embedding_keys: null
96
+ modality_keys:
97
+ - left_hand
98
+ - right_hand
99
+ - left_arm
100
+ - right_arm
101
+ - rpy
102
+ - height
103
+ - torso_vx
104
+ - torso_vy
105
+ - torso_vyaw
106
+ - target_yaw
107
+ sin_cos_embedding_keys: null
108
+ language: !!python/object:gr00t.data.types.ModalityConfig
109
+ action_configs: null
110
+ delta_indices:
111
+ - 0
112
+ mean_std_embedding_keys: null
113
+ modality_keys:
114
+ - annotation.human.task_description
115
+ sin_cos_embedding_keys: null
116
+ state: !!python/object:gr00t.data.types.ModalityConfig
117
+ action_configs: null
118
+ delta_indices:
119
+ - 0
120
+ mean_std_embedding_keys: null
121
+ modality_keys:
122
+ - left_hand
123
+ - right_hand
124
+ - left_arm
125
+ - right_arm
126
+ - rpy
127
+ - height
128
+ sin_cos_embedding_keys: null
129
+ video: !!python/object:gr00t.data.types.ModalityConfig
130
+ action_configs: null
131
+ delta_indices:
132
+ - 0
133
+ mean_std_embedding_keys: null
134
+ modality_keys:
135
+ - rs_view
136
+ sin_cos_embedding_keys: null
137
+ mode: single_turn
138
+ multiprocessing_context: fork
139
+ num_shards_per_epoch: 100000
140
+ override_pretraining_statistics: false
141
+ random_chop: 0.0
142
+ seed: 42
143
+ shard_size: 1024
144
+ shuffle: true
145
+ subsample_ratio: 1.0
146
+ video_backend: torchcodec
147
+ load_config_path: null
148
+ model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
149
+ _attn_implementation_autoset: false
150
+ _attn_implementation_internal: null
151
+ _commit_hash: null
152
+ _name_or_path: ''
153
+ action_horizon: 50
154
+ add_cross_attention: false
155
+ add_pos_embed: true
156
+ apply_sincos_state_encoding: true
157
+ architectures:
158
+ - Gr00tN1d6
159
+ attn_dropout: 0.2
160
+ attn_implementation: null
161
+ backbone_embedding_dim: 2048
162
+ backbone_model_type: eagle
163
+ backbone_trainable_params_fp32: true
164
+ bad_words_ids: null
165
+ begin_suppress_tokens: null
166
+ bos_token_id: null
167
+ chunk_size_feed_forward: 0
168
+ collator_overwrite_image_inputs: false
169
+ color_jitter_params:
170
+ brightness: 0.3
171
+ contrast: 0.4
172
+ hue: 0.08
173
+ saturation: 0.5
174
+ crop_fraction: 0.95
175
+ cross_attention_hidden_size: null
176
+ decoder_start_token_id: null
177
+ diffusion_model_cfg:
178
+ attention_head_dim: 48
179
+ dropout: 0.2
180
+ final_dropout: true
181
+ interleave_self_attention: true
182
+ norm_type: ada_norm
183
+ num_attention_heads: 32
184
+ num_layers: 32
185
+ output_dim: 1024
186
+ positional_embeddings: null
187
+ diversity_penalty: 0.0
188
+ do_sample: false
189
+ eagle_collator: true
190
+ early_stopping: false
191
+ encoder_no_repeat_ngram_size: 0
192
+ eos_token_id: null
193
+ exponential_decay_length_penalty: null
194
+ finetuning_task: null
195
+ forced_bos_token_id: null
196
+ forced_eos_token_id: null
197
+ formalize_language: true
198
+ gemma_collator: false
199
+ hidden_size: 1024
200
+ id2label:
201
+ 0: LABEL_0
202
+ 1: LABEL_1
203
+ image_crop_size: null
204
+ image_target_size: null
205
+ input_embedding_dim: 1536
206
+ is_decoder: false
207
+ is_encoder_decoder: false
208
+ label2id:
209
+ LABEL_0: 0
210
+ LABEL_1: 1
211
+ length_penalty: 1.0
212
+ load_bf16: false
213
+ max_action_dim: 128
214
+ max_length: 20
215
+ max_num_embodiments: 32
216
+ max_seq_len: 1024
217
+ max_state_dim: 128
218
+ min_length: 0
219
+ model_dtype: bfloat16
220
+ model_name: nvidia/Eagle-Block2A-2B-v2
221
+ model_type: Gr00tN1d6
222
+ no_repeat_ngram_size: 0
223
+ noise_beta_alpha: 1.5
224
+ noise_beta_beta: 1.0
225
+ noise_s: 0.999
226
+ num_beam_groups: 1
227
+ num_beams: 1
228
+ num_inference_timesteps: 4
229
+ num_return_sequences: 1
230
+ num_timestep_buckets: 1000
231
+ output_attentions: false
232
+ output_hidden_states: false
233
+ output_scores: false
234
+ pad_token_id: null
235
+ prefix: null
236
+ problem_type: null
237
+ pruned_heads: {}
238
+ random_rotation_angle: null
239
+ remove_invalid_values: false
240
+ repetition_penalty: 1.0
241
+ reproject_vision: false
242
+ return_dict: true
243
+ return_dict_in_generate: false
244
+ select_layer: 16
245
+ sep_token_id: null
246
+ shortest_image_edge: 256
247
+ state_dropout_prob: 0.0
248
+ suppress_tokens: null
249
+ task_specific_params: null
250
+ temperature: 1.0
251
+ tf_legacy_loss: false
252
+ tie_encoder_decoder: false
253
+ tie_word_embeddings: true
254
+ tokenizer_class: null
255
+ top_k: 50
256
+ top_p: 1.0
257
+ torch_dtype: bfloat16
258
+ torchscript: false
259
+ transformers_version: 4.51.3
260
+ tune_diffusion_model: true
261
+ tune_llm: false
262
+ tune_projector: true
263
+ tune_top_llm_layers: 4
264
+ tune_visual: false
265
+ tune_vlln: true
266
+ typical_p: 1.0
267
+ use_albumentations_transforms: true
268
+ use_alternate_vl_dit: true
269
+ use_bfloat16: false
270
+ use_flash_attention: true
271
+ use_relative_action: true
272
+ use_vlln: true
273
+ training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
274
+ add_rl_callback: false
275
+ assert_loss_less_than: null
276
+ batch_size: null
277
+ bf16: true
278
+ dataloader_num_workers: 4
279
+ ddp_bucket_cap_mb: 100
280
+ deepspeed_stage: 2
281
+ enable_open_loop_eval: false
282
+ enable_profiling: false
283
+ eval_batch_size: 2
284
+ eval_bf16: true
285
+ eval_set_split_ratio: 0.1
286
+ eval_steps: 500
287
+ eval_strategy: 'no'
288
+ experiment_name: null
289
+ fp16: false
290
+ global_batch_size: 12
291
+ gradient_accumulation_steps: 1
292
+ gradient_checkpointing: false
293
+ learning_rate: 0.0001
294
+ logging_steps: 10
295
+ lr_scheduler_type: cosine
296
+ max_concurrent_uploads: 2
297
+ max_grad_norm: 1.0
298
+ max_retries: 3
299
+ max_steps: 40000
300
+ num_gpus: 1
301
+ open_loop_eval_plot_indices: null
302
+ open_loop_eval_steps_per_traj: 100
303
+ open_loop_eval_traj_ids:
304
+ - 0
305
+ optim: adamw_torch
306
+ output_dir: /hfm/zhenyu/Psi0/checkpoints/gr00t-real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
307
+ reinit_action_head: false
308
+ remove_unused_columns: false
309
+ save_best_eval_metric_greater_is_better: true
310
+ save_best_eval_metric_name: ''
311
+ save_steps: 40000
312
+ save_total_limit: 4
313
+ save_vl_model: false
314
+ start_from_checkpoint: /hfm/zhenyu/psi/checkpoints/G1WholebodyBendPick-v0-psi0/checkpoint-50000
315
+ tf32: true
316
+ transformers_access_token: null
317
+ transformers_cache_dir: null
318
+ transformers_local_files_only: false
319
+ transformers_trust_remote_code: true
320
+ upload_checkpoints: false
321
+ upload_every: 1000
322
+ upload_last_n_checkpoints: 5
323
+ use_ddp: false
324
+ use_wandb: false
325
+ wandb_project: finetune-gr00t-n1d6
326
+ warmup_ratio: 0.05
327
+ warmup_steps: 0
328
+ weight_decay: 1.0e-05
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,641 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "g1_loco_downstream": {
3
+ "state": {
4
+ "left_hand": {
5
+ "min": [
6
+ -1.0061851739883423,
7
+ -0.08936958014965057,
8
+ 0.22508153319358826,
9
+ -1.2385730743408203,
10
+ -1.7122353315353394,
11
+ -1.1028423309326172,
12
+ -1.5751651525497437
13
+ ],
14
+ "max": [
15
+ 0.31211021542549133,
16
+ 0.8900995850563049,
17
+ 1.2004348039627075,
18
+ 0.05436473712325096,
19
+ -0.02029556967318058,
20
+ -0.02401699870824814,
21
+ -0.03348304331302643
22
+ ],
23
+ "mean": [
24
+ -0.524357259273529,
25
+ 0.5116726160049438,
26
+ 0.6721372604370117,
27
+ -0.4349469840526581,
28
+ -0.508317232131958,
29
+ -0.43180689215660095,
30
+ -0.4434279203414917
31
+ ],
32
+ "std": [
33
+ 0.2243678718805313,
34
+ 0.14296627044677734,
35
+ 0.10860829055309296,
36
+ 0.20835691690444946,
37
+ 0.2688620090484619,
38
+ 0.22207914292812347,
39
+ 0.16989512741565704
40
+ ],
41
+ "q01": [
42
+ -0.9798258036375046,
43
+ 0.2255818009376526,
44
+ 0.5424689680337906,
45
+ -0.8970937162637711,
46
+ -1.4855953991413116,
47
+ -1.000885361433029,
48
+ -1.1528545200824738
49
+ ],
50
+ "q99": [
51
+ -0.1169297218322754,
52
+ 0.8758351147174835,
53
+ 1.0207166373729701,
54
+ -0.052023166045546534,
55
+ -0.0410618856549263,
56
+ -0.039927643537521366,
57
+ -0.10575176924467272
58
+ ]
59
+ },
60
+ "right_hand": {
61
+ "min": [
62
+ -0.9382277131080627,
63
+ -0.7008255124092102,
64
+ -0.9635780453681946,
65
+ -0.16482605040073395,
66
+ 0.022227419540286064,
67
+ 0.011465233750641346,
68
+ 0.019190151244401932
69
+ ],
70
+ "max": [
71
+ 0.07547632604837418,
72
+ -0.22075261175632477,
73
+ -0.2170424908399582,
74
+ 1.2214250564575195,
75
+ 1.6621747016906738,
76
+ 0.9483049511909485,
77
+ 1.2975637912750244
78
+ ],
79
+ "mean": [
80
+ -0.5186124444007874,
81
+ -0.41561663150787354,
82
+ -0.6194753646850586,
83
+ 0.3872115910053253,
84
+ 0.4917716979980469,
85
+ 0.4035561978816986,
86
+ 0.3966030776500702
87
+ ],
88
+ "std": [
89
+ 0.13187938928604126,
90
+ 0.07076243311166763,
91
+ 0.09188053011894226,
92
+ 0.25810912251472473,
93
+ 0.2462475299835205,
94
+ 0.2178368717432022,
95
+ 0.16778795421123505
96
+ ],
97
+ "q01": [
98
+ -0.8237687945365906,
99
+ -0.6008797645568847,
100
+ -0.8918748825788498,
101
+ -0.155739439278841,
102
+ 0.032913535833358765,
103
+ 0.024460958316922188,
104
+ 0.019526662304997444
105
+ ],
106
+ "q99": [
107
+ -0.21736681610346664,
108
+ -0.28435189127922056,
109
+ -0.5137390494346619,
110
+ 0.7676283627748489,
111
+ 1.4599360227584839,
112
+ 0.6817445158958435,
113
+ 1.1090852260589599
114
+ ]
115
+ },
116
+ "left_arm": {
117
+ "min": [
118
+ -1.9539560079574585,
119
+ 0.014932344667613506,
120
+ -0.9010579586029053,
121
+ -0.6989918947219849,
122
+ -0.8808764815330505,
123
+ -1.3907837867736816,
124
+ -1.2593215703964233
125
+ ],
126
+ "max": [
127
+ 0.5761176347732544,
128
+ 1.100870966911316,
129
+ 0.7628798484802246,
130
+ 1.3479257822036743,
131
+ 1.4265302419662476,
132
+ 0.6714234352111816,
133
+ 0.6750589609146118
134
+ ],
135
+ "mean": [
136
+ -0.5292406678199768,
137
+ 0.4849226474761963,
138
+ -0.055224090814590454,
139
+ 0.2777264714241028,
140
+ 0.0669676661491394,
141
+ -0.42649492621421814,
142
+ -0.586437463760376
143
+ ],
144
+ "std": [
145
+ 0.676323652267456,
146
+ 0.1855972856283188,
147
+ 0.2814444303512573,
148
+ 0.43967705965042114,
149
+ 0.38776856660842896,
150
+ 0.25116971135139465,
151
+ 0.24483007192611694
152
+ ],
153
+ "q01": [
154
+ -1.6034354269504547,
155
+ 0.15880117490887644,
156
+ -0.6832776308059693,
157
+ -0.49059823155403137,
158
+ -0.6681205630302429,
159
+ -0.943893313407898,
160
+ -1.059616094827652
161
+ ],
162
+ "q99": [
163
+ 0.47410314828157396,
164
+ 0.9649536401033401,
165
+ 0.5035945236682883,
166
+ 1.2618268728256226,
167
+ 0.8312959581613535,
168
+ 0.16624134108424143,
169
+ 0.03442713152617198
170
+ ]
171
+ },
172
+ "right_arm": {
173
+ "min": [
174
+ -1.301702618598938,
175
+ -0.6721951961517334,
176
+ -0.2562706768512726,
177
+ -0.6034896373748779,
178
+ -0.4228993356227875,
179
+ -1.0893871784210205,
180
+ -0.7712242007255554
181
+ ],
182
+ "max": [
183
+ 0.7493496537208557,
184
+ 0.11452125757932663,
185
+ 0.42554786801338196,
186
+ 1.298634648323059,
187
+ 0.48713478446006775,
188
+ 0.6239774227142334,
189
+ 1.0436362028121948
190
+ ],
191
+ "mean": [
192
+ -0.4488898515701294,
193
+ -0.052791301161050797,
194
+ 0.05462970584630966,
195
+ 0.5688164830207825,
196
+ -0.055344775319099426,
197
+ -0.22808769345283508,
198
+ 0.10722843557596207
199
+ ],
200
+ "std": [
201
+ 0.4263588488101959,
202
+ 0.12731978297233582,
203
+ 0.09462971240282059,
204
+ 0.5017045736312866,
205
+ 0.09741094708442688,
206
+ 0.307020902633667,
207
+ 0.205648273229599
208
+ ],
209
+ "q01": [
210
+ -1.0721351623535156,
211
+ -0.49051433801651,
212
+ -0.1922467440366745,
213
+ -0.3724535495042801,
214
+ -0.261208176612854,
215
+ -0.8866654604673385,
216
+ -0.3525379553437233
217
+ ],
218
+ "q99": [
219
+ 0.5084475308656682,
220
+ 0.09847437590360641,
221
+ 0.31654173284769005,
222
+ 1.2630654573440552,
223
+ 0.2419453300535678,
224
+ 0.5242840051651001,
225
+ 0.8154258728027344
226
+ ]
227
+ },
228
+ "rpy": {
229
+ "min": [
230
+ -0.1804429143667221,
231
+ -0.13963909447193146,
232
+ -0.6072064638137817
233
+ ],
234
+ "max": [
235
+ 0.08730942010879517,
236
+ 0.28669121861457825,
237
+ 0.3420403003692627
238
+ ],
239
+ "mean": [
240
+ -0.02583920769393444,
241
+ 0.120397187769413,
242
+ -0.1572193056344986
243
+ ],
244
+ "std": [
245
+ 0.031078113242983818,
246
+ 0.06188545376062393,
247
+ 0.1938641220331192
248
+ ],
249
+ "q01": [
250
+ -0.08460558950901031,
251
+ -0.05298442132771015,
252
+ -0.5683319628238678
253
+ ],
254
+ "q99": [
255
+ 0.059636705368757204,
256
+ 0.23967182561755168,
257
+ 0.1393231652677059
258
+ ]
259
+ },
260
+ "height": {
261
+ "min": [
262
+ 0.6901783347129822
263
+ ],
264
+ "max": [
265
+ 0.839439868927002
266
+ ],
267
+ "mean": [
268
+ 0.7572857737541199
269
+ ],
270
+ "std": [
271
+ 0.016777420416473413
272
+ ],
273
+ "q01": [
274
+ 0.7088072448968887
275
+ ],
276
+ "q99": [
277
+ 0.7952491819858551
278
+ ]
279
+ }
280
+ },
281
+ "action": {
282
+ "left_hand": {
283
+ "min": [
284
+ -1.0481975078582764,
285
+ -0.27768605947494507,
286
+ 0.3844982385635376,
287
+ -1.2728569507598877,
288
+ -1.7463293075561523,
289
+ -1.1422605514526367,
290
+ -1.6586812734603882
291
+ ],
292
+ "max": [
293
+ 0.4100879728794098,
294
+ 0.9210000038146973,
295
+ 1.254960298538208,
296
+ 0.0010000000474974513,
297
+ 0.0010000000474974513,
298
+ 0.0010000000474974513,
299
+ 0.0010000000474974513
300
+ ],
301
+ "mean": [
302
+ -0.5317995548248291,
303
+ 0.5330548882484436,
304
+ 0.6755692362785339,
305
+ -0.43436452746391296,
306
+ -0.5135447382926941,
307
+ -0.4360726773738861,
308
+ -0.44873759150505066
309
+ ],
310
+ "std": [
311
+ 0.23812484741210938,
312
+ 0.16801197826862335,
313
+ 0.13376471400260925,
314
+ 0.23222216963768005,
315
+ 0.29362037777900696,
316
+ 0.2515135407447815,
317
+ 0.19146300852298737
318
+ ],
319
+ "q01": [
320
+ -0.9957153797149658,
321
+ 0.18187148869037628,
322
+ 0.519504725933075,
323
+ -0.9811010956764221,
324
+ -1.5662461996078492,
325
+ -1.0506287038326263,
326
+ -1.223601657152176
327
+ ],
328
+ "q99": [
329
+ -0.12964366376399994,
330
+ 0.9210000038146973,
331
+ 1.0971701681613921,
332
+ 0.0010000000474974513,
333
+ 0.0010000000474974513,
334
+ 0.0010000000474974513,
335
+ -0.08886573091149377
336
+ ]
337
+ },
338
+ "right_hand": {
339
+ "min": [
340
+ -0.9686063528060913,
341
+ -0.7862647771835327,
342
+ -1.034379243850708,
343
+ -0.0010000000474974513,
344
+ -0.0010000000474974513,
345
+ -0.0010000000474974513,
346
+ -0.0010000000474974513
347
+ ],
348
+ "max": [
349
+ 0.11807795614004135,
350
+ -0.13724753260612488,
351
+ -0.4963247776031494,
352
+ 1.2912133932113647,
353
+ 1.7463293075561523,
354
+ 0.9923550486564636,
355
+ 1.6118189096450806
356
+ ],
357
+ "mean": [
358
+ -0.47038841247558594,
359
+ -0.5184828639030457,
360
+ -0.6593118906021118,
361
+ 0.6895503997802734,
362
+ 0.5141050219535828,
363
+ 0.47509223222732544,
364
+ 0.4021771550178528
365
+ ],
366
+ "std": [
367
+ 0.15588711202144623,
368
+ 0.07128255069255829,
369
+ 0.11441590636968613,
370
+ 0.4033379852771759,
371
+ 0.2591473460197449,
372
+ 0.28332775831222534,
373
+ 0.19387704133987427
374
+ ],
375
+ "q01": [
376
+ -0.8572984337806702,
377
+ -0.7032237648963928,
378
+ -0.9383467495441437,
379
+ -0.0010000000474974513,
380
+ -0.0010000000474974513,
381
+ -0.0010000000474974513,
382
+ -0.0010000000474974513
383
+ ],
384
+ "q99": [
385
+ -0.15663587525487035,
386
+ -0.3310527935624125,
387
+ -0.5397384762763977,
388
+ 1.2012066185474395,
389
+ 1.5254341661930078,
390
+ 0.8683193325996399,
391
+ 1.1839898169040537
392
+ ]
393
+ },
394
+ "left_arm": {
395
+ "min": [
396
+ -2.0079522132873535,
397
+ 0.04465832561254501,
398
+ -1.0082837343215942,
399
+ -0.8037082552909851,
400
+ -1.1698691844940186,
401
+ -1.4439691305160522,
402
+ -1.2980564832687378
403
+ ],
404
+ "max": [
405
+ 0.6930765509605408,
406
+ 1.1467127799987793,
407
+ 0.887980043888092,
408
+ 1.3561404943466187,
409
+ 1.6023852825164795,
410
+ 0.7071925401687622,
411
+ 0.8807530403137207
412
+ ],
413
+ "mean": [
414
+ -0.537239134311676,
415
+ 0.49795472621917725,
416
+ -0.06242304667830467,
417
+ 0.27459681034088135,
418
+ 0.0673600509762764,
419
+ -0.43358004093170166,
420
+ -0.5968970060348511
421
+ ],
422
+ "std": [
423
+ 0.6874366998672485,
424
+ 0.18853120505809784,
425
+ 0.2929178476333618,
426
+ 0.45434024930000305,
427
+ 0.4136706292629242,
428
+ 0.2595718801021576,
429
+ 0.24986836314201355
430
+ ],
431
+ "q01": [
432
+ -1.6286808907985688,
433
+ 0.20261594727635385,
434
+ -0.7263113796710968,
435
+ -0.5180197864770889,
436
+ -0.6968675792217255,
437
+ -0.9553359180688858,
438
+ -1.0851860821247101
439
+ ],
440
+ "q99": [
441
+ 0.5095598667860028,
442
+ 0.9890699118375776,
443
+ 0.5474127471446989,
444
+ 1.2788401186466207,
445
+ 0.9281438887119284,
446
+ 0.20716316923499092,
447
+ 0.09342115409672125
448
+ ]
449
+ },
450
+ "right_arm": {
451
+ "min": [
452
+ -1.3225891590118408,
453
+ -0.8259854316711426,
454
+ -0.2885793149471283,
455
+ -0.8615938425064087,
456
+ -0.444199800491333,
457
+ -1.2984219789505005,
458
+ -0.8555077910423279
459
+ ],
460
+ "max": [
461
+ 0.9672761559486389,
462
+ 0.357116162776947,
463
+ 0.4602624475955963,
464
+ 1.3487075567245483,
465
+ 0.729375422000885,
466
+ 0.6513803601264954,
467
+ 1.052833914756775
468
+ ],
469
+ "mean": [
470
+ -0.42227938771247864,
471
+ -0.05299549922347069,
472
+ 0.057760849595069885,
473
+ 0.5489597320556641,
474
+ -0.04834311082959175,
475
+ -0.2310728132724762,
476
+ 0.11325573176145554
477
+ ],
478
+ "std": [
479
+ 0.4346851408481598,
480
+ 0.14826571941375732,
481
+ 0.0984012559056282,
482
+ 0.5546484589576721,
483
+ 0.11210273206233978,
484
+ 0.3245542347431183,
485
+ 0.21472640335559845
486
+ ],
487
+ "q01": [
488
+ -1.0687924444675445,
489
+ -0.4947426915168762,
490
+ -0.18176313862204552,
491
+ -0.5111332565546036,
492
+ -0.2779590994119644,
493
+ -0.9719170898199081,
494
+ -0.399456313252449
495
+ ],
496
+ "q99": [
497
+ 0.6194304704666128,
498
+ 0.21891917586326573,
499
+ 0.3378777891397475,
500
+ 1.3038406193256378,
501
+ 0.28321229368448236,
502
+ 0.5291351854801176,
503
+ 0.8172934919595717
504
+ ]
505
+ },
506
+ "rpy": {
507
+ "min": [
508
+ -0.1804429143667221,
509
+ -0.13963909447193146,
510
+ -0.6072064638137817
511
+ ],
512
+ "max": [
513
+ 0.08730942010879517,
514
+ 0.28669121861457825,
515
+ 0.3420403003692627
516
+ ],
517
+ "mean": [
518
+ -0.025878138840198517,
519
+ 0.1203930452466011,
520
+ -0.15729419887065887
521
+ ],
522
+ "std": [
523
+ 0.031076112762093544,
524
+ 0.061915408819913864,
525
+ 0.19382591545581818
526
+ ],
527
+ "q01": [
528
+ -0.08460558950901031,
529
+ -0.05354986395686865,
530
+ -0.5683319628238678
531
+ ],
532
+ "q99": [
533
+ 0.059636705368757204,
534
+ 0.23967182561755168,
535
+ 0.1393231652677059
536
+ ]
537
+ },
538
+ "height": {
539
+ "min": [
540
+ 0.6901783347129822
541
+ ],
542
+ "max": [
543
+ 0.839439868927002
544
+ ],
545
+ "mean": [
546
+ 0.7572655081748962
547
+ ],
548
+ "std": [
549
+ 0.016800325363874023
550
+ ],
551
+ "q01": [
552
+ 0.7088072448968887
553
+ ],
554
+ "q99": [
555
+ 0.7952491819858551
556
+ ]
557
+ },
558
+ "torso_vx": {
559
+ "min": [
560
+ 0.0
561
+ ],
562
+ "max": [
563
+ 0.0
564
+ ],
565
+ "mean": [
566
+ 0.0
567
+ ],
568
+ "std": [
569
+ 0.0
570
+ ],
571
+ "q01": [
572
+ 0.0
573
+ ],
574
+ "q99": [
575
+ 0.0
576
+ ]
577
+ },
578
+ "torso_vy": {
579
+ "min": [
580
+ 0.0
581
+ ],
582
+ "max": [
583
+ 0.0
584
+ ],
585
+ "mean": [
586
+ 0.0
587
+ ],
588
+ "std": [
589
+ 0.0
590
+ ],
591
+ "q01": [
592
+ 0.0
593
+ ],
594
+ "q99": [
595
+ 0.0
596
+ ]
597
+ },
598
+ "torso_vyaw": {
599
+ "min": [
600
+ 0.0
601
+ ],
602
+ "max": [
603
+ 0.0
604
+ ],
605
+ "mean": [
606
+ 0.0
607
+ ],
608
+ "std": [
609
+ 0.0
610
+ ],
611
+ "q01": [
612
+ 0.0
613
+ ],
614
+ "q99": [
615
+ 0.0
616
+ ]
617
+ },
618
+ "target_yaw": {
619
+ "min": [
620
+ 0.0
621
+ ],
622
+ "max": [
623
+ 0.0
624
+ ],
625
+ "mean": [
626
+ 0.0
627
+ ],
628
+ "std": [
629
+ 0.0
630
+ ],
631
+ "q01": [
632
+ 0.0
633
+ ],
634
+ "q99": [
635
+ 0.0
636
+ ]
637
+ }
638
+ },
639
+ "relative_action": {}
640
+ }
641
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/final_model_config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "Gr00tN1d6",
3
+ "model_dtype": "bfloat16",
4
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
5
+ "backbone_model_type": "eagle",
6
+ "model_revision": null,
7
+ "tune_top_llm_layers": 4,
8
+ "backbone_embedding_dim": 2048,
9
+ "tune_llm": false,
10
+ "tune_visual": false,
11
+ "select_layer": 16,
12
+ "reproject_vision": false,
13
+ "use_flash_attention": true,
14
+ "load_bf16": true,
15
+ "collator_overwrite_image_inputs": false,
16
+ "eagle_collator": true,
17
+ "backbone_trainable_params_fp32": true,
18
+ "apply_sincos_state_encoding": true,
19
+ "use_relative_action": true,
20
+ "max_state_dim": 128,
21
+ "max_action_dim": 128,
22
+ "action_horizon": 50,
23
+ "hidden_size": 1024,
24
+ "input_embedding_dim": 1536,
25
+ "add_pos_embed": true,
26
+ "attn_dropout": 0.2,
27
+ "use_vlln": true,
28
+ "max_seq_len": 1024,
29
+ "use_alternate_vl_dit": true,
30
+ "attend_text_every_n_blocks": 2,
31
+ "diffusion_model_cfg": {
32
+ "attention_head_dim": 48,
33
+ "dropout": 0.2,
34
+ "final_dropout": true,
35
+ "interleave_self_attention": true,
36
+ "norm_type": "ada_norm",
37
+ "num_attention_heads": 32,
38
+ "num_layers": 32,
39
+ "output_dim": 1024,
40
+ "positional_embeddings": null
41
+ },
42
+ "num_inference_timesteps": 4,
43
+ "noise_beta_alpha": 1.5,
44
+ "noise_beta_beta": 1.0,
45
+ "noise_s": 0.999,
46
+ "num_timestep_buckets": 1000,
47
+ "tune_projector": true,
48
+ "tune_diffusion_model": true,
49
+ "tune_vlln": true,
50
+ "state_dropout_prob": 0.0,
51
+ "state_additive_noise_scale": 0.0,
52
+ "max_num_embodiments": 32
53
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18753a5fcdf7d4c4ba71e0ba22571fc4d182795cc7d5e040339ecd054339ebb0
3
+ size 4990120184
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e9cc87647ddbf0961ed9d28ddf98cc592d6f2b94bf4baf3c564d60cf1da5755
3
+ size 4823190320
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d13e7e2ba9bcfc7d189a1214a5bba3a15c5562329e7fbf00f75e76f90046aff
3
+ size 12960193762
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/processor_config.json ADDED
@@ -0,0 +1,650 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "processor_class": "Gr00tN1d6Processor",
3
+ "processor_kwargs": {
4
+ "modality_configs": {
5
+ "behavior_r1_pro": {
6
+ "video": {
7
+ "delta_indices": [
8
+ 0
9
+ ],
10
+ "modality_keys": [
11
+ "observation.images.rgb.head_256_256",
12
+ "observation.images.rgb.left_wrist_256_256",
13
+ "observation.images.rgb.right_wrist_256_256"
14
+ ],
15
+ "sin_cos_embedding_keys": null,
16
+ "mean_std_embedding_keys": null,
17
+ "action_configs": null
18
+ },
19
+ "state": {
20
+ "delta_indices": [
21
+ 0
22
+ ],
23
+ "modality_keys": [
24
+ "robot_pos",
25
+ "robot_ori_cos",
26
+ "robot_ori_sin",
27
+ "robot_2d_ori",
28
+ "robot_2d_ori_cos",
29
+ "robot_2d_ori_sin",
30
+ "robot_lin_vel",
31
+ "robot_ang_vel",
32
+ "arm_left_qpos",
33
+ "arm_left_qpos_sin",
34
+ "arm_left_qpos_cos",
35
+ "eef_left_pos",
36
+ "eef_left_quat",
37
+ "gripper_left_qpos",
38
+ "arm_right_qpos",
39
+ "arm_right_qpos_sin",
40
+ "arm_right_qpos_cos",
41
+ "eef_right_pos",
42
+ "eef_right_quat",
43
+ "gripper_right_qpos",
44
+ "trunk_qpos"
45
+ ],
46
+ "sin_cos_embedding_keys": null,
47
+ "mean_std_embedding_keys": null,
48
+ "action_configs": null
49
+ },
50
+ "action": {
51
+ "delta_indices": [
52
+ 0,
53
+ 1,
54
+ 2,
55
+ 3,
56
+ 4,
57
+ 5,
58
+ 6,
59
+ 7,
60
+ 8,
61
+ 9,
62
+ 10,
63
+ 11,
64
+ 12,
65
+ 13,
66
+ 14,
67
+ 15,
68
+ 16,
69
+ 17,
70
+ 18,
71
+ 19,
72
+ 20,
73
+ 21,
74
+ 22,
75
+ 23,
76
+ 24,
77
+ 25,
78
+ 26,
79
+ 27,
80
+ 28,
81
+ 29,
82
+ 30,
83
+ 31
84
+ ],
85
+ "modality_keys": [
86
+ "base",
87
+ "torso",
88
+ "left_arm",
89
+ "left_gripper",
90
+ "right_arm",
91
+ "right_gripper"
92
+ ],
93
+ "sin_cos_embedding_keys": null,
94
+ "mean_std_embedding_keys": null,
95
+ "action_configs": [
96
+ {
97
+ "rep": "ABSOLUTE",
98
+ "type": "NON_EEF",
99
+ "format": "DEFAULT",
100
+ "state_key": null
101
+ },
102
+ {
103
+ "rep": "RELATIVE",
104
+ "type": "NON_EEF",
105
+ "format": "DEFAULT",
106
+ "state_key": "trunk_qpos"
107
+ },
108
+ {
109
+ "rep": "RELATIVE",
110
+ "type": "NON_EEF",
111
+ "format": "DEFAULT",
112
+ "state_key": "arm_left_qpos"
113
+ },
114
+ {
115
+ "rep": "ABSOLUTE",
116
+ "type": "NON_EEF",
117
+ "format": "DEFAULT",
118
+ "state_key": null
119
+ },
120
+ {
121
+ "rep": "RELATIVE",
122
+ "type": "NON_EEF",
123
+ "format": "DEFAULT",
124
+ "state_key": "arm_right_qpos"
125
+ },
126
+ {
127
+ "rep": "ABSOLUTE",
128
+ "type": "NON_EEF",
129
+ "format": "DEFAULT",
130
+ "state_key": null
131
+ }
132
+ ]
133
+ },
134
+ "language": {
135
+ "delta_indices": [
136
+ 0
137
+ ],
138
+ "modality_keys": [
139
+ "annotation.human.coarse_action"
140
+ ],
141
+ "sin_cos_embedding_keys": null,
142
+ "mean_std_embedding_keys": null,
143
+ "action_configs": null
144
+ }
145
+ },
146
+ "gr1": {
147
+ "video": {
148
+ "delta_indices": [
149
+ 0
150
+ ],
151
+ "modality_keys": [
152
+ "ego_view_bg_crop_pad_res256_freq20"
153
+ ],
154
+ "sin_cos_embedding_keys": null,
155
+ "mean_std_embedding_keys": null,
156
+ "action_configs": null
157
+ },
158
+ "state": {
159
+ "delta_indices": [
160
+ 0
161
+ ],
162
+ "modality_keys": [
163
+ "left_arm",
164
+ "right_arm",
165
+ "left_hand",
166
+ "right_hand",
167
+ "waist"
168
+ ],
169
+ "sin_cos_embedding_keys": [
170
+ "left_arm",
171
+ "right_arm",
172
+ "left_hand",
173
+ "right_hand",
174
+ "waist"
175
+ ],
176
+ "mean_std_embedding_keys": null,
177
+ "action_configs": null
178
+ },
179
+ "action": {
180
+ "delta_indices": [
181
+ 0,
182
+ 1,
183
+ 2,
184
+ 3,
185
+ 4,
186
+ 5,
187
+ 6,
188
+ 7,
189
+ 8,
190
+ 9,
191
+ 10,
192
+ 11,
193
+ 12,
194
+ 13,
195
+ 14,
196
+ 15
197
+ ],
198
+ "modality_keys": [
199
+ "left_arm",
200
+ "right_arm",
201
+ "left_hand",
202
+ "right_hand",
203
+ "waist"
204
+ ],
205
+ "sin_cos_embedding_keys": null,
206
+ "mean_std_embedding_keys": null,
207
+ "action_configs": [
208
+ {
209
+ "rep": "RELATIVE",
210
+ "type": "NON_EEF",
211
+ "format": "DEFAULT",
212
+ "state_key": null
213
+ },
214
+ {
215
+ "rep": "RELATIVE",
216
+ "type": "NON_EEF",
217
+ "format": "DEFAULT",
218
+ "state_key": null
219
+ },
220
+ {
221
+ "rep": "RELATIVE",
222
+ "type": "NON_EEF",
223
+ "format": "DEFAULT",
224
+ "state_key": null
225
+ },
226
+ {
227
+ "rep": "RELATIVE",
228
+ "type": "NON_EEF",
229
+ "format": "DEFAULT",
230
+ "state_key": null
231
+ },
232
+ {
233
+ "rep": "ABSOLUTE",
234
+ "type": "NON_EEF",
235
+ "format": "DEFAULT",
236
+ "state_key": null
237
+ }
238
+ ]
239
+ },
240
+ "language": {
241
+ "delta_indices": [
242
+ 0
243
+ ],
244
+ "modality_keys": [
245
+ "task"
246
+ ],
247
+ "sin_cos_embedding_keys": null,
248
+ "mean_std_embedding_keys": null,
249
+ "action_configs": null
250
+ }
251
+ },
252
+ "robocasa_panda_omron": {
253
+ "video": {
254
+ "delta_indices": [
255
+ 0
256
+ ],
257
+ "modality_keys": [
258
+ "res256_image_side_0",
259
+ "res256_image_side_1",
260
+ "res256_image_wrist_0"
261
+ ],
262
+ "sin_cos_embedding_keys": null,
263
+ "mean_std_embedding_keys": null,
264
+ "action_configs": null
265
+ },
266
+ "state": {
267
+ "delta_indices": [
268
+ 0
269
+ ],
270
+ "modality_keys": [
271
+ "end_effector_position_relative",
272
+ "end_effector_rotation_relative",
273
+ "gripper_qpos",
274
+ "base_position",
275
+ "base_rotation"
276
+ ],
277
+ "sin_cos_embedding_keys": null,
278
+ "mean_std_embedding_keys": null,
279
+ "action_configs": null
280
+ },
281
+ "action": {
282
+ "delta_indices": [
283
+ 0,
284
+ 1,
285
+ 2,
286
+ 3,
287
+ 4,
288
+ 5,
289
+ 6,
290
+ 7,
291
+ 8,
292
+ 9,
293
+ 10,
294
+ 11,
295
+ 12,
296
+ 13,
297
+ 14,
298
+ 15
299
+ ],
300
+ "modality_keys": [
301
+ "end_effector_position",
302
+ "end_effector_rotation",
303
+ "gripper_close",
304
+ "base_motion",
305
+ "control_mode"
306
+ ],
307
+ "sin_cos_embedding_keys": null,
308
+ "mean_std_embedding_keys": null,
309
+ "action_configs": [
310
+ {
311
+ "rep": "ABSOLUTE",
312
+ "type": "NON_EEF",
313
+ "format": "DEFAULT",
314
+ "state_key": null
315
+ },
316
+ {
317
+ "rep": "ABSOLUTE",
318
+ "type": "NON_EEF",
319
+ "format": "DEFAULT",
320
+ "state_key": null
321
+ },
322
+ {
323
+ "rep": "ABSOLUTE",
324
+ "type": "NON_EEF",
325
+ "format": "DEFAULT",
326
+ "state_key": null
327
+ },
328
+ {
329
+ "rep": "ABSOLUTE",
330
+ "type": "NON_EEF",
331
+ "format": "DEFAULT",
332
+ "state_key": null
333
+ },
334
+ {
335
+ "rep": "ABSOLUTE",
336
+ "type": "NON_EEF",
337
+ "format": "DEFAULT",
338
+ "state_key": null
339
+ }
340
+ ]
341
+ },
342
+ "language": {
343
+ "delta_indices": [
344
+ 0
345
+ ],
346
+ "modality_keys": [
347
+ "annotation.human.action.task_description"
348
+ ],
349
+ "sin_cos_embedding_keys": null,
350
+ "mean_std_embedding_keys": null,
351
+ "action_configs": null
352
+ }
353
+ },
354
+ "new_embodiment": {
355
+ "video": {
356
+ "delta_indices": [
357
+ 0
358
+ ],
359
+ "modality_keys": [
360
+ "rs_view"
361
+ ],
362
+ "sin_cos_embedding_keys": null,
363
+ "mean_std_embedding_keys": null,
364
+ "action_configs": null
365
+ },
366
+ "state": {
367
+ "delta_indices": [
368
+ 0
369
+ ],
370
+ "modality_keys": [
371
+ "left_hand",
372
+ "right_hand",
373
+ "left_arm",
374
+ "right_arm",
375
+ "rpy",
376
+ "height"
377
+ ],
378
+ "sin_cos_embedding_keys": null,
379
+ "mean_std_embedding_keys": null,
380
+ "action_configs": null
381
+ },
382
+ "action": {
383
+ "delta_indices": [
384
+ 0,
385
+ 1,
386
+ 2,
387
+ 3,
388
+ 4,
389
+ 5,
390
+ 6,
391
+ 7,
392
+ 8,
393
+ 9,
394
+ 10,
395
+ 11,
396
+ 12,
397
+ 13,
398
+ 14,
399
+ 15
400
+ ],
401
+ "modality_keys": [
402
+ "left_hand",
403
+ "right_hand",
404
+ "left_arm",
405
+ "right_arm",
406
+ "rpy",
407
+ "height",
408
+ "torso_vx",
409
+ "torso_vy",
410
+ "torso_vyaw",
411
+ "target_yaw"
412
+ ],
413
+ "sin_cos_embedding_keys": null,
414
+ "mean_std_embedding_keys": null,
415
+ "action_configs": [
416
+ {
417
+ "rep": "ABSOLUTE",
418
+ "type": "NON_EEF",
419
+ "format": "DEFAULT",
420
+ "state_key": null
421
+ },
422
+ {
423
+ "rep": "ABSOLUTE",
424
+ "type": "NON_EEF",
425
+ "format": "DEFAULT",
426
+ "state_key": null
427
+ },
428
+ {
429
+ "rep": "ABSOLUTE",
430
+ "type": "NON_EEF",
431
+ "format": "DEFAULT",
432
+ "state_key": null
433
+ },
434
+ {
435
+ "rep": "ABSOLUTE",
436
+ "type": "NON_EEF",
437
+ "format": "DEFAULT",
438
+ "state_key": null
439
+ },
440
+ {
441
+ "rep": "ABSOLUTE",
442
+ "type": "NON_EEF",
443
+ "format": "DEFAULT",
444
+ "state_key": null
445
+ },
446
+ {
447
+ "rep": "ABSOLUTE",
448
+ "type": "NON_EEF",
449
+ "format": "DEFAULT",
450
+ "state_key": null
451
+ },
452
+ {
453
+ "rep": "ABSOLUTE",
454
+ "type": "NON_EEF",
455
+ "format": "DEFAULT",
456
+ "state_key": null
457
+ },
458
+ {
459
+ "rep": "ABSOLUTE",
460
+ "type": "NON_EEF",
461
+ "format": "DEFAULT",
462
+ "state_key": null
463
+ },
464
+ {
465
+ "rep": "ABSOLUTE",
466
+ "type": "NON_EEF",
467
+ "format": "DEFAULT",
468
+ "state_key": null
469
+ },
470
+ {
471
+ "rep": "ABSOLUTE",
472
+ "type": "NON_EEF",
473
+ "format": "DEFAULT",
474
+ "state_key": null
475
+ }
476
+ ]
477
+ },
478
+ "language": {
479
+ "delta_indices": [
480
+ 0
481
+ ],
482
+ "modality_keys": [
483
+ "annotation.human.task_description"
484
+ ],
485
+ "sin_cos_embedding_keys": null,
486
+ "mean_std_embedding_keys": null,
487
+ "action_configs": null
488
+ }
489
+ },
490
+ "g1_loco_downstream": {
491
+ "video": {
492
+ "delta_indices": [
493
+ 0
494
+ ],
495
+ "modality_keys": [
496
+ "rs_view"
497
+ ],
498
+ "sin_cos_embedding_keys": null,
499
+ "mean_std_embedding_keys": null,
500
+ "action_configs": null
501
+ },
502
+ "state": {
503
+ "delta_indices": [
504
+ 0
505
+ ],
506
+ "modality_keys": [
507
+ "left_hand",
508
+ "right_hand",
509
+ "left_arm",
510
+ "right_arm",
511
+ "rpy",
512
+ "height"
513
+ ],
514
+ "sin_cos_embedding_keys": null,
515
+ "mean_std_embedding_keys": null,
516
+ "action_configs": null
517
+ },
518
+ "action": {
519
+ "delta_indices": [
520
+ 0,
521
+ 1,
522
+ 2,
523
+ 3,
524
+ 4,
525
+ 5,
526
+ 6,
527
+ 7,
528
+ 8,
529
+ 9,
530
+ 10,
531
+ 11,
532
+ 12,
533
+ 13,
534
+ 14,
535
+ 15
536
+ ],
537
+ "modality_keys": [
538
+ "left_hand",
539
+ "right_hand",
540
+ "left_arm",
541
+ "right_arm",
542
+ "rpy",
543
+ "height",
544
+ "torso_vx",
545
+ "torso_vy",
546
+ "torso_vyaw",
547
+ "target_yaw"
548
+ ],
549
+ "sin_cos_embedding_keys": null,
550
+ "mean_std_embedding_keys": null,
551
+ "action_configs": [
552
+ {
553
+ "rep": "ABSOLUTE",
554
+ "type": "NON_EEF",
555
+ "format": "DEFAULT",
556
+ "state_key": null
557
+ },
558
+ {
559
+ "rep": "ABSOLUTE",
560
+ "type": "NON_EEF",
561
+ "format": "DEFAULT",
562
+ "state_key": null
563
+ },
564
+ {
565
+ "rep": "ABSOLUTE",
566
+ "type": "NON_EEF",
567
+ "format": "DEFAULT",
568
+ "state_key": null
569
+ },
570
+ {
571
+ "rep": "ABSOLUTE",
572
+ "type": "NON_EEF",
573
+ "format": "DEFAULT",
574
+ "state_key": null
575
+ },
576
+ {
577
+ "rep": "ABSOLUTE",
578
+ "type": "NON_EEF",
579
+ "format": "DEFAULT",
580
+ "state_key": null
581
+ },
582
+ {
583
+ "rep": "ABSOLUTE",
584
+ "type": "NON_EEF",
585
+ "format": "DEFAULT",
586
+ "state_key": null
587
+ },
588
+ {
589
+ "rep": "ABSOLUTE",
590
+ "type": "NON_EEF",
591
+ "format": "DEFAULT",
592
+ "state_key": null
593
+ },
594
+ {
595
+ "rep": "ABSOLUTE",
596
+ "type": "NON_EEF",
597
+ "format": "DEFAULT",
598
+ "state_key": null
599
+ },
600
+ {
601
+ "rep": "ABSOLUTE",
602
+ "type": "NON_EEF",
603
+ "format": "DEFAULT",
604
+ "state_key": null
605
+ },
606
+ {
607
+ "rep": "ABSOLUTE",
608
+ "type": "NON_EEF",
609
+ "format": "DEFAULT",
610
+ "state_key": null
611
+ }
612
+ ]
613
+ },
614
+ "language": {
615
+ "delta_indices": [
616
+ 0
617
+ ],
618
+ "modality_keys": [
619
+ "annotation.human.task_description"
620
+ ],
621
+ "sin_cos_embedding_keys": null,
622
+ "mean_std_embedding_keys": null,
623
+ "action_configs": null
624
+ }
625
+ }
626
+ },
627
+ "image_crop_size": null,
628
+ "image_target_size": null,
629
+ "use_albumentations": true,
630
+ "random_rotation_angle": null,
631
+ "color_jitter_params": {
632
+ "brightness": 0.3,
633
+ "contrast": 0.4,
634
+ "saturation": 0.5,
635
+ "hue": 0.08
636
+ },
637
+ "shortest_image_edge": 256,
638
+ "crop_fraction": 0.95,
639
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
640
+ "model_type": "eagle",
641
+ "formalize_language": true,
642
+ "max_state_dim": 128,
643
+ "max_action_dim": 128,
644
+ "max_action_horizon": 50,
645
+ "use_percentiles": false,
646
+ "clip_outliers": true,
647
+ "apply_sincos_state_encoding": true,
648
+ "use_relative_action": true
649
+ }
650
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f0c369361b6dfa9a44c159f60682c5608b54b4778c6fa7593fac6a19045bca7
3
+ size 14645
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:566e63d72ec3169fd51f5dc061a30bf9c5195d001f715224cdd5f6178206f3e5
3
+ size 1465
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea69deca11ea127417d3806567ecd03668e914d426ea2a32a176fdbab73bb048
3
+ size 5905
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/checkpoint-40000/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "finetune-gr00t-n1d6", "run_id": "Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water"}
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_horizon": 50,
3
+ "add_pos_embed": true,
4
+ "apply_sincos_state_encoding": true,
5
+ "architectures": [
6
+ "Gr00tN1d6"
7
+ ],
8
+ "attn_dropout": 0.2,
9
+ "attn_implementation": null,
10
+ "backbone_embedding_dim": 2048,
11
+ "backbone_model_type": "eagle",
12
+ "backbone_trainable_params_fp32": true,
13
+ "collator_overwrite_image_inputs": false,
14
+ "color_jitter_params": {
15
+ "brightness": 0.1,
16
+ "contrast": 0.1,
17
+ "hue": 0.1,
18
+ "saturation": 0.1
19
+ },
20
+ "crop_fraction": 0.95,
21
+ "diffusion_model_cfg": {
22
+ "attention_head_dim": 48,
23
+ "dropout": 0.2,
24
+ "final_dropout": true,
25
+ "interleave_self_attention": true,
26
+ "norm_type": "ada_norm",
27
+ "num_attention_heads": 32,
28
+ "num_layers": 32,
29
+ "output_dim": 1024,
30
+ "positional_embeddings": null
31
+ },
32
+ "eagle_collator": true,
33
+ "formalize_language": true,
34
+ "gemma_collator": false,
35
+ "hidden_size": 1024,
36
+ "image_crop_size": null,
37
+ "image_target_size": null,
38
+ "input_embedding_dim": 1536,
39
+ "load_bf16": true,
40
+ "max_action_dim": 128,
41
+ "max_num_embodiments": 32,
42
+ "max_seq_len": 1024,
43
+ "max_state_dim": 128,
44
+ "model_dtype": "bfloat16",
45
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
46
+ "model_type": "Gr00tN1d6",
47
+ "noise_beta_alpha": 1.5,
48
+ "noise_beta_beta": 1.0,
49
+ "noise_s": 0.999,
50
+ "num_inference_timesteps": 4,
51
+ "num_timestep_buckets": 1000,
52
+ "random_rotation_angle": null,
53
+ "reproject_vision": false,
54
+ "select_layer": 16,
55
+ "shortest_image_edge": 256,
56
+ "state_dropout_prob": 0.0,
57
+ "torch_dtype": "bfloat16",
58
+ "transformers_version": "4.51.3",
59
+ "tune_diffusion_model": true,
60
+ "tune_llm": false,
61
+ "tune_projector": true,
62
+ "tune_top_llm_layers": 4,
63
+ "tune_visual": false,
64
+ "tune_vlln": true,
65
+ "use_albumentations_transforms": true,
66
+ "use_alternate_vl_dit": true,
67
+ "use_flash_attention": true,
68
+ "use_relative_action": true,
69
+ "use_vlln": true
70
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/conf.yaml ADDED
@@ -0,0 +1,253 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ load_config_path: null
2
+ model:
3
+ model_type: Gr00tN1d6
4
+ model_dtype: bfloat16
5
+ model_name: nvidia/Eagle-Block2A-2B-v2
6
+ backbone_model_type: eagle
7
+ model_revision: null
8
+ tune_top_llm_layers: 4
9
+ backbone_embedding_dim: 2048
10
+ tune_llm: false
11
+ tune_visual: false
12
+ select_layer: 16
13
+ reproject_vision: false
14
+ use_flash_attention: true
15
+ load_bf16: false
16
+ collator_overwrite_image_inputs: false
17
+ eagle_collator: true
18
+ backbone_trainable_params_fp32: true
19
+ image_crop_size: null
20
+ image_target_size: null
21
+ shortest_image_edge: 256
22
+ crop_fraction: 0.95
23
+ random_rotation_angle: null
24
+ color_jitter_params:
25
+ brightness: 0.3
26
+ contrast: 0.4
27
+ saturation: 0.5
28
+ hue: 0.08
29
+ use_albumentations_transforms: true
30
+ formalize_language: true
31
+ apply_sincos_state_encoding: true
32
+ use_relative_action: true
33
+ max_state_dim: 128
34
+ max_action_dim: 128
35
+ action_horizon: 50
36
+ hidden_size: 1024
37
+ input_embedding_dim: 1536
38
+ add_pos_embed: true
39
+ attn_dropout: 0.2
40
+ use_vlln: true
41
+ max_seq_len: 1024
42
+ use_alternate_vl_dit: true
43
+ attend_text_every_n_blocks: 2
44
+ diffusion_model_cfg:
45
+ attention_head_dim: 48
46
+ dropout: 0.2
47
+ final_dropout: true
48
+ interleave_self_attention: true
49
+ norm_type: ada_norm
50
+ num_attention_heads: 32
51
+ num_layers: 32
52
+ output_dim: 1024
53
+ positional_embeddings: null
54
+ num_inference_timesteps: 4
55
+ noise_beta_alpha: 1.5
56
+ noise_beta_beta: 1.0
57
+ noise_s: 0.999
58
+ num_timestep_buckets: 1000
59
+ tune_projector: true
60
+ tune_diffusion_model: true
61
+ tune_vlln: true
62
+ state_dropout_prob: 0.0
63
+ state_additive_noise_scale: 0.0
64
+ max_num_embodiments: 32
65
+ data:
66
+ datasets:
67
+ - dataset_paths:
68
+ - /hfm/data/real_teleop_g1/lerobot/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
69
+ embodiment_tag: g1_loco_downstream
70
+ mix_ratio: 1.0
71
+ dataset_type: physical_embodiment
72
+ val_dataset_path: null
73
+ modality_configs:
74
+ g1_loco_downstream:
75
+ video:
76
+ delta_indices:
77
+ - 0
78
+ modality_keys:
79
+ - rs_view
80
+ sin_cos_embedding_keys: null
81
+ mean_std_embedding_keys: null
82
+ action_configs: null
83
+ state:
84
+ delta_indices:
85
+ - 0
86
+ modality_keys:
87
+ - left_hand
88
+ - right_hand
89
+ - left_arm
90
+ - right_arm
91
+ - rpy
92
+ - height
93
+ sin_cos_embedding_keys: null
94
+ mean_std_embedding_keys: null
95
+ action_configs: null
96
+ action:
97
+ delta_indices:
98
+ - 0
99
+ - 1
100
+ - 2
101
+ - 3
102
+ - 4
103
+ - 5
104
+ - 6
105
+ - 7
106
+ - 8
107
+ - 9
108
+ - 10
109
+ - 11
110
+ - 12
111
+ - 13
112
+ - 14
113
+ - 15
114
+ modality_keys:
115
+ - left_hand
116
+ - right_hand
117
+ - left_arm
118
+ - right_arm
119
+ - rpy
120
+ - height
121
+ - torso_vx
122
+ - torso_vy
123
+ - torso_vyaw
124
+ - target_yaw
125
+ sin_cos_embedding_keys: null
126
+ mean_std_embedding_keys: null
127
+ action_configs:
128
+ - rep: ABSOLUTE
129
+ type: NON_EEF
130
+ format: DEFAULT
131
+ state_key: null
132
+ - rep: ABSOLUTE
133
+ type: NON_EEF
134
+ format: DEFAULT
135
+ state_key: null
136
+ - rep: ABSOLUTE
137
+ type: NON_EEF
138
+ format: DEFAULT
139
+ state_key: null
140
+ - rep: ABSOLUTE
141
+ type: NON_EEF
142
+ format: DEFAULT
143
+ state_key: null
144
+ - rep: ABSOLUTE
145
+ type: NON_EEF
146
+ format: DEFAULT
147
+ state_key: null
148
+ - rep: ABSOLUTE
149
+ type: NON_EEF
150
+ format: DEFAULT
151
+ state_key: null
152
+ - rep: ABSOLUTE
153
+ type: NON_EEF
154
+ format: DEFAULT
155
+ state_key: null
156
+ - rep: ABSOLUTE
157
+ type: NON_EEF
158
+ format: DEFAULT
159
+ state_key: null
160
+ - rep: ABSOLUTE
161
+ type: NON_EEF
162
+ format: DEFAULT
163
+ state_key: null
164
+ - rep: ABSOLUTE
165
+ type: NON_EEF
166
+ format: DEFAULT
167
+ state_key: null
168
+ language:
169
+ delta_indices:
170
+ - 0
171
+ modality_keys:
172
+ - annotation.human.task_description
173
+ sin_cos_embedding_keys: null
174
+ mean_std_embedding_keys: null
175
+ action_configs: null
176
+ download_cache: false
177
+ shard_size: 1024
178
+ episode_sampling_rate: 0.1
179
+ num_shards_per_epoch: 100000
180
+ override_pretraining_statistics: false
181
+ mode: single_turn
182
+ random_chop: 0.0
183
+ mock_dataset_mode: false
184
+ shuffle: true
185
+ seed: 42
186
+ multiprocessing_context: fork
187
+ allow_padding: false
188
+ subsample_ratio: 1.0
189
+ image_crop_size:
190
+ - 244
191
+ - 244
192
+ image_target_size:
193
+ - 224
194
+ - 224
195
+ video_backend: torchcodec
196
+ training:
197
+ output_dir: /hfm/zhenyu/Psi0/checkpoints/gr00t-real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
198
+ experiment_name: null
199
+ max_steps: 40000
200
+ global_batch_size: 12
201
+ batch_size: null
202
+ gradient_accumulation_steps: 1
203
+ learning_rate: 0.0001
204
+ lr_scheduler_type: cosine
205
+ weight_decay: 1.0e-05
206
+ warmup_ratio: 0.05
207
+ warmup_steps: 0
208
+ max_grad_norm: 1.0
209
+ optim: adamw_torch
210
+ start_from_checkpoint: /hfm/zhenyu/psi/checkpoints/G1WholebodyBendPick-v0-psi0/checkpoint-50000
211
+ reinit_action_head: false
212
+ tf32: true
213
+ fp16: false
214
+ bf16: true
215
+ eval_bf16: true
216
+ logging_steps: 10
217
+ save_steps: 40000
218
+ save_total_limit: 4
219
+ save_vl_model: false
220
+ upload_checkpoints: false
221
+ upload_every: 1000
222
+ upload_last_n_checkpoints: 5
223
+ max_concurrent_uploads: 2
224
+ eval_strategy: 'no'
225
+ eval_steps: 500
226
+ eval_set_split_ratio: 0.1
227
+ eval_batch_size: 2
228
+ save_best_eval_metric_name: ''
229
+ save_best_eval_metric_greater_is_better: true
230
+ deepspeed_stage: 2
231
+ gradient_checkpointing: false
232
+ transformers_trust_remote_code: true
233
+ transformers_local_files_only: false
234
+ transformers_cache_dir: null
235
+ transformers_access_token: null
236
+ use_ddp: false
237
+ ddp_bucket_cap_mb: 100
238
+ num_gpus: 1
239
+ dataloader_num_workers: 4
240
+ remove_unused_columns: false
241
+ use_wandb: false
242
+ wandb_project: finetune-gr00t-n1d6
243
+ enable_profiling: false
244
+ max_retries: 3
245
+ assert_loss_less_than: null
246
+ add_rl_callback: false
247
+ enable_open_loop_eval: false
248
+ open_loop_eval_traj_ids:
249
+ - 0
250
+ open_loop_eval_steps_per_traj: 100
251
+ open_loop_eval_plot_indices: null
252
+ max_steps: 40000
253
+ save_steps: 40000
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/config.yaml ADDED
@@ -0,0 +1,328 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object:gr00t.configs.base_config.Config
2
+ data: !!python/object:gr00t.configs.data.data_config.DataConfig
3
+ allow_padding: false
4
+ datasets:
5
+ - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
6
+ dataset_paths:
7
+ - /hfm/data/real_teleop_g1/lerobot/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
8
+ dataset_type: physical_embodiment
9
+ embodiment_tag: g1_loco_downstream
10
+ mix_ratio: 1.0
11
+ val_dataset_path: null
12
+ download_cache: false
13
+ episode_sampling_rate: 0.1
14
+ image_crop_size:
15
+ - 244
16
+ - 244
17
+ image_target_size:
18
+ - 224
19
+ - 224
20
+ mock_dataset_mode: false
21
+ modality_configs:
22
+ g1_loco_downstream:
23
+ action: !!python/object:gr00t.data.types.ModalityConfig
24
+ action_configs:
25
+ - !!python/object:gr00t.data.types.ActionConfig
26
+ format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
27
+ - default
28
+ rep: &id002 !!python/object/apply:gr00t.data.types.ActionRepresentation
29
+ - absolute
30
+ state_key: null
31
+ type: &id003 !!python/object/apply:gr00t.data.types.ActionType
32
+ - non_eef
33
+ - !!python/object:gr00t.data.types.ActionConfig
34
+ format: *id001
35
+ rep: *id002
36
+ state_key: null
37
+ type: *id003
38
+ - !!python/object:gr00t.data.types.ActionConfig
39
+ format: *id001
40
+ rep: *id002
41
+ state_key: null
42
+ type: *id003
43
+ - !!python/object:gr00t.data.types.ActionConfig
44
+ format: *id001
45
+ rep: *id002
46
+ state_key: null
47
+ type: *id003
48
+ - !!python/object:gr00t.data.types.ActionConfig
49
+ format: *id001
50
+ rep: *id002
51
+ state_key: null
52
+ type: *id003
53
+ - !!python/object:gr00t.data.types.ActionConfig
54
+ format: *id001
55
+ rep: *id002
56
+ state_key: null
57
+ type: *id003
58
+ - !!python/object:gr00t.data.types.ActionConfig
59
+ format: *id001
60
+ rep: *id002
61
+ state_key: null
62
+ type: *id003
63
+ - !!python/object:gr00t.data.types.ActionConfig
64
+ format: *id001
65
+ rep: *id002
66
+ state_key: null
67
+ type: *id003
68
+ - !!python/object:gr00t.data.types.ActionConfig
69
+ format: *id001
70
+ rep: *id002
71
+ state_key: null
72
+ type: *id003
73
+ - !!python/object:gr00t.data.types.ActionConfig
74
+ format: *id001
75
+ rep: *id002
76
+ state_key: null
77
+ type: *id003
78
+ delta_indices:
79
+ - 0
80
+ - 1
81
+ - 2
82
+ - 3
83
+ - 4
84
+ - 5
85
+ - 6
86
+ - 7
87
+ - 8
88
+ - 9
89
+ - 10
90
+ - 11
91
+ - 12
92
+ - 13
93
+ - 14
94
+ - 15
95
+ mean_std_embedding_keys: null
96
+ modality_keys:
97
+ - left_hand
98
+ - right_hand
99
+ - left_arm
100
+ - right_arm
101
+ - rpy
102
+ - height
103
+ - torso_vx
104
+ - torso_vy
105
+ - torso_vyaw
106
+ - target_yaw
107
+ sin_cos_embedding_keys: null
108
+ language: !!python/object:gr00t.data.types.ModalityConfig
109
+ action_configs: null
110
+ delta_indices:
111
+ - 0
112
+ mean_std_embedding_keys: null
113
+ modality_keys:
114
+ - annotation.human.task_description
115
+ sin_cos_embedding_keys: null
116
+ state: !!python/object:gr00t.data.types.ModalityConfig
117
+ action_configs: null
118
+ delta_indices:
119
+ - 0
120
+ mean_std_embedding_keys: null
121
+ modality_keys:
122
+ - left_hand
123
+ - right_hand
124
+ - left_arm
125
+ - right_arm
126
+ - rpy
127
+ - height
128
+ sin_cos_embedding_keys: null
129
+ video: !!python/object:gr00t.data.types.ModalityConfig
130
+ action_configs: null
131
+ delta_indices:
132
+ - 0
133
+ mean_std_embedding_keys: null
134
+ modality_keys:
135
+ - rs_view
136
+ sin_cos_embedding_keys: null
137
+ mode: single_turn
138
+ multiprocessing_context: fork
139
+ num_shards_per_epoch: 100000
140
+ override_pretraining_statistics: false
141
+ random_chop: 0.0
142
+ seed: 42
143
+ shard_size: 1024
144
+ shuffle: true
145
+ subsample_ratio: 1.0
146
+ video_backend: torchcodec
147
+ load_config_path: null
148
+ model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
149
+ _attn_implementation_autoset: false
150
+ _attn_implementation_internal: null
151
+ _commit_hash: null
152
+ _name_or_path: ''
153
+ action_horizon: 50
154
+ add_cross_attention: false
155
+ add_pos_embed: true
156
+ apply_sincos_state_encoding: true
157
+ architectures:
158
+ - Gr00tN1d6
159
+ attn_dropout: 0.2
160
+ attn_implementation: null
161
+ backbone_embedding_dim: 2048
162
+ backbone_model_type: eagle
163
+ backbone_trainable_params_fp32: true
164
+ bad_words_ids: null
165
+ begin_suppress_tokens: null
166
+ bos_token_id: null
167
+ chunk_size_feed_forward: 0
168
+ collator_overwrite_image_inputs: false
169
+ color_jitter_params:
170
+ brightness: 0.3
171
+ contrast: 0.4
172
+ hue: 0.08
173
+ saturation: 0.5
174
+ crop_fraction: 0.95
175
+ cross_attention_hidden_size: null
176
+ decoder_start_token_id: null
177
+ diffusion_model_cfg:
178
+ attention_head_dim: 48
179
+ dropout: 0.2
180
+ final_dropout: true
181
+ interleave_self_attention: true
182
+ norm_type: ada_norm
183
+ num_attention_heads: 32
184
+ num_layers: 32
185
+ output_dim: 1024
186
+ positional_embeddings: null
187
+ diversity_penalty: 0.0
188
+ do_sample: false
189
+ eagle_collator: true
190
+ early_stopping: false
191
+ encoder_no_repeat_ngram_size: 0
192
+ eos_token_id: null
193
+ exponential_decay_length_penalty: null
194
+ finetuning_task: null
195
+ forced_bos_token_id: null
196
+ forced_eos_token_id: null
197
+ formalize_language: true
198
+ gemma_collator: false
199
+ hidden_size: 1024
200
+ id2label:
201
+ 0: LABEL_0
202
+ 1: LABEL_1
203
+ image_crop_size: null
204
+ image_target_size: null
205
+ input_embedding_dim: 1536
206
+ is_decoder: false
207
+ is_encoder_decoder: false
208
+ label2id:
209
+ LABEL_0: 0
210
+ LABEL_1: 1
211
+ length_penalty: 1.0
212
+ load_bf16: false
213
+ max_action_dim: 128
214
+ max_length: 20
215
+ max_num_embodiments: 32
216
+ max_seq_len: 1024
217
+ max_state_dim: 128
218
+ min_length: 0
219
+ model_dtype: bfloat16
220
+ model_name: nvidia/Eagle-Block2A-2B-v2
221
+ model_type: Gr00tN1d6
222
+ no_repeat_ngram_size: 0
223
+ noise_beta_alpha: 1.5
224
+ noise_beta_beta: 1.0
225
+ noise_s: 0.999
226
+ num_beam_groups: 1
227
+ num_beams: 1
228
+ num_inference_timesteps: 4
229
+ num_return_sequences: 1
230
+ num_timestep_buckets: 1000
231
+ output_attentions: false
232
+ output_hidden_states: false
233
+ output_scores: false
234
+ pad_token_id: null
235
+ prefix: null
236
+ problem_type: null
237
+ pruned_heads: {}
238
+ random_rotation_angle: null
239
+ remove_invalid_values: false
240
+ repetition_penalty: 1.0
241
+ reproject_vision: false
242
+ return_dict: true
243
+ return_dict_in_generate: false
244
+ select_layer: 16
245
+ sep_token_id: null
246
+ shortest_image_edge: 256
247
+ state_dropout_prob: 0.0
248
+ suppress_tokens: null
249
+ task_specific_params: null
250
+ temperature: 1.0
251
+ tf_legacy_loss: false
252
+ tie_encoder_decoder: false
253
+ tie_word_embeddings: true
254
+ tokenizer_class: null
255
+ top_k: 50
256
+ top_p: 1.0
257
+ torch_dtype: bfloat16
258
+ torchscript: false
259
+ transformers_version: 4.51.3
260
+ tune_diffusion_model: true
261
+ tune_llm: false
262
+ tune_projector: true
263
+ tune_top_llm_layers: 4
264
+ tune_visual: false
265
+ tune_vlln: true
266
+ typical_p: 1.0
267
+ use_albumentations_transforms: true
268
+ use_alternate_vl_dit: true
269
+ use_bfloat16: false
270
+ use_flash_attention: true
271
+ use_relative_action: true
272
+ use_vlln: true
273
+ training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
274
+ add_rl_callback: false
275
+ assert_loss_less_than: null
276
+ batch_size: null
277
+ bf16: true
278
+ dataloader_num_workers: 4
279
+ ddp_bucket_cap_mb: 100
280
+ deepspeed_stage: 2
281
+ enable_open_loop_eval: false
282
+ enable_profiling: false
283
+ eval_batch_size: 2
284
+ eval_bf16: true
285
+ eval_set_split_ratio: 0.1
286
+ eval_steps: 500
287
+ eval_strategy: 'no'
288
+ experiment_name: null
289
+ fp16: false
290
+ global_batch_size: 12
291
+ gradient_accumulation_steps: 1
292
+ gradient_checkpointing: false
293
+ learning_rate: 0.0001
294
+ logging_steps: 10
295
+ lr_scheduler_type: cosine
296
+ max_concurrent_uploads: 2
297
+ max_grad_norm: 1.0
298
+ max_retries: 3
299
+ max_steps: 40000
300
+ num_gpus: 1
301
+ open_loop_eval_plot_indices: null
302
+ open_loop_eval_steps_per_traj: 100
303
+ open_loop_eval_traj_ids:
304
+ - 0
305
+ optim: adamw_torch
306
+ output_dir: /hfm/zhenyu/Psi0/checkpoints/gr00t-real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water
307
+ reinit_action_head: false
308
+ remove_unused_columns: false
309
+ save_best_eval_metric_greater_is_better: true
310
+ save_best_eval_metric_name: ''
311
+ save_steps: 40000
312
+ save_total_limit: 4
313
+ save_vl_model: false
314
+ start_from_checkpoint: /hfm/zhenyu/psi/checkpoints/G1WholebodyBendPick-v0-psi0/checkpoint-50000
315
+ tf32: true
316
+ transformers_access_token: null
317
+ transformers_cache_dir: null
318
+ transformers_local_files_only: false
319
+ transformers_trust_remote_code: true
320
+ upload_checkpoints: false
321
+ upload_every: 1000
322
+ upload_last_n_checkpoints: 5
323
+ use_ddp: false
324
+ use_wandb: false
325
+ wandb_project: finetune-gr00t-n1d6
326
+ warmup_ratio: 0.05
327
+ warmup_steps: 0
328
+ weight_decay: 1.0e-05
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,641 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "g1_loco_downstream": {
3
+ "state": {
4
+ "left_hand": {
5
+ "min": [
6
+ -1.0061851739883423,
7
+ -0.08936958014965057,
8
+ 0.22508153319358826,
9
+ -1.2385730743408203,
10
+ -1.7122353315353394,
11
+ -1.1028423309326172,
12
+ -1.5751651525497437
13
+ ],
14
+ "max": [
15
+ 0.31211021542549133,
16
+ 0.8900995850563049,
17
+ 1.2004348039627075,
18
+ 0.05436473712325096,
19
+ -0.02029556967318058,
20
+ -0.02401699870824814,
21
+ -0.03348304331302643
22
+ ],
23
+ "mean": [
24
+ -0.524357259273529,
25
+ 0.5116726160049438,
26
+ 0.6721372604370117,
27
+ -0.4349469840526581,
28
+ -0.508317232131958,
29
+ -0.43180689215660095,
30
+ -0.4434279203414917
31
+ ],
32
+ "std": [
33
+ 0.2243678718805313,
34
+ 0.14296627044677734,
35
+ 0.10860829055309296,
36
+ 0.20835691690444946,
37
+ 0.2688620090484619,
38
+ 0.22207914292812347,
39
+ 0.16989512741565704
40
+ ],
41
+ "q01": [
42
+ -0.9798258036375046,
43
+ 0.2255818009376526,
44
+ 0.5424689680337906,
45
+ -0.8970937162637711,
46
+ -1.4855953991413116,
47
+ -1.000885361433029,
48
+ -1.1528545200824738
49
+ ],
50
+ "q99": [
51
+ -0.1169297218322754,
52
+ 0.8758351147174835,
53
+ 1.0207166373729701,
54
+ -0.052023166045546534,
55
+ -0.0410618856549263,
56
+ -0.039927643537521366,
57
+ -0.10575176924467272
58
+ ]
59
+ },
60
+ "right_hand": {
61
+ "min": [
62
+ -0.9382277131080627,
63
+ -0.7008255124092102,
64
+ -0.9635780453681946,
65
+ -0.16482605040073395,
66
+ 0.022227419540286064,
67
+ 0.011465233750641346,
68
+ 0.019190151244401932
69
+ ],
70
+ "max": [
71
+ 0.07547632604837418,
72
+ -0.22075261175632477,
73
+ -0.2170424908399582,
74
+ 1.2214250564575195,
75
+ 1.6621747016906738,
76
+ 0.9483049511909485,
77
+ 1.2975637912750244
78
+ ],
79
+ "mean": [
80
+ -0.5186124444007874,
81
+ -0.41561663150787354,
82
+ -0.6194753646850586,
83
+ 0.3872115910053253,
84
+ 0.4917716979980469,
85
+ 0.4035561978816986,
86
+ 0.3966030776500702
87
+ ],
88
+ "std": [
89
+ 0.13187938928604126,
90
+ 0.07076243311166763,
91
+ 0.09188053011894226,
92
+ 0.25810912251472473,
93
+ 0.2462475299835205,
94
+ 0.2178368717432022,
95
+ 0.16778795421123505
96
+ ],
97
+ "q01": [
98
+ -0.8237687945365906,
99
+ -0.6008797645568847,
100
+ -0.8918748825788498,
101
+ -0.155739439278841,
102
+ 0.032913535833358765,
103
+ 0.024460958316922188,
104
+ 0.019526662304997444
105
+ ],
106
+ "q99": [
107
+ -0.21736681610346664,
108
+ -0.28435189127922056,
109
+ -0.5137390494346619,
110
+ 0.7676283627748489,
111
+ 1.4599360227584839,
112
+ 0.6817445158958435,
113
+ 1.1090852260589599
114
+ ]
115
+ },
116
+ "left_arm": {
117
+ "min": [
118
+ -1.9539560079574585,
119
+ 0.014932344667613506,
120
+ -0.9010579586029053,
121
+ -0.6989918947219849,
122
+ -0.8808764815330505,
123
+ -1.3907837867736816,
124
+ -1.2593215703964233
125
+ ],
126
+ "max": [
127
+ 0.5761176347732544,
128
+ 1.100870966911316,
129
+ 0.7628798484802246,
130
+ 1.3479257822036743,
131
+ 1.4265302419662476,
132
+ 0.6714234352111816,
133
+ 0.6750589609146118
134
+ ],
135
+ "mean": [
136
+ -0.5292406678199768,
137
+ 0.4849226474761963,
138
+ -0.055224090814590454,
139
+ 0.2777264714241028,
140
+ 0.0669676661491394,
141
+ -0.42649492621421814,
142
+ -0.586437463760376
143
+ ],
144
+ "std": [
145
+ 0.676323652267456,
146
+ 0.1855972856283188,
147
+ 0.2814444303512573,
148
+ 0.43967705965042114,
149
+ 0.38776856660842896,
150
+ 0.25116971135139465,
151
+ 0.24483007192611694
152
+ ],
153
+ "q01": [
154
+ -1.6034354269504547,
155
+ 0.15880117490887644,
156
+ -0.6832776308059693,
157
+ -0.49059823155403137,
158
+ -0.6681205630302429,
159
+ -0.943893313407898,
160
+ -1.059616094827652
161
+ ],
162
+ "q99": [
163
+ 0.47410314828157396,
164
+ 0.9649536401033401,
165
+ 0.5035945236682883,
166
+ 1.2618268728256226,
167
+ 0.8312959581613535,
168
+ 0.16624134108424143,
169
+ 0.03442713152617198
170
+ ]
171
+ },
172
+ "right_arm": {
173
+ "min": [
174
+ -1.301702618598938,
175
+ -0.6721951961517334,
176
+ -0.2562706768512726,
177
+ -0.6034896373748779,
178
+ -0.4228993356227875,
179
+ -1.0893871784210205,
180
+ -0.7712242007255554
181
+ ],
182
+ "max": [
183
+ 0.7493496537208557,
184
+ 0.11452125757932663,
185
+ 0.42554786801338196,
186
+ 1.298634648323059,
187
+ 0.48713478446006775,
188
+ 0.6239774227142334,
189
+ 1.0436362028121948
190
+ ],
191
+ "mean": [
192
+ -0.4488898515701294,
193
+ -0.052791301161050797,
194
+ 0.05462970584630966,
195
+ 0.5688164830207825,
196
+ -0.055344775319099426,
197
+ -0.22808769345283508,
198
+ 0.10722843557596207
199
+ ],
200
+ "std": [
201
+ 0.4263588488101959,
202
+ 0.12731978297233582,
203
+ 0.09462971240282059,
204
+ 0.5017045736312866,
205
+ 0.09741094708442688,
206
+ 0.307020902633667,
207
+ 0.205648273229599
208
+ ],
209
+ "q01": [
210
+ -1.0721351623535156,
211
+ -0.49051433801651,
212
+ -0.1922467440366745,
213
+ -0.3724535495042801,
214
+ -0.261208176612854,
215
+ -0.8866654604673385,
216
+ -0.3525379553437233
217
+ ],
218
+ "q99": [
219
+ 0.5084475308656682,
220
+ 0.09847437590360641,
221
+ 0.31654173284769005,
222
+ 1.2630654573440552,
223
+ 0.2419453300535678,
224
+ 0.5242840051651001,
225
+ 0.8154258728027344
226
+ ]
227
+ },
228
+ "rpy": {
229
+ "min": [
230
+ -0.1804429143667221,
231
+ -0.13963909447193146,
232
+ -0.6072064638137817
233
+ ],
234
+ "max": [
235
+ 0.08730942010879517,
236
+ 0.28669121861457825,
237
+ 0.3420403003692627
238
+ ],
239
+ "mean": [
240
+ -0.02583920769393444,
241
+ 0.120397187769413,
242
+ -0.1572193056344986
243
+ ],
244
+ "std": [
245
+ 0.031078113242983818,
246
+ 0.06188545376062393,
247
+ 0.1938641220331192
248
+ ],
249
+ "q01": [
250
+ -0.08460558950901031,
251
+ -0.05298442132771015,
252
+ -0.5683319628238678
253
+ ],
254
+ "q99": [
255
+ 0.059636705368757204,
256
+ 0.23967182561755168,
257
+ 0.1393231652677059
258
+ ]
259
+ },
260
+ "height": {
261
+ "min": [
262
+ 0.6901783347129822
263
+ ],
264
+ "max": [
265
+ 0.839439868927002
266
+ ],
267
+ "mean": [
268
+ 0.7572857737541199
269
+ ],
270
+ "std": [
271
+ 0.016777420416473413
272
+ ],
273
+ "q01": [
274
+ 0.7088072448968887
275
+ ],
276
+ "q99": [
277
+ 0.7952491819858551
278
+ ]
279
+ }
280
+ },
281
+ "action": {
282
+ "left_hand": {
283
+ "min": [
284
+ -1.0481975078582764,
285
+ -0.27768605947494507,
286
+ 0.3844982385635376,
287
+ -1.2728569507598877,
288
+ -1.7463293075561523,
289
+ -1.1422605514526367,
290
+ -1.6586812734603882
291
+ ],
292
+ "max": [
293
+ 0.4100879728794098,
294
+ 0.9210000038146973,
295
+ 1.254960298538208,
296
+ 0.0010000000474974513,
297
+ 0.0010000000474974513,
298
+ 0.0010000000474974513,
299
+ 0.0010000000474974513
300
+ ],
301
+ "mean": [
302
+ -0.5317995548248291,
303
+ 0.5330548882484436,
304
+ 0.6755692362785339,
305
+ -0.43436452746391296,
306
+ -0.5135447382926941,
307
+ -0.4360726773738861,
308
+ -0.44873759150505066
309
+ ],
310
+ "std": [
311
+ 0.23812484741210938,
312
+ 0.16801197826862335,
313
+ 0.13376471400260925,
314
+ 0.23222216963768005,
315
+ 0.29362037777900696,
316
+ 0.2515135407447815,
317
+ 0.19146300852298737
318
+ ],
319
+ "q01": [
320
+ -0.9957153797149658,
321
+ 0.18187148869037628,
322
+ 0.519504725933075,
323
+ -0.9811010956764221,
324
+ -1.5662461996078492,
325
+ -1.0506287038326263,
326
+ -1.223601657152176
327
+ ],
328
+ "q99": [
329
+ -0.12964366376399994,
330
+ 0.9210000038146973,
331
+ 1.0971701681613921,
332
+ 0.0010000000474974513,
333
+ 0.0010000000474974513,
334
+ 0.0010000000474974513,
335
+ -0.08886573091149377
336
+ ]
337
+ },
338
+ "right_hand": {
339
+ "min": [
340
+ -0.9686063528060913,
341
+ -0.7862647771835327,
342
+ -1.034379243850708,
343
+ -0.0010000000474974513,
344
+ -0.0010000000474974513,
345
+ -0.0010000000474974513,
346
+ -0.0010000000474974513
347
+ ],
348
+ "max": [
349
+ 0.11807795614004135,
350
+ -0.13724753260612488,
351
+ -0.4963247776031494,
352
+ 1.2912133932113647,
353
+ 1.7463293075561523,
354
+ 0.9923550486564636,
355
+ 1.6118189096450806
356
+ ],
357
+ "mean": [
358
+ -0.47038841247558594,
359
+ -0.5184828639030457,
360
+ -0.6593118906021118,
361
+ 0.6895503997802734,
362
+ 0.5141050219535828,
363
+ 0.47509223222732544,
364
+ 0.4021771550178528
365
+ ],
366
+ "std": [
367
+ 0.15588711202144623,
368
+ 0.07128255069255829,
369
+ 0.11441590636968613,
370
+ 0.4033379852771759,
371
+ 0.2591473460197449,
372
+ 0.28332775831222534,
373
+ 0.19387704133987427
374
+ ],
375
+ "q01": [
376
+ -0.8572984337806702,
377
+ -0.7032237648963928,
378
+ -0.9383467495441437,
379
+ -0.0010000000474974513,
380
+ -0.0010000000474974513,
381
+ -0.0010000000474974513,
382
+ -0.0010000000474974513
383
+ ],
384
+ "q99": [
385
+ -0.15663587525487035,
386
+ -0.3310527935624125,
387
+ -0.5397384762763977,
388
+ 1.2012066185474395,
389
+ 1.5254341661930078,
390
+ 0.8683193325996399,
391
+ 1.1839898169040537
392
+ ]
393
+ },
394
+ "left_arm": {
395
+ "min": [
396
+ -2.0079522132873535,
397
+ 0.04465832561254501,
398
+ -1.0082837343215942,
399
+ -0.8037082552909851,
400
+ -1.1698691844940186,
401
+ -1.4439691305160522,
402
+ -1.2980564832687378
403
+ ],
404
+ "max": [
405
+ 0.6930765509605408,
406
+ 1.1467127799987793,
407
+ 0.887980043888092,
408
+ 1.3561404943466187,
409
+ 1.6023852825164795,
410
+ 0.7071925401687622,
411
+ 0.8807530403137207
412
+ ],
413
+ "mean": [
414
+ -0.537239134311676,
415
+ 0.49795472621917725,
416
+ -0.06242304667830467,
417
+ 0.27459681034088135,
418
+ 0.0673600509762764,
419
+ -0.43358004093170166,
420
+ -0.5968970060348511
421
+ ],
422
+ "std": [
423
+ 0.6874366998672485,
424
+ 0.18853120505809784,
425
+ 0.2929178476333618,
426
+ 0.45434024930000305,
427
+ 0.4136706292629242,
428
+ 0.2595718801021576,
429
+ 0.24986836314201355
430
+ ],
431
+ "q01": [
432
+ -1.6286808907985688,
433
+ 0.20261594727635385,
434
+ -0.7263113796710968,
435
+ -0.5180197864770889,
436
+ -0.6968675792217255,
437
+ -0.9553359180688858,
438
+ -1.0851860821247101
439
+ ],
440
+ "q99": [
441
+ 0.5095598667860028,
442
+ 0.9890699118375776,
443
+ 0.5474127471446989,
444
+ 1.2788401186466207,
445
+ 0.9281438887119284,
446
+ 0.20716316923499092,
447
+ 0.09342115409672125
448
+ ]
449
+ },
450
+ "right_arm": {
451
+ "min": [
452
+ -1.3225891590118408,
453
+ -0.8259854316711426,
454
+ -0.2885793149471283,
455
+ -0.8615938425064087,
456
+ -0.444199800491333,
457
+ -1.2984219789505005,
458
+ -0.8555077910423279
459
+ ],
460
+ "max": [
461
+ 0.9672761559486389,
462
+ 0.357116162776947,
463
+ 0.4602624475955963,
464
+ 1.3487075567245483,
465
+ 0.729375422000885,
466
+ 0.6513803601264954,
467
+ 1.052833914756775
468
+ ],
469
+ "mean": [
470
+ -0.42227938771247864,
471
+ -0.05299549922347069,
472
+ 0.057760849595069885,
473
+ 0.5489597320556641,
474
+ -0.04834311082959175,
475
+ -0.2310728132724762,
476
+ 0.11325573176145554
477
+ ],
478
+ "std": [
479
+ 0.4346851408481598,
480
+ 0.14826571941375732,
481
+ 0.0984012559056282,
482
+ 0.5546484589576721,
483
+ 0.11210273206233978,
484
+ 0.3245542347431183,
485
+ 0.21472640335559845
486
+ ],
487
+ "q01": [
488
+ -1.0687924444675445,
489
+ -0.4947426915168762,
490
+ -0.18176313862204552,
491
+ -0.5111332565546036,
492
+ -0.2779590994119644,
493
+ -0.9719170898199081,
494
+ -0.399456313252449
495
+ ],
496
+ "q99": [
497
+ 0.6194304704666128,
498
+ 0.21891917586326573,
499
+ 0.3378777891397475,
500
+ 1.3038406193256378,
501
+ 0.28321229368448236,
502
+ 0.5291351854801176,
503
+ 0.8172934919595717
504
+ ]
505
+ },
506
+ "rpy": {
507
+ "min": [
508
+ -0.1804429143667221,
509
+ -0.13963909447193146,
510
+ -0.6072064638137817
511
+ ],
512
+ "max": [
513
+ 0.08730942010879517,
514
+ 0.28669121861457825,
515
+ 0.3420403003692627
516
+ ],
517
+ "mean": [
518
+ -0.025878138840198517,
519
+ 0.1203930452466011,
520
+ -0.15729419887065887
521
+ ],
522
+ "std": [
523
+ 0.031076112762093544,
524
+ 0.061915408819913864,
525
+ 0.19382591545581818
526
+ ],
527
+ "q01": [
528
+ -0.08460558950901031,
529
+ -0.05354986395686865,
530
+ -0.5683319628238678
531
+ ],
532
+ "q99": [
533
+ 0.059636705368757204,
534
+ 0.23967182561755168,
535
+ 0.1393231652677059
536
+ ]
537
+ },
538
+ "height": {
539
+ "min": [
540
+ 0.6901783347129822
541
+ ],
542
+ "max": [
543
+ 0.839439868927002
544
+ ],
545
+ "mean": [
546
+ 0.7572655081748962
547
+ ],
548
+ "std": [
549
+ 0.016800325363874023
550
+ ],
551
+ "q01": [
552
+ 0.7088072448968887
553
+ ],
554
+ "q99": [
555
+ 0.7952491819858551
556
+ ]
557
+ },
558
+ "torso_vx": {
559
+ "min": [
560
+ 0.0
561
+ ],
562
+ "max": [
563
+ 0.0
564
+ ],
565
+ "mean": [
566
+ 0.0
567
+ ],
568
+ "std": [
569
+ 0.0
570
+ ],
571
+ "q01": [
572
+ 0.0
573
+ ],
574
+ "q99": [
575
+ 0.0
576
+ ]
577
+ },
578
+ "torso_vy": {
579
+ "min": [
580
+ 0.0
581
+ ],
582
+ "max": [
583
+ 0.0
584
+ ],
585
+ "mean": [
586
+ 0.0
587
+ ],
588
+ "std": [
589
+ 0.0
590
+ ],
591
+ "q01": [
592
+ 0.0
593
+ ],
594
+ "q99": [
595
+ 0.0
596
+ ]
597
+ },
598
+ "torso_vyaw": {
599
+ "min": [
600
+ 0.0
601
+ ],
602
+ "max": [
603
+ 0.0
604
+ ],
605
+ "mean": [
606
+ 0.0
607
+ ],
608
+ "std": [
609
+ 0.0
610
+ ],
611
+ "q01": [
612
+ 0.0
613
+ ],
614
+ "q99": [
615
+ 0.0
616
+ ]
617
+ },
618
+ "target_yaw": {
619
+ "min": [
620
+ 0.0
621
+ ],
622
+ "max": [
623
+ 0.0
624
+ ],
625
+ "mean": [
626
+ 0.0
627
+ ],
628
+ "std": [
629
+ 0.0
630
+ ],
631
+ "q01": [
632
+ 0.0
633
+ ],
634
+ "q99": [
635
+ 0.0
636
+ ]
637
+ }
638
+ },
639
+ "relative_action": {}
640
+ }
641
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/final_model_config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "Gr00tN1d6",
3
+ "model_dtype": "bfloat16",
4
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
5
+ "backbone_model_type": "eagle",
6
+ "model_revision": null,
7
+ "tune_top_llm_layers": 4,
8
+ "backbone_embedding_dim": 2048,
9
+ "tune_llm": false,
10
+ "tune_visual": false,
11
+ "select_layer": 16,
12
+ "reproject_vision": false,
13
+ "use_flash_attention": true,
14
+ "load_bf16": true,
15
+ "collator_overwrite_image_inputs": false,
16
+ "eagle_collator": true,
17
+ "backbone_trainable_params_fp32": true,
18
+ "apply_sincos_state_encoding": true,
19
+ "use_relative_action": true,
20
+ "max_state_dim": 128,
21
+ "max_action_dim": 128,
22
+ "action_horizon": 50,
23
+ "hidden_size": 1024,
24
+ "input_embedding_dim": 1536,
25
+ "add_pos_embed": true,
26
+ "attn_dropout": 0.2,
27
+ "use_vlln": true,
28
+ "max_seq_len": 1024,
29
+ "use_alternate_vl_dit": true,
30
+ "attend_text_every_n_blocks": 2,
31
+ "diffusion_model_cfg": {
32
+ "attention_head_dim": 48,
33
+ "dropout": 0.2,
34
+ "final_dropout": true,
35
+ "interleave_self_attention": true,
36
+ "norm_type": "ada_norm",
37
+ "num_attention_heads": 32,
38
+ "num_layers": 32,
39
+ "output_dim": 1024,
40
+ "positional_embeddings": null
41
+ },
42
+ "num_inference_timesteps": 4,
43
+ "noise_beta_alpha": 1.5,
44
+ "noise_beta_beta": 1.0,
45
+ "noise_s": 0.999,
46
+ "num_timestep_buckets": 1000,
47
+ "tune_projector": true,
48
+ "tune_diffusion_model": true,
49
+ "tune_vlln": true,
50
+ "state_dropout_prob": 0.0,
51
+ "state_additive_noise_scale": 0.0,
52
+ "max_num_embodiments": 32
53
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18753a5fcdf7d4c4ba71e0ba22571fc4d182795cc7d5e040339ecd054339ebb0
3
+ size 4990120184
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e9cc87647ddbf0961ed9d28ddf98cc592d6f2b94bf4baf3c564d60cf1da5755
3
+ size 4823190320
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/openloop_eval_summary.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_path": "/hfm/data/real_teleop_g1/lerobot/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water",
3
+ "episode_slots": [
4
+ 0,
5
+ 1
6
+ ],
7
+ "evaluated_episodes": [
8
+ 0,
9
+ 1
10
+ ],
11
+ "samples": 2,
12
+ "first_frame_l1": {
13
+ "left_hand": [
14
+ 0.06225765496492386,
15
+ 0.055987387895584106,
16
+ 0.07100054621696472,
17
+ 0.11199969053268433,
18
+ 0.020465150475502014,
19
+ 0.10501638054847717,
20
+ 0.020014509558677673
21
+ ],
22
+ "right_hand": [
23
+ 0.02996090054512024,
24
+ 0.02876916527748108,
25
+ 0.08727657794952393,
26
+ 0.07637888938188553,
27
+ 0.06915086507797241,
28
+ 0.024984657764434814,
29
+ 0.06461960077285767
30
+ ],
31
+ "left_arm": [
32
+ 0.041444290429353714,
33
+ 0.031082943081855774,
34
+ 0.02451770007610321,
35
+ 0.02921357750892639,
36
+ 0.04166471213102341,
37
+ 0.039951205253601074,
38
+ 0.08570513874292374
39
+ ],
40
+ "right_arm": [
41
+ 0.035057730972766876,
42
+ 0.0155758336186409,
43
+ 0.010679751634597778,
44
+ 0.050048828125,
45
+ 0.025948233902454376,
46
+ 0.08881008625030518,
47
+ 0.10962827503681183
48
+ ],
49
+ "rpy": [
50
+ 0.015656711533665657,
51
+ 0.02759091928601265,
52
+ 0.16597026586532593
53
+ ],
54
+ "height": [
55
+ 0.004619598388671875
56
+ ],
57
+ "torso_vx": [
58
+ 0.0
59
+ ],
60
+ "torso_vy": [
61
+ 0.0
62
+ ],
63
+ "torso_vyaw": [
64
+ 0.0
65
+ ],
66
+ "target_yaw": [
67
+ 0.0
68
+ ]
69
+ },
70
+ "full_horizon_l1": {
71
+ "left_hand": 0.046287477016448975,
72
+ "right_hand": 0.057397855179650445,
73
+ "left_arm": 0.04075341458831515,
74
+ "right_arm": 0.049117714166641235,
75
+ "rpy": 0.07359595596790314,
76
+ "height": 0.002772154286503792,
77
+ "torso_vx": 0.0,
78
+ "torso_vy": 0.0,
79
+ "torso_vyaw": 0.0,
80
+ "target_yaw": 0.0
81
+ }
82
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/embodiment_id.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "robocasa_panda_omron": 13,
3
+ "gr1": 20,
4
+ "behavior_r1_pro": 24,
5
+ "unitree_g1": 8,
6
+ "oxe_google": 0,
7
+ "oxe_widowx": 1,
8
+ "libero_panda": 2,
9
+ "new_embodiment": 10,
10
+ "g1_ee_a16": 10,
11
+ "h1_ee_a16": 11,
12
+ "g1_upper_a16": 10,
13
+ "g1_loco_downstream": 10
14
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/processor_config.json ADDED
@@ -0,0 +1,650 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "processor_class": "Gr00tN1d6Processor",
3
+ "processor_kwargs": {
4
+ "modality_configs": {
5
+ "behavior_r1_pro": {
6
+ "video": {
7
+ "delta_indices": [
8
+ 0
9
+ ],
10
+ "modality_keys": [
11
+ "observation.images.rgb.head_256_256",
12
+ "observation.images.rgb.left_wrist_256_256",
13
+ "observation.images.rgb.right_wrist_256_256"
14
+ ],
15
+ "sin_cos_embedding_keys": null,
16
+ "mean_std_embedding_keys": null,
17
+ "action_configs": null
18
+ },
19
+ "state": {
20
+ "delta_indices": [
21
+ 0
22
+ ],
23
+ "modality_keys": [
24
+ "robot_pos",
25
+ "robot_ori_cos",
26
+ "robot_ori_sin",
27
+ "robot_2d_ori",
28
+ "robot_2d_ori_cos",
29
+ "robot_2d_ori_sin",
30
+ "robot_lin_vel",
31
+ "robot_ang_vel",
32
+ "arm_left_qpos",
33
+ "arm_left_qpos_sin",
34
+ "arm_left_qpos_cos",
35
+ "eef_left_pos",
36
+ "eef_left_quat",
37
+ "gripper_left_qpos",
38
+ "arm_right_qpos",
39
+ "arm_right_qpos_sin",
40
+ "arm_right_qpos_cos",
41
+ "eef_right_pos",
42
+ "eef_right_quat",
43
+ "gripper_right_qpos",
44
+ "trunk_qpos"
45
+ ],
46
+ "sin_cos_embedding_keys": null,
47
+ "mean_std_embedding_keys": null,
48
+ "action_configs": null
49
+ },
50
+ "action": {
51
+ "delta_indices": [
52
+ 0,
53
+ 1,
54
+ 2,
55
+ 3,
56
+ 4,
57
+ 5,
58
+ 6,
59
+ 7,
60
+ 8,
61
+ 9,
62
+ 10,
63
+ 11,
64
+ 12,
65
+ 13,
66
+ 14,
67
+ 15,
68
+ 16,
69
+ 17,
70
+ 18,
71
+ 19,
72
+ 20,
73
+ 21,
74
+ 22,
75
+ 23,
76
+ 24,
77
+ 25,
78
+ 26,
79
+ 27,
80
+ 28,
81
+ 29,
82
+ 30,
83
+ 31
84
+ ],
85
+ "modality_keys": [
86
+ "base",
87
+ "torso",
88
+ "left_arm",
89
+ "left_gripper",
90
+ "right_arm",
91
+ "right_gripper"
92
+ ],
93
+ "sin_cos_embedding_keys": null,
94
+ "mean_std_embedding_keys": null,
95
+ "action_configs": [
96
+ {
97
+ "rep": "ABSOLUTE",
98
+ "type": "NON_EEF",
99
+ "format": "DEFAULT",
100
+ "state_key": null
101
+ },
102
+ {
103
+ "rep": "RELATIVE",
104
+ "type": "NON_EEF",
105
+ "format": "DEFAULT",
106
+ "state_key": "trunk_qpos"
107
+ },
108
+ {
109
+ "rep": "RELATIVE",
110
+ "type": "NON_EEF",
111
+ "format": "DEFAULT",
112
+ "state_key": "arm_left_qpos"
113
+ },
114
+ {
115
+ "rep": "ABSOLUTE",
116
+ "type": "NON_EEF",
117
+ "format": "DEFAULT",
118
+ "state_key": null
119
+ },
120
+ {
121
+ "rep": "RELATIVE",
122
+ "type": "NON_EEF",
123
+ "format": "DEFAULT",
124
+ "state_key": "arm_right_qpos"
125
+ },
126
+ {
127
+ "rep": "ABSOLUTE",
128
+ "type": "NON_EEF",
129
+ "format": "DEFAULT",
130
+ "state_key": null
131
+ }
132
+ ]
133
+ },
134
+ "language": {
135
+ "delta_indices": [
136
+ 0
137
+ ],
138
+ "modality_keys": [
139
+ "annotation.human.coarse_action"
140
+ ],
141
+ "sin_cos_embedding_keys": null,
142
+ "mean_std_embedding_keys": null,
143
+ "action_configs": null
144
+ }
145
+ },
146
+ "gr1": {
147
+ "video": {
148
+ "delta_indices": [
149
+ 0
150
+ ],
151
+ "modality_keys": [
152
+ "ego_view_bg_crop_pad_res256_freq20"
153
+ ],
154
+ "sin_cos_embedding_keys": null,
155
+ "mean_std_embedding_keys": null,
156
+ "action_configs": null
157
+ },
158
+ "state": {
159
+ "delta_indices": [
160
+ 0
161
+ ],
162
+ "modality_keys": [
163
+ "left_arm",
164
+ "right_arm",
165
+ "left_hand",
166
+ "right_hand",
167
+ "waist"
168
+ ],
169
+ "sin_cos_embedding_keys": [
170
+ "left_arm",
171
+ "right_arm",
172
+ "left_hand",
173
+ "right_hand",
174
+ "waist"
175
+ ],
176
+ "mean_std_embedding_keys": null,
177
+ "action_configs": null
178
+ },
179
+ "action": {
180
+ "delta_indices": [
181
+ 0,
182
+ 1,
183
+ 2,
184
+ 3,
185
+ 4,
186
+ 5,
187
+ 6,
188
+ 7,
189
+ 8,
190
+ 9,
191
+ 10,
192
+ 11,
193
+ 12,
194
+ 13,
195
+ 14,
196
+ 15
197
+ ],
198
+ "modality_keys": [
199
+ "left_arm",
200
+ "right_arm",
201
+ "left_hand",
202
+ "right_hand",
203
+ "waist"
204
+ ],
205
+ "sin_cos_embedding_keys": null,
206
+ "mean_std_embedding_keys": null,
207
+ "action_configs": [
208
+ {
209
+ "rep": "RELATIVE",
210
+ "type": "NON_EEF",
211
+ "format": "DEFAULT",
212
+ "state_key": null
213
+ },
214
+ {
215
+ "rep": "RELATIVE",
216
+ "type": "NON_EEF",
217
+ "format": "DEFAULT",
218
+ "state_key": null
219
+ },
220
+ {
221
+ "rep": "RELATIVE",
222
+ "type": "NON_EEF",
223
+ "format": "DEFAULT",
224
+ "state_key": null
225
+ },
226
+ {
227
+ "rep": "RELATIVE",
228
+ "type": "NON_EEF",
229
+ "format": "DEFAULT",
230
+ "state_key": null
231
+ },
232
+ {
233
+ "rep": "ABSOLUTE",
234
+ "type": "NON_EEF",
235
+ "format": "DEFAULT",
236
+ "state_key": null
237
+ }
238
+ ]
239
+ },
240
+ "language": {
241
+ "delta_indices": [
242
+ 0
243
+ ],
244
+ "modality_keys": [
245
+ "task"
246
+ ],
247
+ "sin_cos_embedding_keys": null,
248
+ "mean_std_embedding_keys": null,
249
+ "action_configs": null
250
+ }
251
+ },
252
+ "robocasa_panda_omron": {
253
+ "video": {
254
+ "delta_indices": [
255
+ 0
256
+ ],
257
+ "modality_keys": [
258
+ "res256_image_side_0",
259
+ "res256_image_side_1",
260
+ "res256_image_wrist_0"
261
+ ],
262
+ "sin_cos_embedding_keys": null,
263
+ "mean_std_embedding_keys": null,
264
+ "action_configs": null
265
+ },
266
+ "state": {
267
+ "delta_indices": [
268
+ 0
269
+ ],
270
+ "modality_keys": [
271
+ "end_effector_position_relative",
272
+ "end_effector_rotation_relative",
273
+ "gripper_qpos",
274
+ "base_position",
275
+ "base_rotation"
276
+ ],
277
+ "sin_cos_embedding_keys": null,
278
+ "mean_std_embedding_keys": null,
279
+ "action_configs": null
280
+ },
281
+ "action": {
282
+ "delta_indices": [
283
+ 0,
284
+ 1,
285
+ 2,
286
+ 3,
287
+ 4,
288
+ 5,
289
+ 6,
290
+ 7,
291
+ 8,
292
+ 9,
293
+ 10,
294
+ 11,
295
+ 12,
296
+ 13,
297
+ 14,
298
+ 15
299
+ ],
300
+ "modality_keys": [
301
+ "end_effector_position",
302
+ "end_effector_rotation",
303
+ "gripper_close",
304
+ "base_motion",
305
+ "control_mode"
306
+ ],
307
+ "sin_cos_embedding_keys": null,
308
+ "mean_std_embedding_keys": null,
309
+ "action_configs": [
310
+ {
311
+ "rep": "ABSOLUTE",
312
+ "type": "NON_EEF",
313
+ "format": "DEFAULT",
314
+ "state_key": null
315
+ },
316
+ {
317
+ "rep": "ABSOLUTE",
318
+ "type": "NON_EEF",
319
+ "format": "DEFAULT",
320
+ "state_key": null
321
+ },
322
+ {
323
+ "rep": "ABSOLUTE",
324
+ "type": "NON_EEF",
325
+ "format": "DEFAULT",
326
+ "state_key": null
327
+ },
328
+ {
329
+ "rep": "ABSOLUTE",
330
+ "type": "NON_EEF",
331
+ "format": "DEFAULT",
332
+ "state_key": null
333
+ },
334
+ {
335
+ "rep": "ABSOLUTE",
336
+ "type": "NON_EEF",
337
+ "format": "DEFAULT",
338
+ "state_key": null
339
+ }
340
+ ]
341
+ },
342
+ "language": {
343
+ "delta_indices": [
344
+ 0
345
+ ],
346
+ "modality_keys": [
347
+ "annotation.human.action.task_description"
348
+ ],
349
+ "sin_cos_embedding_keys": null,
350
+ "mean_std_embedding_keys": null,
351
+ "action_configs": null
352
+ }
353
+ },
354
+ "new_embodiment": {
355
+ "video": {
356
+ "delta_indices": [
357
+ 0
358
+ ],
359
+ "modality_keys": [
360
+ "rs_view"
361
+ ],
362
+ "sin_cos_embedding_keys": null,
363
+ "mean_std_embedding_keys": null,
364
+ "action_configs": null
365
+ },
366
+ "state": {
367
+ "delta_indices": [
368
+ 0
369
+ ],
370
+ "modality_keys": [
371
+ "left_hand",
372
+ "right_hand",
373
+ "left_arm",
374
+ "right_arm",
375
+ "rpy",
376
+ "height"
377
+ ],
378
+ "sin_cos_embedding_keys": null,
379
+ "mean_std_embedding_keys": null,
380
+ "action_configs": null
381
+ },
382
+ "action": {
383
+ "delta_indices": [
384
+ 0,
385
+ 1,
386
+ 2,
387
+ 3,
388
+ 4,
389
+ 5,
390
+ 6,
391
+ 7,
392
+ 8,
393
+ 9,
394
+ 10,
395
+ 11,
396
+ 12,
397
+ 13,
398
+ 14,
399
+ 15
400
+ ],
401
+ "modality_keys": [
402
+ "left_hand",
403
+ "right_hand",
404
+ "left_arm",
405
+ "right_arm",
406
+ "rpy",
407
+ "height",
408
+ "torso_vx",
409
+ "torso_vy",
410
+ "torso_vyaw",
411
+ "target_yaw"
412
+ ],
413
+ "sin_cos_embedding_keys": null,
414
+ "mean_std_embedding_keys": null,
415
+ "action_configs": [
416
+ {
417
+ "rep": "ABSOLUTE",
418
+ "type": "NON_EEF",
419
+ "format": "DEFAULT",
420
+ "state_key": null
421
+ },
422
+ {
423
+ "rep": "ABSOLUTE",
424
+ "type": "NON_EEF",
425
+ "format": "DEFAULT",
426
+ "state_key": null
427
+ },
428
+ {
429
+ "rep": "ABSOLUTE",
430
+ "type": "NON_EEF",
431
+ "format": "DEFAULT",
432
+ "state_key": null
433
+ },
434
+ {
435
+ "rep": "ABSOLUTE",
436
+ "type": "NON_EEF",
437
+ "format": "DEFAULT",
438
+ "state_key": null
439
+ },
440
+ {
441
+ "rep": "ABSOLUTE",
442
+ "type": "NON_EEF",
443
+ "format": "DEFAULT",
444
+ "state_key": null
445
+ },
446
+ {
447
+ "rep": "ABSOLUTE",
448
+ "type": "NON_EEF",
449
+ "format": "DEFAULT",
450
+ "state_key": null
451
+ },
452
+ {
453
+ "rep": "ABSOLUTE",
454
+ "type": "NON_EEF",
455
+ "format": "DEFAULT",
456
+ "state_key": null
457
+ },
458
+ {
459
+ "rep": "ABSOLUTE",
460
+ "type": "NON_EEF",
461
+ "format": "DEFAULT",
462
+ "state_key": null
463
+ },
464
+ {
465
+ "rep": "ABSOLUTE",
466
+ "type": "NON_EEF",
467
+ "format": "DEFAULT",
468
+ "state_key": null
469
+ },
470
+ {
471
+ "rep": "ABSOLUTE",
472
+ "type": "NON_EEF",
473
+ "format": "DEFAULT",
474
+ "state_key": null
475
+ }
476
+ ]
477
+ },
478
+ "language": {
479
+ "delta_indices": [
480
+ 0
481
+ ],
482
+ "modality_keys": [
483
+ "annotation.human.task_description"
484
+ ],
485
+ "sin_cos_embedding_keys": null,
486
+ "mean_std_embedding_keys": null,
487
+ "action_configs": null
488
+ }
489
+ },
490
+ "g1_loco_downstream": {
491
+ "video": {
492
+ "delta_indices": [
493
+ 0
494
+ ],
495
+ "modality_keys": [
496
+ "rs_view"
497
+ ],
498
+ "sin_cos_embedding_keys": null,
499
+ "mean_std_embedding_keys": null,
500
+ "action_configs": null
501
+ },
502
+ "state": {
503
+ "delta_indices": [
504
+ 0
505
+ ],
506
+ "modality_keys": [
507
+ "left_hand",
508
+ "right_hand",
509
+ "left_arm",
510
+ "right_arm",
511
+ "rpy",
512
+ "height"
513
+ ],
514
+ "sin_cos_embedding_keys": null,
515
+ "mean_std_embedding_keys": null,
516
+ "action_configs": null
517
+ },
518
+ "action": {
519
+ "delta_indices": [
520
+ 0,
521
+ 1,
522
+ 2,
523
+ 3,
524
+ 4,
525
+ 5,
526
+ 6,
527
+ 7,
528
+ 8,
529
+ 9,
530
+ 10,
531
+ 11,
532
+ 12,
533
+ 13,
534
+ 14,
535
+ 15
536
+ ],
537
+ "modality_keys": [
538
+ "left_hand",
539
+ "right_hand",
540
+ "left_arm",
541
+ "right_arm",
542
+ "rpy",
543
+ "height",
544
+ "torso_vx",
545
+ "torso_vy",
546
+ "torso_vyaw",
547
+ "target_yaw"
548
+ ],
549
+ "sin_cos_embedding_keys": null,
550
+ "mean_std_embedding_keys": null,
551
+ "action_configs": [
552
+ {
553
+ "rep": "ABSOLUTE",
554
+ "type": "NON_EEF",
555
+ "format": "DEFAULT",
556
+ "state_key": null
557
+ },
558
+ {
559
+ "rep": "ABSOLUTE",
560
+ "type": "NON_EEF",
561
+ "format": "DEFAULT",
562
+ "state_key": null
563
+ },
564
+ {
565
+ "rep": "ABSOLUTE",
566
+ "type": "NON_EEF",
567
+ "format": "DEFAULT",
568
+ "state_key": null
569
+ },
570
+ {
571
+ "rep": "ABSOLUTE",
572
+ "type": "NON_EEF",
573
+ "format": "DEFAULT",
574
+ "state_key": null
575
+ },
576
+ {
577
+ "rep": "ABSOLUTE",
578
+ "type": "NON_EEF",
579
+ "format": "DEFAULT",
580
+ "state_key": null
581
+ },
582
+ {
583
+ "rep": "ABSOLUTE",
584
+ "type": "NON_EEF",
585
+ "format": "DEFAULT",
586
+ "state_key": null
587
+ },
588
+ {
589
+ "rep": "ABSOLUTE",
590
+ "type": "NON_EEF",
591
+ "format": "DEFAULT",
592
+ "state_key": null
593
+ },
594
+ {
595
+ "rep": "ABSOLUTE",
596
+ "type": "NON_EEF",
597
+ "format": "DEFAULT",
598
+ "state_key": null
599
+ },
600
+ {
601
+ "rep": "ABSOLUTE",
602
+ "type": "NON_EEF",
603
+ "format": "DEFAULT",
604
+ "state_key": null
605
+ },
606
+ {
607
+ "rep": "ABSOLUTE",
608
+ "type": "NON_EEF",
609
+ "format": "DEFAULT",
610
+ "state_key": null
611
+ }
612
+ ]
613
+ },
614
+ "language": {
615
+ "delta_indices": [
616
+ 0
617
+ ],
618
+ "modality_keys": [
619
+ "annotation.human.task_description"
620
+ ],
621
+ "sin_cos_embedding_keys": null,
622
+ "mean_std_embedding_keys": null,
623
+ "action_configs": null
624
+ }
625
+ }
626
+ },
627
+ "image_crop_size": null,
628
+ "image_target_size": null,
629
+ "use_albumentations": true,
630
+ "random_rotation_angle": null,
631
+ "color_jitter_params": {
632
+ "brightness": 0.3,
633
+ "contrast": 0.4,
634
+ "saturation": 0.5,
635
+ "hue": 0.08
636
+ },
637
+ "shortest_image_edge": 256,
638
+ "crop_fraction": 0.95,
639
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
640
+ "model_type": "eagle",
641
+ "formalize_language": true,
642
+ "max_state_dim": 128,
643
+ "max_action_dim": 128,
644
+ "max_action_horizon": 50,
645
+ "use_percentiles": false,
646
+ "clip_outliers": true,
647
+ "apply_sincos_state_encoding": true,
648
+ "use_relative_action": true
649
+ }
650
+ }
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/processor/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea69deca11ea127417d3806567ecd03668e914d426ea2a32a176fdbab73bb048
3
+ size 5905
gr00t-n1.6/real/Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "finetune-gr00t-n1d6", "run_id": "Remove_the_cap_turn_on_the_faucet_and_fill_the_bottle_with_water"}