Safetensors
English
zhenyuzhao commited on
Commit
397063f
·
verified ·
1 Parent(s): 35699cc

Add GR00T real checkpoint for Rotate_to_pour_ham_into_plate_and_push_the_cart_forward

Browse files
Files changed (33) hide show
  1. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/config.json +70 -0
  2. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/embodiment_id.json +14 -0
  3. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/experiment_cfg/conf.yaml +253 -0
  4. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/experiment_cfg/config.yaml +328 -0
  5. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/experiment_cfg/dataset_statistics.json +641 -0
  6. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/experiment_cfg/final_model_config.json +53 -0
  7. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/experiment_cfg/final_processor_config.json +0 -0
  8. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model-00001-of-00002.safetensors +3 -0
  9. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model-00002-of-00002.safetensors +3 -0
  10. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model.safetensors.index.json +0 -0
  11. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/optimizer.pt +3 -0
  12. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/processor_config.json +650 -0
  13. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/rng_state.pth +3 -0
  14. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/scheduler.pt +3 -0
  15. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/statistics.json +0 -0
  16. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/trainer_state.json +0 -0
  17. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/training_args.bin +3 -0
  18. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/wandb_config.json +1 -0
  19. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/config.json +70 -0
  20. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/conf.yaml +253 -0
  21. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/config.yaml +328 -0
  22. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/dataset_statistics.json +641 -0
  23. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/final_model_config.json +53 -0
  24. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/final_processor_config.json +0 -0
  25. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model-00001-of-00002.safetensors +3 -0
  26. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model-00002-of-00002.safetensors +3 -0
  27. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model.safetensors.index.json +0 -0
  28. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/openloop_eval_summary.json +82 -0
  29. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/processor/embodiment_id.json +14 -0
  30. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/processor/processor_config.json +650 -0
  31. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/processor/statistics.json +0 -0
  32. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/training_args.bin +3 -0
  33. gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/wandb_config.json +1 -0
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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
+ -0.9269875884056091,
7
+ -0.06699541956186295,
8
+ 0.4786558449268341,
9
+ -1.4195833206176758,
10
+ -1.7287458181381226,
11
+ -1.184408187866211,
12
+ -1.543308138847351
13
+ ],
14
+ "max": [
15
+ 0.11218374967575073,
16
+ 1.0542062520980835,
17
+ 1.3448737859725952,
18
+ 0.01063089445233345,
19
+ -0.02630593813955784,
20
+ 0.03173275291919708,
21
+ -0.02618894912302494
22
+ ],
23
+ "mean": [
24
+ -0.46002745628356934,
25
+ 0.4747629165649414,
26
+ 0.6948250532150269,
27
+ -0.31353455781936646,
28
+ -0.5853810906410217,
29
+ -0.241254523396492,
30
+ -0.404990017414093
31
+ ],
32
+ "std": [
33
+ 0.16049478948116302,
34
+ 0.13932937383651733,
35
+ 0.10406248271465302,
36
+ 0.314911425113678,
37
+ 0.36967816948890686,
38
+ 0.2794082760810852,
39
+ 0.19444450736045837
40
+ ],
41
+ "q01": [
42
+ -0.8130688619613647,
43
+ 0.19560351073741913,
44
+ 0.546482310295105,
45
+ -1.212935438156128,
46
+ -1.610870542526245,
47
+ -1.0323941123485565,
48
+ -1.2250821590423584
49
+ ],
50
+ "q99": [
51
+ -0.18172204494476318,
52
+ 0.840534197688103,
53
+ 0.9946710854768758,
54
+ -0.01472496685571968,
55
+ -0.07811787776648912,
56
+ -0.002453993773087859,
57
+ -0.08073746308684347
58
+ ]
59
+ },
60
+ "right_hand": {
61
+ "min": [
62
+ -0.9820996522903442,
63
+ -0.7896795272827148,
64
+ -1.0993529558181763,
65
+ -0.20172026753425598,
66
+ 0.030457261949777603,
67
+ 0.006668585818260908,
68
+ 0.01882481388747692
69
+ ],
70
+ "max": [
71
+ 0.017766516655683517,
72
+ -0.1264796108007431,
73
+ -0.49203604459762573,
74
+ 1.328361988067627,
75
+ 1.7149782180786133,
76
+ 1.0285054445266724,
77
+ 1.5147463083267212
78
+ ],
79
+ "mean": [
80
+ -0.4261653423309326,
81
+ -0.4688602685928345,
82
+ -0.6508861184120178,
83
+ 0.39778369665145874,
84
+ 0.6887039542198181,
85
+ 0.39132845401763916,
86
+ 0.4321276545524597
87
+ ],
88
+ "std": [
89
+ 0.16000154614448547,
90
+ 0.10341810435056686,
91
+ 0.07819007337093353,
92
+ 0.3130604922771454,
93
+ 0.38070809841156006,
94
+ 0.26158344745635986,
95
+ 0.25310713052749634
96
+ ],
97
+ "q01": [
98
+ -0.8208091944456101,
99
+ -0.7230183482170105,
100
+ -0.8764568066596985,
101
+ -0.1599019467830658,
102
+ 0.038338125310838224,
103
+ 0.021174823231995106,
104
+ 0.019403893500566483
105
+ ],
106
+ "q99": [
107
+ -0.1260720820724962,
108
+ -0.2484179437160492,
109
+ -0.5271240472793579,
110
+ 0.9174872511625299,
111
+ 1.6465024292469028,
112
+ 0.8363010001182556,
113
+ 1.1525878620147705
114
+ ]
115
+ },
116
+ "left_arm": {
117
+ "min": [
118
+ -1.0591537952423096,
119
+ -0.1634288728237152,
120
+ -0.7029227614402771,
121
+ -1.0480923652648926,
122
+ -0.7582659125328064,
123
+ -1.4715324640274048,
124
+ -0.8460901379585266
125
+ ],
126
+ "max": [
127
+ 1.2089686393737793,
128
+ 0.7988085150718689,
129
+ 0.8867607712745667,
130
+ 1.1790919303894043,
131
+ 1.6713320016860962,
132
+ 0.5832118391990662,
133
+ 0.7345390915870667
134
+ ],
135
+ "mean": [
136
+ 0.5126563310623169,
137
+ 0.2898343503475189,
138
+ -0.038487911224365234,
139
+ -0.41970038414001465,
140
+ 0.35927656292915344,
141
+ -0.3993854224681854,
142
+ -0.20340844988822937
143
+ ],
144
+ "std": [
145
+ 0.4427265524864197,
146
+ 0.12085731327533722,
147
+ 0.2294389307498932,
148
+ 0.36075279116630554,
149
+ 0.6096572875976562,
150
+ 0.4142565429210663,
151
+ 0.3043171465396881
152
+ ],
153
+ "q01": [
154
+ -0.6334736919403077,
155
+ -0.020731991082429888,
156
+ -0.5797120887041092,
157
+ -0.9999154782295228,
158
+ -0.47161510527133943,
159
+ -1.107250144481659,
160
+ -0.6664532256126404
161
+ ],
162
+ "q99": [
163
+ 1.1159828901290894,
164
+ 0.5759617680311203,
165
+ 0.5208223456144333,
166
+ 0.6016402041912099,
167
+ 1.5184007847309116,
168
+ 0.4425380656123162,
169
+ 0.5416640043258667
170
+ ]
171
+ },
172
+ "right_arm": {
173
+ "min": [
174
+ -1.7347166538238525,
175
+ -0.9909635782241821,
176
+ -0.9644664525985718,
177
+ -1.0476369857788086,
178
+ -1.4481498003005981,
179
+ -1.6282670497894287,
180
+ -0.6750589609146118
181
+ ],
182
+ "max": [
183
+ 1.0711021423339844,
184
+ 0.11333481967449188,
185
+ 0.9750365614891052,
186
+ 1.3633613586425781,
187
+ 0.9291010499000549,
188
+ 1.1024490594863892,
189
+ 1.283244013786316
190
+ ],
191
+ "mean": [
192
+ -0.41795027256011963,
193
+ -0.32765018939971924,
194
+ 0.14392685890197754,
195
+ 0.05127895623445511,
196
+ -0.33854445815086365,
197
+ 0.10009922087192535,
198
+ 0.31569716334342957
199
+ ],
200
+ "std": [
201
+ 0.5552160739898682,
202
+ 0.15981826186180115,
203
+ 0.24790969491004944,
204
+ 0.6697200536727905,
205
+ 0.4505113363265991,
206
+ 0.3168523907661438,
207
+ 0.2951470911502838
208
+ ],
209
+ "q01": [
210
+ -1.5834815180301667,
211
+ -0.7632960504293442,
212
+ -0.34250891089439395,
213
+ -0.9945466506481171,
214
+ -1.2838559031486512,
215
+ -0.7027624332904816,
216
+ -0.39147184550762176
217
+ ],
218
+ "q99": [
219
+ 0.7133604252338412,
220
+ 0.05241899937391281,
221
+ 0.7506796699762346,
222
+ 1.3331971168518066,
223
+ 0.5412076115608215,
224
+ 0.8809573137760164,
225
+ 0.9873766976594929
226
+ ]
227
+ },
228
+ "rpy": {
229
+ "min": [
230
+ -0.31443771719932556,
231
+ -0.25234249234199524,
232
+ -0.5012049078941345
233
+ ],
234
+ "max": [
235
+ 0.25545963644981384,
236
+ 0.2277967780828476,
237
+ 0.22437766194343567
238
+ ],
239
+ "mean": [
240
+ -0.034549929201602936,
241
+ 0.013610188849270344,
242
+ -0.024429943412542343
243
+ ],
244
+ "std": [
245
+ 0.03911573067307472,
246
+ 0.06750434637069702,
247
+ 0.08074839413166046
248
+ ],
249
+ "q01": [
250
+ -0.1405536088347435,
251
+ -0.14539323568344117,
252
+ -0.2829143413901329
253
+ ],
254
+ "q99": [
255
+ 0.045323085039854064,
256
+ 0.15672585278749473,
257
+ 0.14398116096854235
258
+ ]
259
+ },
260
+ "height": {
261
+ "min": [
262
+ 0.6416011452674866
263
+ ],
264
+ "max": [
265
+ 0.8164317607879639
266
+ ],
267
+ "mean": [
268
+ 0.7201999425888062
269
+ ],
270
+ "std": [
271
+ 0.014960889704524746
272
+ ],
273
+ "q01": [
274
+ 0.6767415118217468
275
+ ],
276
+ "q99": [
277
+ 0.7662041783332825
278
+ ]
279
+ }
280
+ },
281
+ "action": {
282
+ "left_hand": {
283
+ "min": [
284
+ -1.0409882068634033,
285
+ -0.12273073196411133,
286
+ 0.436837762594223,
287
+ -1.5717962980270386,
288
+ -1.7463293075561523,
289
+ -1.2559611797332764,
290
+ -1.7463293075561523
291
+ ],
292
+ "max": [
293
+ 0.4421468675136566,
294
+ 0.9210000038146973,
295
+ 1.7463293075561523,
296
+ 0.0010000000474974513,
297
+ 0.0010000000474974513,
298
+ 0.0010000000474974513,
299
+ 0.0010000000474974513
300
+ ],
301
+ "mean": [
302
+ -0.4807356894016266,
303
+ 0.47509241104125977,
304
+ 0.6906853914260864,
305
+ -0.31041640043258667,
306
+ -0.5898368954658508,
307
+ -0.23241764307022095,
308
+ -0.40629705786705017
309
+ ],
310
+ "std": [
311
+ 0.1750338226556778,
312
+ 0.15048623085021973,
313
+ 0.13044340908527374,
314
+ 0.33980855345726013,
315
+ 0.38186123967170715,
316
+ 0.28935426473617554,
317
+ 0.2044779658317566
318
+ ],
319
+ "q01": [
320
+ -0.8321770209074021,
321
+ 0.1542169316112995,
322
+ 0.5194920480251313,
323
+ -1.2916747808456421,
324
+ -1.642631198167801,
325
+ -1.0422625398635865,
326
+ -1.2671113669872285
327
+ ],
328
+ "q99": [
329
+ -0.17152927815914154,
330
+ 0.8419997036457066,
331
+ 1.0500152027606964,
332
+ 0.0010000000474974513,
333
+ -0.04017595946788788,
334
+ 0.0010000000474974513,
335
+ -0.07496687382459628
336
+ ]
337
+ },
338
+ "right_hand": {
339
+ "min": [
340
+ -1.0481975078582764,
341
+ -0.9210000038146973,
342
+ -1.7463293075561523,
343
+ -0.0010000000474974513,
344
+ -0.0010000000474974513,
345
+ -0.0010000000474974513,
346
+ -0.0010000000474974513
347
+ ],
348
+ "max": [
349
+ 0.5198602080345154,
350
+ -0.014095980674028397,
351
+ -0.46265918016433716,
352
+ 1.5717962980270386,
353
+ 1.7463293075561523,
354
+ 1.118023157119751,
355
+ 1.7463293075561523
356
+ ],
357
+ "mean": [
358
+ -0.40839895606040955,
359
+ -0.5726506114006042,
360
+ -0.6818944811820984,
361
+ 0.5284730792045593,
362
+ 0.7199811935424805,
363
+ 0.4318573772907257,
364
+ 0.4562909007072449
365
+ ],
366
+ "std": [
367
+ 0.18769894540309906,
368
+ 0.14306479692459106,
369
+ 0.12480109930038452,
370
+ 0.32611697912216187,
371
+ 0.419857919216156,
372
+ 0.3162503242492676,
373
+ 0.3113788366317749
374
+ ],
375
+ "q01": [
376
+ -0.8399935835599899,
377
+ -0.8023136258125305,
378
+ -1.0682287335395813,
379
+ -0.0010000000474974513,
380
+ -0.0010000000474974513,
381
+ -0.0010000000474974513,
382
+ -0.0010000000474974513
383
+ ],
384
+ "q99": [
385
+ -0.09633898876607405,
386
+ -0.24337346106767624,
387
+ -0.5391130346059799,
388
+ 0.9635337591171265,
389
+ 1.6898248541355134,
390
+ 0.9045331478118896,
391
+ 1.411249041557312
392
+ ]
393
+ },
394
+ "left_arm": {
395
+ "min": [
396
+ -1.019361138343811,
397
+ -0.6054571270942688,
398
+ -1.021012783050537,
399
+ -1.0471999645233154,
400
+ -0.7927936315536499,
401
+ -1.560844898223877,
402
+ -0.9046999216079712
403
+ ],
404
+ "max": [
405
+ 1.2334027290344238,
406
+ 1.0112191438674927,
407
+ 0.9427134990692139,
408
+ 1.2165793180465698,
409
+ 1.7361043691635132,
410
+ 0.7488318085670471,
411
+ 0.9007768630981445
412
+ ],
413
+ "mean": [
414
+ 0.5241657495498657,
415
+ 0.2945125997066498,
416
+ -0.04500480368733406,
417
+ -0.42049187421798706,
418
+ 0.36284521222114563,
419
+ -0.40989160537719727,
420
+ -0.21080413460731506
421
+ ],
422
+ "std": [
423
+ 0.43741071224212646,
424
+ 0.12982703745365143,
425
+ 0.23653879761695862,
426
+ 0.3736835718154907,
427
+ 0.618385374546051,
428
+ 0.42224231362342834,
429
+ 0.30724525451660156
430
+ ],
431
+ "q01": [
432
+ -0.5942269682884216,
433
+ -0.07577644497156143,
434
+ -0.6093236482143403,
435
+ -1.0471994876861572,
436
+ -0.4628263860940933,
437
+ -1.1512476015090942,
438
+ -0.6814799612760544
439
+ ],
440
+ "q99": [
441
+ 1.1341198837757112,
442
+ 0.5896527397632599,
443
+ 0.5230552434921265,
444
+ 0.6644547766447099,
445
+ 1.526016002893448,
446
+ 0.4597533589601519,
447
+ 0.5558918887376787
448
+ ]
449
+ },
450
+ "right_arm": {
451
+ "min": [
452
+ -1.7759873867034912,
453
+ -1.0457597970962524,
454
+ -0.9954499006271362,
455
+ -1.0471999645233154,
456
+ -1.4607900381088257,
457
+ -1.614428997039795,
458
+ -0.9024532437324524
459
+ ],
460
+ "max": [
461
+ 1.1621601581573486,
462
+ 0.3022633492946625,
463
+ 1.0151734352111816,
464
+ 1.3677823543548584,
465
+ 0.9460119009017944,
466
+ 1.121222972869873,
467
+ 1.3166650533676147
468
+ ],
469
+ "mean": [
470
+ -0.4022386372089386,
471
+ -0.33494579792022705,
472
+ 0.13704007863998413,
473
+ 0.020859461277723312,
474
+ -0.330166757106781,
475
+ 0.08796475827693939,
476
+ 0.32162994146347046
477
+ ],
478
+ "std": [
479
+ 0.5662075877189636,
480
+ 0.17241664230823517,
481
+ 0.2581793963909149,
482
+ 0.7005768418312073,
483
+ 0.4578431248664856,
484
+ 0.3462810218334198,
485
+ 0.30215367674827576
486
+ ],
487
+ "q01": [
488
+ -1.5976583051681519,
489
+ -0.7911965763568878,
490
+ -0.4213506481051445,
491
+ -1.0471998453140259,
492
+ -1.2858412957191467,
493
+ -0.8454385197162628,
494
+ -0.3983448427915573
495
+ ],
496
+ "q99": [
497
+ 0.7685537481307989,
498
+ 0.09822795458138024,
499
+ 0.764597846865654,
500
+ 1.342433763742447,
501
+ 0.5681095659732828,
502
+ 0.8840116006135945,
503
+ 1.0199516999721532
504
+ ]
505
+ },
506
+ "rpy": {
507
+ "min": [
508
+ -0.31443771719932556,
509
+ -0.25234249234199524,
510
+ -0.5012049078941345
511
+ ],
512
+ "max": [
513
+ 0.25545963644981384,
514
+ 0.2277967780828476,
515
+ 0.22437766194343567
516
+ ],
517
+ "mean": [
518
+ -0.03456060215830803,
519
+ 0.013608204200863838,
520
+ -0.024464966729283333
521
+ ],
522
+ "std": [
523
+ 0.03911809250712395,
524
+ 0.06751924008131027,
525
+ 0.080777607858181
526
+ ],
527
+ "q01": [
528
+ -0.1405536088347435,
529
+ -0.14539323568344117,
530
+ -0.2829238149523735
531
+ ],
532
+ "q99": [
533
+ 0.04534956537187113,
534
+ 0.15672843694686892,
535
+ 0.14398758068680767
536
+ ]
537
+ },
538
+ "height": {
539
+ "min": [
540
+ 0.6416011452674866
541
+ ],
542
+ "max": [
543
+ 0.8164317607879639
544
+ ],
545
+ "mean": [
546
+ 0.7201656103134155
547
+ ],
548
+ "std": [
549
+ 0.014956292696296909
550
+ ],
551
+ "q01": [
552
+ 0.676702984571457
553
+ ],
554
+ "q99": [
555
+ 0.7662041783332825
556
+ ]
557
+ },
558
+ "torso_vx": {
559
+ "min": [
560
+ -0.3499999940395355
561
+ ],
562
+ "max": [
563
+ 0.3499999940395355
564
+ ],
565
+ "mean": [
566
+ 0.03277398273348808
567
+ ],
568
+ "std": [
569
+ 0.10225828737020493
570
+ ],
571
+ "q01": [
572
+ 0.0
573
+ ],
574
+ "q99": [
575
+ 0.3499999940395355
576
+ ]
577
+ },
578
+ "torso_vy": {
579
+ "min": [
580
+ -0.5
581
+ ],
582
+ "max": [
583
+ 0.0
584
+ ],
585
+ "mean": [
586
+ -4.3974599975626916e-05
587
+ ],
588
+ "std": [
589
+ 0.004688670858740807
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.5
604
+ ],
605
+ "mean": [
606
+ 0.035689834505319595
607
+ ],
608
+ "std": [
609
+ 0.12832395732402802
610
+ ],
611
+ "q01": [
612
+ 0.0
613
+ ],
614
+ "q99": [
615
+ 0.5
616
+ ]
617
+ },
618
+ "target_yaw": {
619
+ "min": [
620
+ 0.0
621
+ ],
622
+ "max": [
623
+ 1.9220000505447388
624
+ ],
625
+ "mean": [
626
+ 1.2327402830123901
627
+ ],
628
+ "std": [
629
+ 0.7543802261352539
630
+ ],
631
+ "q01": [
632
+ 0.0
633
+ ],
634
+ "q99": [
635
+ 1.8980000019073486
636
+ ]
637
+ }
638
+ },
639
+ "relative_action": {}
640
+ }
641
+ }
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f7c728386945885fff7b9d3dcd777aa0fdc9b287b0477f38566d673ad4cb842
3
+ size 4990120184
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:122447d7cb264bcac4b2343fbc2a1bf14b0e52b87cac7f5dd44361d0767602ae
3
+ size 4823190320
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5005930eb12baf8c9ab71e0f2b2a1d02002d533781266b7a7422905b6dc0612
3
+ size 12960193762
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b6059bfb47312475171da5c417a76ec376bb39f71ca533e666e64855487450e
3
+ size 5841
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/checkpoint-40000/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "finetune-gr00t-n1d6", "run_id": "Rotate_to_pour_ham_into_plate_and_push_the_cart_forward"}
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward
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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,641 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "g1_loco_downstream": {
3
+ "state": {
4
+ "left_hand": {
5
+ "min": [
6
+ -0.9269875884056091,
7
+ -0.06699541956186295,
8
+ 0.4786558449268341,
9
+ -1.4195833206176758,
10
+ -1.7287458181381226,
11
+ -1.184408187866211,
12
+ -1.543308138847351
13
+ ],
14
+ "max": [
15
+ 0.11218374967575073,
16
+ 1.0542062520980835,
17
+ 1.3448737859725952,
18
+ 0.01063089445233345,
19
+ -0.02630593813955784,
20
+ 0.03173275291919708,
21
+ -0.02618894912302494
22
+ ],
23
+ "mean": [
24
+ -0.46002745628356934,
25
+ 0.4747629165649414,
26
+ 0.6948250532150269,
27
+ -0.31353455781936646,
28
+ -0.5853810906410217,
29
+ -0.241254523396492,
30
+ -0.404990017414093
31
+ ],
32
+ "std": [
33
+ 0.16049478948116302,
34
+ 0.13932937383651733,
35
+ 0.10406248271465302,
36
+ 0.314911425113678,
37
+ 0.36967816948890686,
38
+ 0.2794082760810852,
39
+ 0.19444450736045837
40
+ ],
41
+ "q01": [
42
+ -0.8130688619613647,
43
+ 0.19560351073741913,
44
+ 0.546482310295105,
45
+ -1.212935438156128,
46
+ -1.610870542526245,
47
+ -1.0323941123485565,
48
+ -1.2250821590423584
49
+ ],
50
+ "q99": [
51
+ -0.18172204494476318,
52
+ 0.840534197688103,
53
+ 0.9946710854768758,
54
+ -0.01472496685571968,
55
+ -0.07811787776648912,
56
+ -0.002453993773087859,
57
+ -0.08073746308684347
58
+ ]
59
+ },
60
+ "right_hand": {
61
+ "min": [
62
+ -0.9820996522903442,
63
+ -0.7896795272827148,
64
+ -1.0993529558181763,
65
+ -0.20172026753425598,
66
+ 0.030457261949777603,
67
+ 0.006668585818260908,
68
+ 0.01882481388747692
69
+ ],
70
+ "max": [
71
+ 0.017766516655683517,
72
+ -0.1264796108007431,
73
+ -0.49203604459762573,
74
+ 1.328361988067627,
75
+ 1.7149782180786133,
76
+ 1.0285054445266724,
77
+ 1.5147463083267212
78
+ ],
79
+ "mean": [
80
+ -0.4261653423309326,
81
+ -0.4688602685928345,
82
+ -0.6508861184120178,
83
+ 0.39778369665145874,
84
+ 0.6887039542198181,
85
+ 0.39132845401763916,
86
+ 0.4321276545524597
87
+ ],
88
+ "std": [
89
+ 0.16000154614448547,
90
+ 0.10341810435056686,
91
+ 0.07819007337093353,
92
+ 0.3130604922771454,
93
+ 0.38070809841156006,
94
+ 0.26158344745635986,
95
+ 0.25310713052749634
96
+ ],
97
+ "q01": [
98
+ -0.8208091944456101,
99
+ -0.7230183482170105,
100
+ -0.8764568066596985,
101
+ -0.1599019467830658,
102
+ 0.038338125310838224,
103
+ 0.021174823231995106,
104
+ 0.019403893500566483
105
+ ],
106
+ "q99": [
107
+ -0.1260720820724962,
108
+ -0.2484179437160492,
109
+ -0.5271240472793579,
110
+ 0.9174872511625299,
111
+ 1.6465024292469028,
112
+ 0.8363010001182556,
113
+ 1.1525878620147705
114
+ ]
115
+ },
116
+ "left_arm": {
117
+ "min": [
118
+ -1.0591537952423096,
119
+ -0.1634288728237152,
120
+ -0.7029227614402771,
121
+ -1.0480923652648926,
122
+ -0.7582659125328064,
123
+ -1.4715324640274048,
124
+ -0.8460901379585266
125
+ ],
126
+ "max": [
127
+ 1.2089686393737793,
128
+ 0.7988085150718689,
129
+ 0.8867607712745667,
130
+ 1.1790919303894043,
131
+ 1.6713320016860962,
132
+ 0.5832118391990662,
133
+ 0.7345390915870667
134
+ ],
135
+ "mean": [
136
+ 0.5126563310623169,
137
+ 0.2898343503475189,
138
+ -0.038487911224365234,
139
+ -0.41970038414001465,
140
+ 0.35927656292915344,
141
+ -0.3993854224681854,
142
+ -0.20340844988822937
143
+ ],
144
+ "std": [
145
+ 0.4427265524864197,
146
+ 0.12085731327533722,
147
+ 0.2294389307498932,
148
+ 0.36075279116630554,
149
+ 0.6096572875976562,
150
+ 0.4142565429210663,
151
+ 0.3043171465396881
152
+ ],
153
+ "q01": [
154
+ -0.6334736919403077,
155
+ -0.020731991082429888,
156
+ -0.5797120887041092,
157
+ -0.9999154782295228,
158
+ -0.47161510527133943,
159
+ -1.107250144481659,
160
+ -0.6664532256126404
161
+ ],
162
+ "q99": [
163
+ 1.1159828901290894,
164
+ 0.5759617680311203,
165
+ 0.5208223456144333,
166
+ 0.6016402041912099,
167
+ 1.5184007847309116,
168
+ 0.4425380656123162,
169
+ 0.5416640043258667
170
+ ]
171
+ },
172
+ "right_arm": {
173
+ "min": [
174
+ -1.7347166538238525,
175
+ -0.9909635782241821,
176
+ -0.9644664525985718,
177
+ -1.0476369857788086,
178
+ -1.4481498003005981,
179
+ -1.6282670497894287,
180
+ -0.6750589609146118
181
+ ],
182
+ "max": [
183
+ 1.0711021423339844,
184
+ 0.11333481967449188,
185
+ 0.9750365614891052,
186
+ 1.3633613586425781,
187
+ 0.9291010499000549,
188
+ 1.1024490594863892,
189
+ 1.283244013786316
190
+ ],
191
+ "mean": [
192
+ -0.41795027256011963,
193
+ -0.32765018939971924,
194
+ 0.14392685890197754,
195
+ 0.05127895623445511,
196
+ -0.33854445815086365,
197
+ 0.10009922087192535,
198
+ 0.31569716334342957
199
+ ],
200
+ "std": [
201
+ 0.5552160739898682,
202
+ 0.15981826186180115,
203
+ 0.24790969491004944,
204
+ 0.6697200536727905,
205
+ 0.4505113363265991,
206
+ 0.3168523907661438,
207
+ 0.2951470911502838
208
+ ],
209
+ "q01": [
210
+ -1.5834815180301667,
211
+ -0.7632960504293442,
212
+ -0.34250891089439395,
213
+ -0.9945466506481171,
214
+ -1.2838559031486512,
215
+ -0.7027624332904816,
216
+ -0.39147184550762176
217
+ ],
218
+ "q99": [
219
+ 0.7133604252338412,
220
+ 0.05241899937391281,
221
+ 0.7506796699762346,
222
+ 1.3331971168518066,
223
+ 0.5412076115608215,
224
+ 0.8809573137760164,
225
+ 0.9873766976594929
226
+ ]
227
+ },
228
+ "rpy": {
229
+ "min": [
230
+ -0.31443771719932556,
231
+ -0.25234249234199524,
232
+ -0.5012049078941345
233
+ ],
234
+ "max": [
235
+ 0.25545963644981384,
236
+ 0.2277967780828476,
237
+ 0.22437766194343567
238
+ ],
239
+ "mean": [
240
+ -0.034549929201602936,
241
+ 0.013610188849270344,
242
+ -0.024429943412542343
243
+ ],
244
+ "std": [
245
+ 0.03911573067307472,
246
+ 0.06750434637069702,
247
+ 0.08074839413166046
248
+ ],
249
+ "q01": [
250
+ -0.1405536088347435,
251
+ -0.14539323568344117,
252
+ -0.2829143413901329
253
+ ],
254
+ "q99": [
255
+ 0.045323085039854064,
256
+ 0.15672585278749473,
257
+ 0.14398116096854235
258
+ ]
259
+ },
260
+ "height": {
261
+ "min": [
262
+ 0.6416011452674866
263
+ ],
264
+ "max": [
265
+ 0.8164317607879639
266
+ ],
267
+ "mean": [
268
+ 0.7201999425888062
269
+ ],
270
+ "std": [
271
+ 0.014960889704524746
272
+ ],
273
+ "q01": [
274
+ 0.6767415118217468
275
+ ],
276
+ "q99": [
277
+ 0.7662041783332825
278
+ ]
279
+ }
280
+ },
281
+ "action": {
282
+ "left_hand": {
283
+ "min": [
284
+ -1.0409882068634033,
285
+ -0.12273073196411133,
286
+ 0.436837762594223,
287
+ -1.5717962980270386,
288
+ -1.7463293075561523,
289
+ -1.2559611797332764,
290
+ -1.7463293075561523
291
+ ],
292
+ "max": [
293
+ 0.4421468675136566,
294
+ 0.9210000038146973,
295
+ 1.7463293075561523,
296
+ 0.0010000000474974513,
297
+ 0.0010000000474974513,
298
+ 0.0010000000474974513,
299
+ 0.0010000000474974513
300
+ ],
301
+ "mean": [
302
+ -0.4807356894016266,
303
+ 0.47509241104125977,
304
+ 0.6906853914260864,
305
+ -0.31041640043258667,
306
+ -0.5898368954658508,
307
+ -0.23241764307022095,
308
+ -0.40629705786705017
309
+ ],
310
+ "std": [
311
+ 0.1750338226556778,
312
+ 0.15048623085021973,
313
+ 0.13044340908527374,
314
+ 0.33980855345726013,
315
+ 0.38186123967170715,
316
+ 0.28935426473617554,
317
+ 0.2044779658317566
318
+ ],
319
+ "q01": [
320
+ -0.8321770209074021,
321
+ 0.1542169316112995,
322
+ 0.5194920480251313,
323
+ -1.2916747808456421,
324
+ -1.642631198167801,
325
+ -1.0422625398635865,
326
+ -1.2671113669872285
327
+ ],
328
+ "q99": [
329
+ -0.17152927815914154,
330
+ 0.8419997036457066,
331
+ 1.0500152027606964,
332
+ 0.0010000000474974513,
333
+ -0.04017595946788788,
334
+ 0.0010000000474974513,
335
+ -0.07496687382459628
336
+ ]
337
+ },
338
+ "right_hand": {
339
+ "min": [
340
+ -1.0481975078582764,
341
+ -0.9210000038146973,
342
+ -1.7463293075561523,
343
+ -0.0010000000474974513,
344
+ -0.0010000000474974513,
345
+ -0.0010000000474974513,
346
+ -0.0010000000474974513
347
+ ],
348
+ "max": [
349
+ 0.5198602080345154,
350
+ -0.014095980674028397,
351
+ -0.46265918016433716,
352
+ 1.5717962980270386,
353
+ 1.7463293075561523,
354
+ 1.118023157119751,
355
+ 1.7463293075561523
356
+ ],
357
+ "mean": [
358
+ -0.40839895606040955,
359
+ -0.5726506114006042,
360
+ -0.6818944811820984,
361
+ 0.5284730792045593,
362
+ 0.7199811935424805,
363
+ 0.4318573772907257,
364
+ 0.4562909007072449
365
+ ],
366
+ "std": [
367
+ 0.18769894540309906,
368
+ 0.14306479692459106,
369
+ 0.12480109930038452,
370
+ 0.32611697912216187,
371
+ 0.419857919216156,
372
+ 0.3162503242492676,
373
+ 0.3113788366317749
374
+ ],
375
+ "q01": [
376
+ -0.8399935835599899,
377
+ -0.8023136258125305,
378
+ -1.0682287335395813,
379
+ -0.0010000000474974513,
380
+ -0.0010000000474974513,
381
+ -0.0010000000474974513,
382
+ -0.0010000000474974513
383
+ ],
384
+ "q99": [
385
+ -0.09633898876607405,
386
+ -0.24337346106767624,
387
+ -0.5391130346059799,
388
+ 0.9635337591171265,
389
+ 1.6898248541355134,
390
+ 0.9045331478118896,
391
+ 1.411249041557312
392
+ ]
393
+ },
394
+ "left_arm": {
395
+ "min": [
396
+ -1.019361138343811,
397
+ -0.6054571270942688,
398
+ -1.021012783050537,
399
+ -1.0471999645233154,
400
+ -0.7927936315536499,
401
+ -1.560844898223877,
402
+ -0.9046999216079712
403
+ ],
404
+ "max": [
405
+ 1.2334027290344238,
406
+ 1.0112191438674927,
407
+ 0.9427134990692139,
408
+ 1.2165793180465698,
409
+ 1.7361043691635132,
410
+ 0.7488318085670471,
411
+ 0.9007768630981445
412
+ ],
413
+ "mean": [
414
+ 0.5241657495498657,
415
+ 0.2945125997066498,
416
+ -0.04500480368733406,
417
+ -0.42049187421798706,
418
+ 0.36284521222114563,
419
+ -0.40989160537719727,
420
+ -0.21080413460731506
421
+ ],
422
+ "std": [
423
+ 0.43741071224212646,
424
+ 0.12982703745365143,
425
+ 0.23653879761695862,
426
+ 0.3736835718154907,
427
+ 0.618385374546051,
428
+ 0.42224231362342834,
429
+ 0.30724525451660156
430
+ ],
431
+ "q01": [
432
+ -0.5942269682884216,
433
+ -0.07577644497156143,
434
+ -0.6093236482143403,
435
+ -1.0471994876861572,
436
+ -0.4628263860940933,
437
+ -1.1512476015090942,
438
+ -0.6814799612760544
439
+ ],
440
+ "q99": [
441
+ 1.1341198837757112,
442
+ 0.5896527397632599,
443
+ 0.5230552434921265,
444
+ 0.6644547766447099,
445
+ 1.526016002893448,
446
+ 0.4597533589601519,
447
+ 0.5558918887376787
448
+ ]
449
+ },
450
+ "right_arm": {
451
+ "min": [
452
+ -1.7759873867034912,
453
+ -1.0457597970962524,
454
+ -0.9954499006271362,
455
+ -1.0471999645233154,
456
+ -1.4607900381088257,
457
+ -1.614428997039795,
458
+ -0.9024532437324524
459
+ ],
460
+ "max": [
461
+ 1.1621601581573486,
462
+ 0.3022633492946625,
463
+ 1.0151734352111816,
464
+ 1.3677823543548584,
465
+ 0.9460119009017944,
466
+ 1.121222972869873,
467
+ 1.3166650533676147
468
+ ],
469
+ "mean": [
470
+ -0.4022386372089386,
471
+ -0.33494579792022705,
472
+ 0.13704007863998413,
473
+ 0.020859461277723312,
474
+ -0.330166757106781,
475
+ 0.08796475827693939,
476
+ 0.32162994146347046
477
+ ],
478
+ "std": [
479
+ 0.5662075877189636,
480
+ 0.17241664230823517,
481
+ 0.2581793963909149,
482
+ 0.7005768418312073,
483
+ 0.4578431248664856,
484
+ 0.3462810218334198,
485
+ 0.30215367674827576
486
+ ],
487
+ "q01": [
488
+ -1.5976583051681519,
489
+ -0.7911965763568878,
490
+ -0.4213506481051445,
491
+ -1.0471998453140259,
492
+ -1.2858412957191467,
493
+ -0.8454385197162628,
494
+ -0.3983448427915573
495
+ ],
496
+ "q99": [
497
+ 0.7685537481307989,
498
+ 0.09822795458138024,
499
+ 0.764597846865654,
500
+ 1.342433763742447,
501
+ 0.5681095659732828,
502
+ 0.8840116006135945,
503
+ 1.0199516999721532
504
+ ]
505
+ },
506
+ "rpy": {
507
+ "min": [
508
+ -0.31443771719932556,
509
+ -0.25234249234199524,
510
+ -0.5012049078941345
511
+ ],
512
+ "max": [
513
+ 0.25545963644981384,
514
+ 0.2277967780828476,
515
+ 0.22437766194343567
516
+ ],
517
+ "mean": [
518
+ -0.03456060215830803,
519
+ 0.013608204200863838,
520
+ -0.024464966729283333
521
+ ],
522
+ "std": [
523
+ 0.03911809250712395,
524
+ 0.06751924008131027,
525
+ 0.080777607858181
526
+ ],
527
+ "q01": [
528
+ -0.1405536088347435,
529
+ -0.14539323568344117,
530
+ -0.2829238149523735
531
+ ],
532
+ "q99": [
533
+ 0.04534956537187113,
534
+ 0.15672843694686892,
535
+ 0.14398758068680767
536
+ ]
537
+ },
538
+ "height": {
539
+ "min": [
540
+ 0.6416011452674866
541
+ ],
542
+ "max": [
543
+ 0.8164317607879639
544
+ ],
545
+ "mean": [
546
+ 0.7201656103134155
547
+ ],
548
+ "std": [
549
+ 0.014956292696296909
550
+ ],
551
+ "q01": [
552
+ 0.676702984571457
553
+ ],
554
+ "q99": [
555
+ 0.7662041783332825
556
+ ]
557
+ },
558
+ "torso_vx": {
559
+ "min": [
560
+ -0.3499999940395355
561
+ ],
562
+ "max": [
563
+ 0.3499999940395355
564
+ ],
565
+ "mean": [
566
+ 0.03277398273348808
567
+ ],
568
+ "std": [
569
+ 0.10225828737020493
570
+ ],
571
+ "q01": [
572
+ 0.0
573
+ ],
574
+ "q99": [
575
+ 0.3499999940395355
576
+ ]
577
+ },
578
+ "torso_vy": {
579
+ "min": [
580
+ -0.5
581
+ ],
582
+ "max": [
583
+ 0.0
584
+ ],
585
+ "mean": [
586
+ -4.3974599975626916e-05
587
+ ],
588
+ "std": [
589
+ 0.004688670858740807
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.5
604
+ ],
605
+ "mean": [
606
+ 0.035689834505319595
607
+ ],
608
+ "std": [
609
+ 0.12832395732402802
610
+ ],
611
+ "q01": [
612
+ 0.0
613
+ ],
614
+ "q99": [
615
+ 0.5
616
+ ]
617
+ },
618
+ "target_yaw": {
619
+ "min": [
620
+ 0.0
621
+ ],
622
+ "max": [
623
+ 1.9220000505447388
624
+ ],
625
+ "mean": [
626
+ 1.2327402830123901
627
+ ],
628
+ "std": [
629
+ 0.7543802261352539
630
+ ],
631
+ "q01": [
632
+ 0.0
633
+ ],
634
+ "q99": [
635
+ 1.8980000019073486
636
+ ]
637
+ }
638
+ },
639
+ "relative_action": {}
640
+ }
641
+ }
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f7c728386945885fff7b9d3dcd777aa0fdc9b287b0477f38566d673ad4cb842
3
+ size 4990120184
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:122447d7cb264bcac4b2343fbc2a1bf14b0e52b87cac7f5dd44361d0767602ae
3
+ size 4823190320
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/openloop_eval_summary.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_path": "/hfm/data/real_teleop_g1/lerobot/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward",
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.07034747302532196,
15
+ 0.015403568744659424,
16
+ 0.02102065086364746,
17
+ 0.0,
18
+ 0.02663065493106842,
19
+ 0.0,
20
+ 0.042267702519893646
21
+ ],
22
+ "right_hand": [
23
+ 0.03497052192687988,
24
+ 0.021196693181991577,
25
+ 0.015848100185394287,
26
+ 0.046609461307525635,
27
+ 0.09788846224546432,
28
+ 0.0013869238318875432,
29
+ 0.03425165265798569
30
+ ],
31
+ "left_arm": [
32
+ 0.047714412212371826,
33
+ 0.023432567715644836,
34
+ 0.01398124173283577,
35
+ 0.0059877485036849976,
36
+ 0.043400369584560394,
37
+ 0.0496959425508976,
38
+ 0.05396580696105957
39
+ ],
40
+ "right_arm": [
41
+ 0.014340952038764954,
42
+ 0.04044799506664276,
43
+ 0.0281422957777977,
44
+ 0.03413500264286995,
45
+ 0.06819961220026016,
46
+ 0.08551736176013947,
47
+ 0.018471568822860718
48
+ ],
49
+ "rpy": [
50
+ 0.012937198393046856,
51
+ 0.04779314249753952,
52
+ 0.03802592307329178
53
+ ],
54
+ "height": [
55
+ 0.0057329535484313965
56
+ ],
57
+ "torso_vx": [
58
+ 0.0003417968691792339
59
+ ],
60
+ "torso_vy": [
61
+ 0.00048828125
62
+ ],
63
+ "torso_vyaw": [
64
+ 0.0
65
+ ],
66
+ "target_yaw": [
67
+ 0.003753906348720193
68
+ ]
69
+ },
70
+ "full_horizon_l1": {
71
+ "left_hand": 0.02401271941406386,
72
+ "right_hand": 0.03394014175449099,
73
+ "left_arm": 0.03508638058389936,
74
+ "right_arm": 0.047939143010548184,
75
+ "rpy": 0.03060765688618024,
76
+ "height": 0.007748669013381004,
77
+ "torso_vx": 0.0025674819480627775,
78
+ "torso_vy": 0.00030517578125,
79
+ "torso_vyaw": 9.1552734375e-05,
80
+ "target_yaw": 0.013842529384419322
81
+ }
82
+ }
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/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/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/processor/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b6059bfb47312475171da5c417a76ec376bb39f71ca533e666e64855487450e
3
+ size 5841
gr00t-n1.6/real/Rotate_to_pour_ham_into_plate_and_push_the_cart_forward/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "finetune-gr00t-n1d6", "run_id": "Rotate_to_pour_ham_into_plate_and_push_the_cart_forward"}