Hugging Face's logo

pravsels
/
groot1.7_eyedrops_basket

Robotics
Safetensors
gr00t
Gr00tN1d7
gr00t-n1.7
vla
Model card Files
xet
 Community
pravsels commited on
Commit
c04c3bc
·
verified ·
1 Parent(s): aa2c0b0

Upload GR00T N1.7 checkpoint step 8000

Browse files
Files changed (14) hide show
  1. checkpoints/8000/pretrained_model/config.json +88 -0
  2. checkpoints/8000/pretrained_model/embodiment_id.json +57 -0
  3. checkpoints/8000/pretrained_model/experiment_cfg/conf.yaml +219 -0
  4. checkpoints/8000/pretrained_model/experiment_cfg/config.yaml +217 -0
  5. checkpoints/8000/pretrained_model/experiment_cfg/dataset_statistics.json +824 -0
  6. checkpoints/8000/pretrained_model/experiment_cfg/final_model_config.json +56 -0
  7. checkpoints/8000/pretrained_model/experiment_cfg/final_processor_config.json +0 -0
  8. checkpoints/8000/pretrained_model/model-00001-of-00003.safetensors +3 -0
  9. checkpoints/8000/pretrained_model/model-00002-of-00003.safetensors +3 -0
  10. checkpoints/8000/pretrained_model/model-00003-of-00003.safetensors +3 -0
  11. checkpoints/8000/pretrained_model/model.safetensors.index.json +0 -0
  12. checkpoints/8000/pretrained_model/processor_config.json +1124 -0
  13. checkpoints/8000/pretrained_model/statistics.json +0 -0
  14. checkpoints/8000/pretrained_model/wandb_config.json +1 -0
checkpoints/8000/pretrained_model/config.json ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_horizon": 40,
3
+ "add_pos_embed": true,
4
+ "apply_sincos_state_encoding": false,
5
+ "architectures": [
6
+ "Gr00tN1d7"
7
+ ],
8
+ "attn_dropout": 0.2,
9
+ "attn_implementation": null,
10
+ "backbone_embedding_dim": 2048,
11
+ "backbone_trainable_params_fp32": true,
12
+ "color_jitter_params": {
13
+ "brightness": 0.3,
14
+ "contrast": 0.4,
15
+ "hue": 0.08,
16
+ "saturation": 0.5
17
+ },
18
+ "crop_fraction": 0.95,
19
+ "diffusion_model_cfg": {
20
+ "attention_head_dim": 48,
21
+ "dropout": 0.2,
22
+ "final_dropout": true,
23
+ "interleave_self_attention": true,
24
+ "norm_type": "ada_norm",
25
+ "num_attention_heads": 32,
26
+ "num_layers": 32,
27
+ "output_dim": 1024,
28
+ "positional_embeddings": null
29
+ },
30
+ "dtype": "float32",
31
+ "exclude_state": false,
32
+ "formalize_language": true,
33
+ "hidden_size": 1024,
34
+ "image_crop_size": [
35
+ 230,
36
+ 230
37
+ ],
38
+ "image_target_size": [
39
+ 256,
40
+ 256
41
+ ],
42
+ "letter_box_transform": false,
43
+ "load_bf16": false,
44
+ "max_action_dim": 132,
45
+ "max_num_embodiments": 32,
46
+ "max_seq_len": 1024,
47
+ "max_state_dim": 132,
48
+ "model_dtype": "bfloat16",
49
+ "model_name": "nvidia/Cosmos-Reason2-2B",
50
+ "model_type": "Gr00tN1d7",
51
+ "noise_beta_alpha": 1.5,
52
+ "noise_beta_beta": 1.0,
53
+ "noise_s": 0.999,
54
+ "num_inference_timesteps": 4,
55
+ "num_timestep_buckets": 1000,
56
+ "random_history_crop": true,
57
+ "random_rotation_angle": 0,
58
+ "reproject_vision": false,
59
+ "rtc_ramp_rate": 6.0,
60
+ "select_layer": 16,
61
+ "shortest_image_edge": 256,
62
+ "state_dropout_prob": 0.2,
63
+ "state_gaussian_noise_std": 0.0,
64
+ "transformers_version": "4.57.3",
65
+ "tune_diffusion_model": true,
66
+ "tune_linear": true,
67
+ "tune_llm": false,
68
+ "tune_projector": true,
69
+ "tune_top_llm_layers": 0,
70
+ "tune_visual": false,
71
+ "tune_vlln": true,
72
+ "use_albumentations": true,
73
+ "use_alternate_vl_dit": true,
74
+ "use_flash_attention": true,
75
+ "use_future_tokens": false,
76
+ "use_mean_std": false,
77
+ "use_percentiles": true,
78
+ "use_vl_self_attention": true,
79
+ "use_vlln": true,
80
+ "vl_self_attention_cfg": {
81
+ "attention_head_dim": 64,
82
+ "dropout": 0.2,
83
+ "final_dropout": true,
84
+ "num_attention_heads": 32,
85
+ "num_layers": 4,
86
+ "positional_embeddings": null
87
+ }
88
+ }
checkpoints/8000/pretrained_model/embodiment_id.json ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "robocasa_panda_omron": 13,
3
+ "oxe_droid": 17,
4
+ "oxe_fractal": 18,
5
+ "oxe_language_table": 19,
6
+ "oxe_bridge": 20,
7
+ "unknown": 22,
8
+ "gr1_unified": 20,
9
+ "agibot": 26,
10
+ "sim_behavior_r1_pro": 23,
11
+ "xdof": 24,
12
+ "xdof_oss_data": 25,
13
+ "unitree_g1_full_body_with_waist_height_nav_cmd": 25,
14
+ "real_r1_pro_sharpa": 27,
15
+ "real_r1_pro_sharpa_add_view": 27,
16
+ "real_r1_pro_sharpa_relative_arm_joint": 26,
17
+ "real_r1_pro_sharpa_delta_eef": 26,
18
+ "real_r1_pro_sharpa_absolute_eef": 26,
19
+ "real_r1_pro_sharpa_meanstd": 26,
20
+ "real_r1_pro_sharpa_relative_eef": 26,
21
+ "real_r1_pro_sharpa_relative_eef_add_view": 26,
22
+ "real_r1_pro_sharpa_relative_eef_relative_hand": 26,
23
+ "real_r1_pro_sharpa_relative_eef_human": 26,
24
+ "real_r1_pro_sharpa_relative_eef_human_add_view": 26,
25
+ "real_r1_pro_sharpa_relative_eef_human_relative_hand": 26,
26
+ "real_r1_pro_sharpa_relative_eef_egodex": 26,
27
+ "real_r1_pro_sharpa_relative_eef_egodex_relative_hand": 26,
28
+ "real_r1_pro_sharpa_relative_eef_egodex_wrist_only": 26,
29
+ "real_r1_pro_sharpa_relative_eef_maxinsights": 26,
30
+ "real_r1_pro_sharpa_relative_eef_maxinsights_relative_hand": 26,
31
+ "real_r1_pro_sharpa_relative_eef_mecka": 26,
32
+ "real_r1_pro_sharpa_relative_eef_mecka_relative_hand": 26,
33
+ "real_g1_relative_eef_absolute_joints": 25,
34
+ "real_g1_relative_eef_absolute_joints_wrist_cam": 25,
35
+ "real_g1_relative_eef_relative_joints": 25,
36
+ "real_r1_pro_sharpa_relative_eef_relative_hand_relative_joint": 26,
37
+ "real_r1_pro_sharpa_relative_joint": 29,
38
+ "oxe_droid_relative_eef_relative_joint": 24,
39
+ "oxe_droid_relative_eef_relative_joint_swapped": 24,
40
+ "oxe_droid_relative_eef_relative_joint_upweight_z": 24,
41
+ "oxe_droid_relative_eef_relative_joint_upweight_z_swapped": 24,
42
+ "oxe_droid_relative_eef_relative_joint_3view": 24,
43
+ "oxe_droid_relative_eef_relative_joint_3view_swapped": 24,
44
+ "oxe_droid_relative_eef": 24,
45
+ "oxe_droid_joint_position_relative": 24,
46
+ "xdof_relative_eef_relative_joint": 27,
47
+ "xdof_relative_eef_relative_joint_subtask": 27,
48
+ "xdof_relative_eef": 27,
49
+ "xdof_relative_joint": 28,
50
+ "simpler_env_google": 0,
51
+ "simpler_env_widowx": 1,
52
+ "libero_sim": 2,
53
+ "droid_sim": 3,
54
+ "robocasa_gr1_tabletop": 10,
55
+ "new_embodiment": 10,
56
+ "unitree_g1_sonic": 11
57
+ }
checkpoints/8000/pretrained_model/experiment_cfg/conf.yaml ADDED
@@ -0,0 +1,219 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ load_config_path: null
2
+ model:
3
+ model_type: Gr00tN1d7
4
+ model_dtype: bfloat16
5
+ model_name: nvidia/Cosmos-Reason2-2B
6
+ backbone_model_type: qwen
7
+ model_revision: null
8
+ tune_top_llm_layers: 0
9
+ backbone_embedding_dim: 2048
10
+ tune_llm: false
11
+ tune_visual: false
12
+ select_layer: 12
13
+ reproject_vision: false
14
+ use_flash_attention: true
15
+ load_bf16: false
16
+ backbone_trainable_params_fp32: true
17
+ image_crop_size:
18
+ - 230
19
+ - 230
20
+ image_target_size:
21
+ - 256
22
+ - 256
23
+ shortest_image_edge: null
24
+ crop_fraction: null
25
+ random_rotation_angle: null
26
+ color_jitter_params:
27
+ brightness: 0.3
28
+ contrast: 0.4
29
+ saturation: 0.5
30
+ hue: 0.08
31
+ use_albumentations_transforms: true
32
+ letter_box_transform: false
33
+ extra_augmentation_config: null
34
+ formalize_language: true
35
+ apply_sincos_state_encoding: false
36
+ use_percentiles: true
37
+ use_relative_action: true
38
+ max_state_dim: 132
39
+ max_action_dim: 132
40
+ action_horizon: 40
41
+ hidden_size: 1024
42
+ input_embedding_dim: 1536
43
+ state_history_length: 1
44
+ add_pos_embed: true
45
+ attn_dropout: 0.2
46
+ use_vlln: true
47
+ max_seq_len: 1024
48
+ use_alternate_vl_dit: true
49
+ attend_text_every_n_blocks: 2
50
+ diffusion_model_cfg:
51
+ positional_embeddings: null
52
+ num_layers: 16
53
+ num_attention_heads: 32
54
+ attention_head_dim: 48
55
+ norm_type: ada_norm
56
+ dropout: 0.2
57
+ final_dropout: true
58
+ output_dim: 1024
59
+ interleave_self_attention: true
60
+ num_inference_timesteps: 4
61
+ noise_beta_alpha: 1.5
62
+ noise_beta_beta: 1.0
63
+ noise_s: 0.999
64
+ num_timestep_buckets: 1000
65
+ tune_projector: true
66
+ tune_diffusion_model: true
67
+ tune_vlln: true
68
+ state_dropout_prob: 0.2
69
+ exclude_state: false
70
+ use_mean_std: false
71
+ max_num_embodiments: 32
72
+ data:
73
+ datasets:
74
+ - dataset_paths:
75
+ - /scratch/u6kr/pravsels.u6kr/groot/datasets/eyedrops_basket
76
+ embodiment_tag: new_embodiment
77
+ mix_ratio: 1.0
78
+ dataset_type: physical_embodiment
79
+ val_dataset_path: null
80
+ modality_configs:
81
+ new_embodiment:
82
+ video:
83
+ delta_indices:
84
+ - 0
85
+ modality_keys:
86
+ - front
87
+ - wrist
88
+ - top
89
+ sin_cos_embedding_keys: null
90
+ mean_std_embedding_keys: null
91
+ action_configs: null
92
+ state:
93
+ delta_indices:
94
+ - 0
95
+ modality_keys:
96
+ - single_arm
97
+ - gripper
98
+ sin_cos_embedding_keys: null
99
+ mean_std_embedding_keys: null
100
+ action_configs: null
101
+ action:
102
+ delta_indices:
103
+ - 0
104
+ - 1
105
+ - 2
106
+ - 3
107
+ - 4
108
+ - 5
109
+ - 6
110
+ - 7
111
+ - 8
112
+ - 9
113
+ - 10
114
+ - 11
115
+ - 12
116
+ - 13
117
+ - 14
118
+ - 15
119
+ modality_keys:
120
+ - single_arm
121
+ - gripper
122
+ sin_cos_embedding_keys: null
123
+ mean_std_embedding_keys: null
124
+ action_configs:
125
+ - rep: RELATIVE
126
+ type: NON_EEF
127
+ format: DEFAULT
128
+ state_key: null
129
+ - rep: ABSOLUTE
130
+ type: NON_EEF
131
+ format: DEFAULT
132
+ state_key: null
133
+ language:
134
+ delta_indices:
135
+ - 0
136
+ modality_keys:
137
+ - annotation.human.task_description
138
+ sin_cos_embedding_keys: null
139
+ mean_std_embedding_keys: null
140
+ action_configs: null
141
+ download_cache: false
142
+ shard_size: 1024
143
+ episode_sampling_rate: 0.1
144
+ num_shards_per_epoch: 100000
145
+ override_pretraining_statistics: true
146
+ mode: single_turn
147
+ random_chop: 0.0
148
+ mock_dataset_mode: false
149
+ shuffle: true
150
+ seed: 42
151
+ multiprocessing_context: fork
152
+ allow_padding: false
153
+ subsample_ratio: 1.0
154
+ image_crop_size:
155
+ - 244
156
+ - 244
157
+ image_target_size:
158
+ - 224
159
+ - 224
160
+ training:
161
+ output_dir: /scratch/u6kr/pravsels.u6kr/groot/checkpoints/groot_eyedrops_basket
162
+ experiment_name: groot_eyedrops_basket
163
+ max_steps: 30000
164
+ global_batch_size: 64
165
+ per_gpu_batch_size: null
166
+ gradient_accumulation_steps: 1
167
+ learning_rate: 0.0001
168
+ lr_scheduler_type: cosine
169
+ weight_decay: 1.0e-05
170
+ warmup_ratio: 0.05
171
+ warmup_steps: 0
172
+ max_grad_norm: 1.0
173
+ optim: adamw_torch
174
+ start_from_checkpoint: nvidia/GR00T-N1.7-3B
175
+ skip_weight_loading: false
176
+ tf32: true
177
+ fp16: false
178
+ bf16: true
179
+ eval_bf16: true
180
+ logging_steps: 10
181
+ save_steps: 2000
182
+ save_total_limit: 5
183
+ save_vl_model: false
184
+ save_only_model: false
185
+ resume_from_checkpoint: false
186
+ upload_checkpoints: false
187
+ upload_every: 1000
188
+ upload_last_n_checkpoints: 5
189
+ max_concurrent_uploads: 2
190
+ eval_strategy: 'no'
191
+ eval_steps: 500
192
+ eval_set_split_ratio: 0.1
193
+ eval_batch_size: 2
194
+ save_best_eval_metric_name: ''
195
+ save_best_eval_metric_greater_is_better: true
196
+ deepspeed_stage: 2
197
+ gradient_checkpointing: false
198
+ transformers_trust_remote_code: true
199
+ transformers_local_files_only: false
200
+ transformers_cache_dir: null
201
+ transformers_access_token: null
202
+ use_ddp: false
203
+ ddp_bucket_cap_mb: 100
204
+ num_gpus: 1
205
+ dataloader_num_workers: 8
206
+ remove_unused_columns: false
207
+ use_wandb: true
208
+ wandb_project: groot_eyedrops_basket
209
+ enable_profiling: false
210
+ max_retries: 3
211
+ assert_loss_less_than: null
212
+ add_rl_callback: false
213
+ enable_open_loop_eval: false
214
+ open_loop_eval_traj_ids:
215
+ - 0
216
+ open_loop_eval_steps_per_traj: 100
217
+ open_loop_eval_plot_indices: null
218
+ max_steps: 30000
219
+ save_steps: 2000
checkpoints/8000/pretrained_model/experiment_cfg/config.yaml ADDED
@@ -0,0 +1,217 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ load_config_path: null
2
+ model:
3
+ model_type: Gr00tN1d7
4
+ model_dtype: bfloat16
5
+ model_name: nvidia/Cosmos-Reason2-2B
6
+ backbone_model_type: qwen
7
+ model_revision: null
8
+ tune_top_llm_layers: 0
9
+ backbone_embedding_dim: 2048
10
+ tune_llm: false
11
+ tune_visual: false
12
+ select_layer: 12
13
+ reproject_vision: false
14
+ use_flash_attention: true
15
+ load_bf16: false
16
+ backbone_trainable_params_fp32: true
17
+ image_crop_size:
18
+ - 230
19
+ - 230
20
+ image_target_size:
21
+ - 256
22
+ - 256
23
+ shortest_image_edge: null
24
+ crop_fraction: null
25
+ random_rotation_angle: null
26
+ color_jitter_params:
27
+ brightness: 0.3
28
+ contrast: 0.4
29
+ saturation: 0.5
30
+ hue: 0.08
31
+ use_albumentations_transforms: true
32
+ letter_box_transform: false
33
+ extra_augmentation_config: null
34
+ formalize_language: true
35
+ apply_sincos_state_encoding: false
36
+ use_percentiles: true
37
+ use_relative_action: true
38
+ max_state_dim: 132
39
+ max_action_dim: 132
40
+ action_horizon: 40
41
+ hidden_size: 1024
42
+ input_embedding_dim: 1536
43
+ state_history_length: 1
44
+ add_pos_embed: true
45
+ attn_dropout: 0.2
46
+ use_vlln: true
47
+ max_seq_len: 1024
48
+ use_alternate_vl_dit: true
49
+ attend_text_every_n_blocks: 2
50
+ diffusion_model_cfg:
51
+ positional_embeddings: null
52
+ num_layers: 16
53
+ num_attention_heads: 32
54
+ attention_head_dim: 48
55
+ norm_type: ada_norm
56
+ dropout: 0.2
57
+ final_dropout: true
58
+ output_dim: 1024
59
+ interleave_self_attention: true
60
+ num_inference_timesteps: 4
61
+ noise_beta_alpha: 1.5
62
+ noise_beta_beta: 1.0
63
+ noise_s: 0.999
64
+ num_timestep_buckets: 1000
65
+ tune_projector: true
66
+ tune_diffusion_model: true
67
+ tune_vlln: true
68
+ state_dropout_prob: 0.2
69
+ exclude_state: false
70
+ use_mean_std: false
71
+ max_num_embodiments: 32
72
+ data:
73
+ datasets:
74
+ - dataset_paths:
75
+ - /scratch/u6kr/pravsels.u6kr/groot/datasets/eyedrops_basket
76
+ embodiment_tag: new_embodiment
77
+ mix_ratio: 1.0
78
+ dataset_type: physical_embodiment
79
+ val_dataset_path: null
80
+ modality_configs:
81
+ new_embodiment:
82
+ video:
83
+ delta_indices:
84
+ - 0
85
+ modality_keys:
86
+ - front
87
+ - wrist
88
+ - top
89
+ sin_cos_embedding_keys: null
90
+ mean_std_embedding_keys: null
91
+ action_configs: null
92
+ state:
93
+ delta_indices:
94
+ - 0
95
+ modality_keys:
96
+ - single_arm
97
+ - gripper
98
+ sin_cos_embedding_keys: null
99
+ mean_std_embedding_keys: null
100
+ action_configs: null
101
+ action:
102
+ delta_indices:
103
+ - 0
104
+ - 1
105
+ - 2
106
+ - 3
107
+ - 4
108
+ - 5
109
+ - 6
110
+ - 7
111
+ - 8
112
+ - 9
113
+ - 10
114
+ - 11
115
+ - 12
116
+ - 13
117
+ - 14
118
+ - 15
119
+ modality_keys:
120
+ - single_arm
121
+ - gripper
122
+ sin_cos_embedding_keys: null
123
+ mean_std_embedding_keys: null
124
+ action_configs:
125
+ - rep: relative
126
+ type: non_eef
127
+ format: default
128
+ state_key: null
129
+ - rep: absolute
130
+ type: non_eef
131
+ format: default
132
+ state_key: null
133
+ language:
134
+ delta_indices:
135
+ - 0
136
+ modality_keys:
137
+ - annotation.human.task_description
138
+ sin_cos_embedding_keys: null
139
+ mean_std_embedding_keys: null
140
+ action_configs: null
141
+ download_cache: false
142
+ shard_size: 1024
143
+ episode_sampling_rate: 0.1
144
+ num_shards_per_epoch: 100000
145
+ override_pretraining_statistics: true
146
+ mode: single_turn
147
+ random_chop: 0.0
148
+ mock_dataset_mode: false
149
+ shuffle: true
150
+ seed: 42
151
+ multiprocessing_context: fork
152
+ allow_padding: false
153
+ subsample_ratio: 1.0
154
+ image_crop_size:
155
+ - 244
156
+ - 244
157
+ image_target_size:
158
+ - 224
159
+ - 224
160
+ training:
161
+ output_dir: /scratch/u6kr/pravsels.u6kr/groot/checkpoints/groot_eyedrops_basket
162
+ experiment_name: groot_eyedrops_basket
163
+ max_steps: 30000
164
+ global_batch_size: 64
165
+ per_gpu_batch_size: null
166
+ gradient_accumulation_steps: 1
167
+ learning_rate: 0.0001
168
+ lr_scheduler_type: cosine
169
+ weight_decay: 1.0e-05
170
+ warmup_ratio: 0.05
171
+ warmup_steps: 0
172
+ max_grad_norm: 1.0
173
+ optim: adamw_torch
174
+ start_from_checkpoint: nvidia/GR00T-N1.7-3B
175
+ skip_weight_loading: false
176
+ tf32: true
177
+ fp16: false
178
+ bf16: true
179
+ eval_bf16: true
180
+ logging_steps: 10
181
+ save_steps: 2000
182
+ save_total_limit: 5
183
+ save_vl_model: false
184
+ save_only_model: false
185
+ resume_from_checkpoint: false
186
+ upload_checkpoints: false
187
+ upload_every: 1000
188
+ upload_last_n_checkpoints: 5
189
+ max_concurrent_uploads: 2
190
+ eval_strategy: 'no'
191
+ eval_steps: 500
192
+ eval_set_split_ratio: 0.1
193
+ eval_batch_size: 2
194
+ save_best_eval_metric_name: ''
195
+ save_best_eval_metric_greater_is_better: true
196
+ deepspeed_stage: 2
197
+ gradient_checkpointing: false
198
+ transformers_trust_remote_code: true
199
+ transformers_local_files_only: false
200
+ transformers_cache_dir: null
201
+ transformers_access_token: null
202
+ use_ddp: false
203
+ ddp_bucket_cap_mb: 100
204
+ num_gpus: 1
205
+ dataloader_num_workers: 8
206
+ remove_unused_columns: false
207
+ use_wandb: true
208
+ wandb_project: groot_eyedrops_basket
209
+ enable_profiling: false
210
+ max_retries: 3
211
+ assert_loss_less_than: null
212
+ add_rl_callback: false
213
+ enable_open_loop_eval: false
214
+ open_loop_eval_traj_ids:
215
+ - 0
216
+ open_loop_eval_steps_per_traj: 100
217
+ open_loop_eval_plot_indices: null
checkpoints/8000/pretrained_model/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,824 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "state": {
4
+ "single_arm": {
5
+ "min": [
6
+ -58.06593322753906,
7
+ -103.12088012695312,
8
+ -68.61538696289062,
9
+ 3.604395627975464,
10
+ -55.340660095214844
11
+ ],
12
+ "max": [
13
+ 46.72527313232422,
14
+ 67.42857360839844,
15
+ 107.20879364013672,
16
+ 99.78022003173828,
17
+ 57.36263656616211
18
+ ],
19
+ "mean": [
20
+ -5.283914566040039,
21
+ -39.428462982177734,
22
+ 34.93605041503906,
23
+ 63.42448806762695,
24
+ 4.100034236907959
25
+ ],
26
+ "std": [
27
+ 19.576589584350586,
28
+ 47.46223449707031,
29
+ 48.44475173950195,
30
+ 17.01551628112793,
31
+ 16.847476959228516
32
+ ],
33
+ "q01": [
34
+ -51.09802261352539,
35
+ -103.03296661376953,
36
+ -58.94505310058594,
37
+ 14.509011077880862,
38
+ -44.673406219482416
39
+ ],
40
+ "q99": [
41
+ 31.25274658203125,
42
+ 44.61362586975094,
43
+ 100.32878890991209,
44
+ 91.2298890686035,
45
+ 52.96703338623047
46
+ ]
47
+ },
48
+ "gripper": {
49
+ "min": [
50
+ 0.8032128810882568
51
+ ],
52
+ "max": [
53
+ 48.12583541870117
54
+ ],
55
+ "mean": [
56
+ 12.644012451171875
57
+ ],
58
+ "std": [
59
+ 13.968839645385742
60
+ ],
61
+ "q01": [
62
+ 0.8032128810882568
63
+ ],
64
+ "q99": [
65
+ 46.58634567260742
66
+ ]
67
+ }
68
+ },
69
+ "action": {
70
+ "single_arm": {
71
+ "min": [
72
+ -58.769229888916016,
73
+ -103.69230651855469,
74
+ -68.83516693115234,
75
+ 2.5054945945739746,
76
+ -56.0439567565918
77
+ ],
78
+ "max": [
79
+ 47.07692337036133,
80
+ 68.61538696289062,
81
+ 97.67032623291016,
82
+ 100.43955993652344,
83
+ 58.06593322753906
84
+ ],
85
+ "mean": [
86
+ -5.355493545532227,
87
+ -40.183467864990234,
88
+ 33.538238525390625,
89
+ 63.233455657958984,
90
+ 4.16102409362793
91
+ ],
92
+ "std": [
93
+ 19.616744995117188,
94
+ 47.00802993774414,
95
+ 48.594383239746094,
96
+ 17.292940139770508,
97
+ 16.9429988861084
98
+ ],
99
+ "q01": [
100
+ -51.27384582519531,
101
+ -103.16483306884766,
102
+ -61.53845977783203,
103
+ 13.406593322753906,
104
+ -44.49406646728515
105
+ ],
106
+ "q99": [
107
+ 31.58153789520262,
108
+ 43.801319274902326,
109
+ 97.31867980957031,
110
+ 91.12088012695312,
111
+ 53.425935287475525
112
+ ]
113
+ },
114
+ "gripper": {
115
+ "min": [
116
+ 0.0
117
+ ],
118
+ "max": [
119
+ 48.40510177612305
120
+ ],
121
+ "mean": [
122
+ 10.331722259521484
123
+ ],
124
+ "std": [
125
+ 15.104353904724121
126
+ ],
127
+ "q01": [
128
+ 0.5582137107849121
129
+ ],
130
+ "q99": [
131
+ 46.88995361328125
132
+ ]
133
+ }
134
+ },
135
+ "relative_action": {
136
+ "single_arm": {
137
+ "min": [
138
+ [
139
+ -33.31867980957031,
140
+ -40.659339904785156,
141
+ -46.54945373535156,
142
+ -24.21977996826172,
143
+ -21.80219841003418
144
+ ],
145
+ [
146
+ -35.16483688354492,
147
+ -45.406593322753906,
148
+ -53.05493927001953,
149
+ -28.351646423339844,
150
+ -27.25274658203125
151
+ ],
152
+ [
153
+ -37.36263656616211,
154
+ -49.71428680419922,
155
+ -60.351646423339844,
156
+ -32.131866455078125,
157
+ -32.263736724853516
158
+ ],
159
+ [
160
+ -39.20878982543945,
161
+ -53.758243560791016,
162
+ -67.29669952392578,
163
+ -36.17582702636719,
164
+ -36.0439567565918
165
+ ],
166
+ [
167
+ -40.61538314819336,
168
+ -56.13187026977539,
169
+ -74.24175262451172,
170
+ -40.835166931152344,
171
+ -39.648353576660156
172
+ ],
173
+ [
174
+ -42.28571319580078,
175
+ -59.0329704284668,
176
+ -80.30769348144531,
177
+ -45.23077392578125,
178
+ -43.69230651855469
179
+ ],
180
+ [
181
+ -46.417579650878906,
182
+ -61.4945068359375,
183
+ -84.6153793334961,
184
+ -50.153846740722656,
185
+ -47.47252655029297
186
+ ],
187
+ [
188
+ -50.63736343383789,
189
+ -65.71428680419922,
190
+ -87.34065246582031,
191
+ -54.54945373535156,
192
+ -50.901100158691406
193
+ ],
194
+ [
195
+ -54.769229888916016,
196
+ -69.67033386230469,
197
+ -90.32967376708984,
198
+ -59.03296661376953,
199
+ -53.27472686767578
200
+ ],
201
+ [
202
+ -57.93406677246094,
203
+ -73.62637329101562,
204
+ -94.63735961914062,
205
+ -62.109893798828125,
206
+ -55.20879364013672
207
+ ],
208
+ [
209
+ -60.92308044433594,
210
+ -77.67033386230469,
211
+ -97.45055389404297,
212
+ -64.92308044433594,
213
+ -56.79121017456055
214
+ ],
215
+ [
216
+ -63.38461685180664,
217
+ -81.89010620117188,
218
+ -101.75823974609375,
219
+ -68.0879135131836,
220
+ -58.72527313232422
221
+ ],
222
+ [
223
+ -65.4945068359375,
224
+ -85.75823974609375,
225
+ -104.3956069946289,
226
+ -69.5824203491211,
227
+ -60.131866455078125
228
+ ],
229
+ [
230
+ -67.16484069824219,
231
+ -90.32966613769531,
232
+ -106.76922607421875,
233
+ -70.46154022216797,
234
+ -61.098899841308594
235
+ ],
236
+ [
237
+ -68.4835205078125,
238
+ -94.19779968261719,
239
+ -109.4065933227539,
240
+ -70.989013671875,
241
+ -61.71428680419922
242
+ ],
243
+ [
244
+ -69.71428680419922,
245
+ -97.62637329101562,
246
+ -112.04396057128906,
247
+ -71.25274658203125,
248
+ -62.5054931640625
249
+ ]
250
+ ],
251
+ "max": [
252
+ [
253
+ 34.373626708984375,
254
+ 40.131866455078125,
255
+ 52.70330047607422,
256
+ 21.670330047607422,
257
+ 36.57143020629883
258
+ ],
259
+ [
260
+ 36.65934371948242,
261
+ 47.076927185058594,
262
+ 56.04396057128906,
263
+ 27.208789825439453,
264
+ 41.05494689941406
265
+ ],
266
+ [
267
+ 40.351646423339844,
268
+ 54.19779968261719,
269
+ 60.087913513183594,
270
+ 31.692306518554688,
271
+ 45.362640380859375
272
+ ],
273
+ [
274
+ 42.63736343383789,
275
+ 60.43955993652344,
276
+ 63.78022003173828,
277
+ 36.61538314819336,
278
+ 49.230770111083984
279
+ ],
280
+ [
281
+ 43.95604705810547,
282
+ 65.71428680419922,
283
+ 67.47252655029297,
284
+ 40.65934371948242,
285
+ 52.483516693115234
286
+ ],
287
+ [
288
+ 44.835166931152344,
289
+ 69.93406677246094,
290
+ 70.9011001586914,
291
+ 44.52747344970703,
292
+ 54.85714340209961
293
+ ],
294
+ [
295
+ 45.274723052978516,
296
+ 72.4835205078125,
297
+ 74.24176025390625,
298
+ 47.164833068847656,
299
+ 56.879119873046875
300
+ ],
301
+ [
302
+ 48.0,
303
+ 75.38461303710938,
304
+ 77.4065933227539,
305
+ 49.80219268798828,
306
+ 57.846153259277344
307
+ ],
308
+ [
309
+ 50.46154022216797,
310
+ 79.60440063476562,
311
+ 82.15384674072266,
312
+ 52.879119873046875,
313
+ 58.72527313232422
314
+ ],
315
+ [
316
+ 52.659339904785156,
317
+ 82.15385437011719,
318
+ 87.51648712158203,
319
+ 56.21977996826172,
320
+ 59.340660095214844
321
+ ],
322
+ [
323
+ 54.85714340209961,
324
+ 85.5824203491211,
325
+ 91.64835357666016,
326
+ 59.47252655029297,
327
+ 59.340660095214844
328
+ ],
329
+ [
330
+ 56.43955993652344,
331
+ 88.13187408447266,
332
+ 94.81318664550781,
333
+ 63.5164794921875,
334
+ 59.340660095214844
335
+ ],
336
+ [
337
+ 57.846153259277344,
338
+ 89.8901138305664,
339
+ 98.85714721679688,
340
+ 65.53845977783203,
341
+ 59.340660095214844
342
+ ],
343
+ [
344
+ 59.07692337036133,
345
+ 92.17582702636719,
346
+ 102.02198028564453,
347
+ 67.9120864868164,
348
+ 59.340660095214844
349
+ ],
350
+ [
351
+ 60.13187026977539,
352
+ 94.37362670898438,
353
+ 104.65933990478516,
354
+ 68.87911987304688,
355
+ 59.340660095214844
356
+ ],
357
+ [
358
+ 60.92307662963867,
359
+ 96.13187408447266,
360
+ 107.64835357666016,
361
+ 68.96703338623047,
362
+ 59.340660095214844
363
+ ]
364
+ ],
365
+ "mean": [
366
+ [
367
+ -0.08280117064714432,
368
+ -0.8370546698570251,
369
+ -1.5024255514144897,
370
+ -0.25165319442749023,
371
+ 0.05799264460802078
372
+ ],
373
+ [
374
+ -0.08147875964641571,
375
+ -0.8376094698905945,
376
+ -1.502388834953308,
377
+ -0.25205153226852417,
378
+ 0.05584130436182022
379
+ ],
380
+ [
381
+ -0.07888694107532501,
382
+ -0.8380405306816101,
383
+ -1.5023177862167358,
384
+ -0.2521041929721832,
385
+ 0.05493888258934021
386
+ ],
387
+ [
388
+ -0.07530812174081802,
389
+ -0.8384293913841248,
390
+ -1.5021636486053467,
391
+ -0.2518942058086395,
392
+ 0.05465531349182129
393
+ ],
394
+ [
395
+ -0.07038646936416626,
396
+ -0.8387541174888611,
397
+ -1.501592993736267,
398
+ -0.25158926844596863,
399
+ 0.05470782518386841
400
+ ],
401
+ [
402
+ -0.06493968516588211,
403
+ -0.8392683863639832,
404
+ -1.5000920295715332,
405
+ -0.2515261769294739,
406
+ 0.05515897274017334
407
+ ],
408
+ [
409
+ -0.05995477735996246,
410
+ -0.8409574627876282,
411
+ -1.4969127178192139,
412
+ -0.25120115280151367,
413
+ 0.05572562664747238
414
+ ],
415
+ [
416
+ -0.05445561185479164,
417
+ -0.8456594347953796,
418
+ -1.4900087118148804,
419
+ -0.25032997131347656,
420
+ 0.05639733746647835
421
+ ],
422
+ [
423
+ -0.04917704686522484,
424
+ -0.8545376658439636,
425
+ -1.478272795677185,
426
+ -0.24875588715076447,
427
+ 0.05719497427344322
428
+ ],
429
+ [
430
+ -0.04355209693312645,
431
+ -0.8691246509552002,
432
+ -1.458924412727356,
433
+ -0.24626904726028442,
434
+ 0.05797160789370537
435
+ ],
436
+ [
437
+ -0.038346823304891586,
438
+ -0.8912784457206726,
439
+ -1.4296759366989136,
440
+ -0.24284760653972626,
441
+ 0.058548688888549805
442
+ ],
443
+ [
444
+ -0.03479977324604988,
445
+ -0.9246606826782227,
446
+ -1.3863885402679443,
447
+ -0.238030806183815,
448
+ 0.05913636460900307
449
+ ],
450
+ [
451
+ -0.03337237611413002,
452
+ -0.9721370935440063,
453
+ -1.3266538381576538,
454
+ -0.2314509153366089,
455
+ 0.059619057923555374
456
+ ],
457
+ [
458
+ -0.03347723186016083,
459
+ -1.0365015268325806,
460
+ -1.2476520538330078,
461
+ -0.22236257791519165,
462
+ 0.05978715792298317
463
+ ],
464
+ [
465
+ -0.035450153052806854,
466
+ -1.1201618909835815,
467
+ -1.1470521688461304,
468
+ -0.21024177968502045,
469
+ 0.059650592505931854
470
+ ],
471
+ [
472
+ -0.03953241929411888,
473
+ -1.2246440649032593,
474
+ -1.0228911638259888,
475
+ -0.19504547119140625,
476
+ 0.06003895401954651
477
+ ]
478
+ ],
479
+ "std": [
480
+ [
481
+ 4.4369001388549805,
482
+ 8.116925239562988,
483
+ 8.529901504516602,
484
+ 4.724339008331299,
485
+ 3.2984979152679443
486
+ ],
487
+ [
488
+ 5.670312404632568,
489
+ 10.43100643157959,
490
+ 10.89982795715332,
491
+ 6.067131042480469,
492
+ 4.200020790100098
493
+ ],
494
+ [
495
+ 6.877181053161621,
496
+ 12.709939956665039,
497
+ 13.234122276306152,
498
+ 7.341812610626221,
499
+ 5.063300609588623
500
+ ],
501
+ [
502
+ 8.04920482635498,
503
+ 14.947751998901367,
504
+ 15.524109840393066,
505
+ 8.548194885253906,
506
+ 5.889414310455322
507
+ ],
508
+ [
509
+ 9.181419372558594,
510
+ 17.14101219177246,
511
+ 17.764511108398438,
512
+ 9.68813419342041,
513
+ 6.680186748504639
514
+ ],
515
+ [
516
+ 10.272740364074707,
517
+ 19.28843879699707,
518
+ 19.95174217224121,
519
+ 10.765113830566406,
520
+ 7.437430381774902
521
+ ],
522
+ [
523
+ 11.32327651977539,
524
+ 21.388935089111328,
525
+ 22.083906173706055,
526
+ 11.783564567565918,
527
+ 8.162642478942871
528
+ ],
529
+ [
530
+ 12.331792831420898,
531
+ 23.441679000854492,
532
+ 24.15879249572754,
533
+ 12.74799633026123,
534
+ 8.857882499694824
535
+ ],
536
+ [
537
+ 13.299581527709961,
538
+ 25.44676399230957,
539
+ 26.176504135131836,
540
+ 13.661735534667969,
541
+ 9.52495002746582
542
+ ],
543
+ [
544
+ 14.22679615020752,
545
+ 27.403940200805664,
546
+ 28.136343002319336,
547
+ 14.52731990814209,
548
+ 10.165623664855957
549
+ ],
550
+ [
551
+ 15.115191459655762,
552
+ 29.312644958496094,
553
+ 30.038375854492188,
554
+ 15.34621524810791,
555
+ 10.781367301940918
556
+ ],
557
+ [
558
+ 15.966975212097168,
559
+ 31.172513961791992,
560
+ 31.88163185119629,
561
+ 16.11883544921875,
562
+ 11.37424087524414
563
+ ],
564
+ [
565
+ 16.784847259521484,
566
+ 32.98200225830078,
567
+ 33.66566848754883,
568
+ 16.846220016479492,
569
+ 11.945833206176758
570
+ ],
571
+ [
572
+ 17.570199966430664,
573
+ 34.74038314819336,
574
+ 35.39009094238281,
575
+ 17.52888298034668,
576
+ 12.4974946975708
577
+ ],
578
+ [
579
+ 18.324724197387695,
580
+ 36.44675064086914,
581
+ 37.05491256713867,
582
+ 18.167613983154297,
583
+ 13.030975341796875
584
+ ],
585
+ [
586
+ 19.050418853759766,
587
+ 38.09906005859375,
588
+ 38.65940475463867,
589
+ 18.763816833496094,
590
+ 13.547266960144043
591
+ ]
592
+ ],
593
+ "q01": [
594
+ [
595
+ -13.8021976852417,
596
+ -20.0,
597
+ -20.791207275390626,
598
+ -14.239998168945311,
599
+ -9.5824169921875
600
+ ],
601
+ [
602
+ -17.691428833007812,
603
+ -25.098899841308594,
604
+ -26.08703155517578,
605
+ -18.15384719848633,
606
+ -12.307692527770996
607
+ ],
608
+ [
609
+ -21.538460235595704,
610
+ -30.349009552001952,
611
+ -31.42856689453125,
612
+ -21.62461601257324,
613
+ -14.702417793273925
614
+ ],
615
+ [
616
+ -24.96703338623047,
617
+ -35.38461395263672,
618
+ -36.460662078857425,
619
+ -24.6558243560791,
620
+ -17.318681716918945
621
+ ],
622
+ [
623
+ -28.349891052246093,
624
+ -40.10725357055664,
625
+ -41.47164749145508,
626
+ -27.47252700805664,
627
+ -19.340660095214844
628
+ ],
629
+ [
630
+ -31.669449462890626,
631
+ -44.92132034301758,
632
+ -46.591650390625,
633
+ -30.28571029663086,
634
+ -21.91120880126953
635
+ ],
636
+ [
637
+ -34.30681243896484,
638
+ -49.20791259765625,
639
+ -51.49362487792969,
640
+ -31.97714012145996,
641
+ -23.669450454711914
642
+ ],
643
+ [
644
+ -37.31692443847656,
645
+ -53.31867980957031,
646
+ -56.152965698242184,
647
+ -33.823298645019534,
648
+ -25.494504928588867
649
+ ],
650
+ [
651
+ -39.82417678833008,
652
+ -57.362637481689454,
653
+ -60.81230850219727,
654
+ -35.40571517944336,
655
+ -26.327911224365234
656
+ ],
657
+ [
658
+ -41.95516510009766,
659
+ -60.81231140136719,
660
+ -65.20791259765625,
661
+ -37.05494400024414,
662
+ -27.71340576171875
663
+ ],
664
+ [
665
+ -43.8892318725586,
666
+ -64.13186645507812,
667
+ -70.02197448730469,
668
+ -38.021976470947266,
669
+ -28.76835220336914
670
+ ],
671
+ [
672
+ -45.58066116333008,
673
+ -67.22988708496094,
674
+ -73.89010620117188,
675
+ -39.07692428588867,
676
+ -29.73538604736328
677
+ ],
678
+ [
679
+ -47.163077697753906,
680
+ -70.39472595214843,
681
+ -77.62461700439454,
682
+ -40.74725341796875,
683
+ -30.857141494750977
684
+ ],
685
+ [
686
+ -48.900219116210934,
687
+ -73.55956481933593,
688
+ -81.36264038085938,
689
+ -41.97801971435547,
690
+ -31.91208694458008
691
+ ],
692
+ [
693
+ -50.394725952148434,
694
+ -76.37275146484374,
695
+ -84.54857269287109,
696
+ -42.85714050292969,
697
+ -32.54857177734375
698
+ ],
699
+ [
700
+ -52.04395294189453,
701
+ -79.62549224853515,
702
+ -88.15296569824218,
703
+ -43.60263854980469,
704
+ -33.05494689941406
705
+ ]
706
+ ],
707
+ "q99": [
708
+ [
709
+ 12.483515968322754,
710
+ 15.956043243408203,
711
+ 21.251871337890606,
712
+ 15.5164794921875,
713
+ 10.658461265563945
714
+ ],
715
+ [
716
+ 16.196923904418927,
717
+ 20.548572692871076,
718
+ 27.317805480957013,
719
+ 19.73626419067383,
720
+ 13.47164836883543
721
+ ],
722
+ [
723
+ 19.692306518554688,
724
+ 25.207912597656232,
725
+ 33.18329971313469,
726
+ 23.51648567199707,
727
+ 16.48175796508785
728
+ ],
729
+ [
730
+ 22.85714195251465,
731
+ 29.582416534423828,
732
+ 38.48263854980467,
733
+ 27.141977920532206,
734
+ 19.62549514770506
735
+ ],
736
+ [
737
+ 25.95516555786131,
738
+ 33.66857086181637,
739
+ 43.736263275146484,
740
+ 30.65846176147459,
741
+ 22.438682250976544
742
+ ],
743
+ [
744
+ 28.68043869018553,
745
+ 37.821535644531195,
746
+ 48.32879562377928,
747
+ 34.417579650878906,
748
+ 25.076045684814435
749
+ ],
750
+ [
751
+ 31.47252655029297,
752
+ 41.71428680419922,
753
+ 52.854503173828064,
754
+ 36.90022201538084,
755
+ 27.78022003173828
756
+ ],
757
+ [
758
+ 33.88835083007809,
759
+ 45.73362976074213,
760
+ 57.40483215332027,
761
+ 39.801316375732405,
762
+ 30.681317291259766
763
+ ],
764
+ [
765
+ 35.99824172973629,
766
+ 49.383735809326154,
767
+ 61.142857513427735,
768
+ 42.68132019042969,
769
+ 33.230770111083984
770
+ ],
771
+ [
772
+ 37.538460693359376,
773
+ 52.79121017456055,
774
+ 64.70153656005856,
775
+ 44.72439498901365,
776
+ 35.625496063232404
777
+ ],
778
+ [
779
+ 38.94505401611328,
780
+ 56.30768966674805,
781
+ 67.69055023193356,
782
+ 46.65846466064451,
783
+ 38.0624157714843
784
+ ],
785
+ [
786
+ 40.79121017456055,
787
+ 59.93318954467772,
788
+ 71.4496710205078,
789
+ 48.04396057128906,
790
+ 39.93318572998045
791
+ ],
792
+ [
793
+ 41.71252761840817,
794
+ 63.27384674072264,
795
+ 73.93230468749996,
796
+ 49.626373291015625,
797
+ 41.69142837524412
798
+ ],
799
+ [
800
+ 42.46153732299805,
801
+ 67.12088012695312,
802
+ 77.05494689941406,
803
+ 51.73626129150391,
804
+ 43.44966903686522
805
+ ],
806
+ [
807
+ 43.537581634521466,
808
+ 70.48263946533201,
809
+ 79.09802459716795,
810
+ 52.48175659179684,
811
+ 44.659339904785156
812
+ ],
813
+ [
814
+ 44.57142639160156,
815
+ 73.73537841796873,
816
+ 81.99912658691404,
817
+ 53.4945068359375,
818
+ 46.28395401000973
819
+ ]
820
+ ]
821
+ }
822
+ }
823
+ }
824
+ }
checkpoints/8000/pretrained_model/experiment_cfg/final_model_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "Gr00tN1d7",
3
+ "model_dtype": "bfloat16",
4
+ "model_name": "nvidia/Cosmos-Reason2-2B",
5
+ "backbone_model_type": "qwen",
6
+ "model_revision": null,
7
+ "tune_top_llm_layers": 0,
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": false,
15
+ "backbone_trainable_params_fp32": true,
16
+ "letter_box_transform": false,
17
+ "extra_augmentation_config": null,
18
+ "apply_sincos_state_encoding": false,
19
+ "use_percentiles": true,
20
+ "use_relative_action": false,
21
+ "max_state_dim": 132,
22
+ "max_action_dim": 132,
23
+ "action_horizon": 40,
24
+ "hidden_size": 1024,
25
+ "input_embedding_dim": 1536,
26
+ "state_history_length": 1,
27
+ "add_pos_embed": true,
28
+ "attn_dropout": 0.2,
29
+ "use_vlln": true,
30
+ "max_seq_len": 1024,
31
+ "use_alternate_vl_dit": true,
32
+ "attend_text_every_n_blocks": 2,
33
+ "diffusion_model_cfg": {
34
+ "attention_head_dim": 48,
35
+ "dropout": 0.2,
36
+ "final_dropout": true,
37
+ "interleave_self_attention": true,
38
+ "norm_type": "ada_norm",
39
+ "num_attention_heads": 32,
40
+ "num_layers": 32,
41
+ "output_dim": 1024,
42
+ "positional_embeddings": null
43
+ },
44
+ "num_inference_timesteps": 4,
45
+ "noise_beta_alpha": 1.5,
46
+ "noise_beta_beta": 1.0,
47
+ "noise_s": 0.999,
48
+ "num_timestep_buckets": 1000,
49
+ "tune_projector": true,
50
+ "tune_diffusion_model": true,
51
+ "tune_vlln": true,
52
+ "state_dropout_prob": 0.2,
53
+ "exclude_state": false,
54
+ "use_mean_std": false,
55
+ "max_num_embodiments": 32
56
+ }
checkpoints/8000/pretrained_model/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoints/8000/pretrained_model/model-00001-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cbcaea5ee88f1e0f1465043920a2647c67e7de17d24adfd1c477742a6168edec
3
+ size 4986649584
checkpoints/8000/pretrained_model/model-00002-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8c15f3f9bdc6f932b22de4652829a0511ffb9e15d95d184d43d6eefe9f173126
3
+ size 4970792616
checkpoints/8000/pretrained_model/model-00003-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b32f935fa3173604ff2f2b7c37b0ef2e2849e455fb4ec8d6bdd0742f851b737
3
+ size 2618758696
checkpoints/8000/pretrained_model/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoints/8000/pretrained_model/processor_config.json ADDED
@@ -0,0 +1,1124 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "processor_class": "Gr00tN1d7Processor",
3
+ "processor_kwargs": {
4
+ "modality_configs": {
5
+ "real_g1_relative_eef_relative_joints": {
6
+ "video": {
7
+ "delta_indices": [
8
+ -20,
9
+ 0
10
+ ],
11
+ "modality_keys": [
12
+ "ego_view"
13
+ ],
14
+ "sin_cos_embedding_keys": null,
15
+ "mean_std_embedding_keys": null,
16
+ "action_configs": null
17
+ },
18
+ "state": {
19
+ "delta_indices": [
20
+ 0
21
+ ],
22
+ "modality_keys": [
23
+ "left_wrist_eef_9d",
24
+ "right_wrist_eef_9d",
25
+ "left_hand",
26
+ "right_hand",
27
+ "left_arm",
28
+ "right_arm",
29
+ "waist"
30
+ ],
31
+ "sin_cos_embedding_keys": null,
32
+ "mean_std_embedding_keys": null,
33
+ "action_configs": null
34
+ },
35
+ "action": {
36
+ "delta_indices": [
37
+ 0,
38
+ 1,
39
+ 2,
40
+ 3,
41
+ 4,
42
+ 5,
43
+ 6,
44
+ 7,
45
+ 8,
46
+ 9,
47
+ 10,
48
+ 11,
49
+ 12,
50
+ 13,
51
+ 14,
52
+ 15,
53
+ 16,
54
+ 17,
55
+ 18,
56
+ 19,
57
+ 20,
58
+ 21,
59
+ 22,
60
+ 23,
61
+ 24,
62
+ 25,
63
+ 26,
64
+ 27,
65
+ 28,
66
+ 29,
67
+ 30,
68
+ 31,
69
+ 32,
70
+ 33,
71
+ 34,
72
+ 35,
73
+ 36,
74
+ 37,
75
+ 38,
76
+ 39
77
+ ],
78
+ "modality_keys": [
79
+ "left_wrist_eef_9d",
80
+ "right_wrist_eef_9d",
81
+ "left_hand",
82
+ "right_hand",
83
+ "left_arm",
84
+ "right_arm",
85
+ "waist",
86
+ "base_height_command",
87
+ "navigate_command"
88
+ ],
89
+ "sin_cos_embedding_keys": null,
90
+ "mean_std_embedding_keys": null,
91
+ "action_configs": [
92
+ {
93
+ "rep": "RELATIVE",
94
+ "type": "EEF",
95
+ "format": "XYZ_ROT6D",
96
+ "state_key": "left_wrist_eef_9d"
97
+ },
98
+ {
99
+ "rep": "RELATIVE",
100
+ "type": "EEF",
101
+ "format": "XYZ_ROT6D",
102
+ "state_key": "right_wrist_eef_9d"
103
+ },
104
+ {
105
+ "rep": "ABSOLUTE",
106
+ "type": "NON_EEF",
107
+ "format": "DEFAULT",
108
+ "state_key": "left_hand"
109
+ },
110
+ {
111
+ "rep": "ABSOLUTE",
112
+ "type": "NON_EEF",
113
+ "format": "DEFAULT",
114
+ "state_key": "right_hand"
115
+ },
116
+ {
117
+ "rep": "RELATIVE",
118
+ "type": "NON_EEF",
119
+ "format": "DEFAULT",
120
+ "state_key": "left_arm"
121
+ },
122
+ {
123
+ "rep": "RELATIVE",
124
+ "type": "NON_EEF",
125
+ "format": "DEFAULT",
126
+ "state_key": "right_arm"
127
+ },
128
+ {
129
+ "rep": "ABSOLUTE",
130
+ "type": "NON_EEF",
131
+ "format": "DEFAULT",
132
+ "state_key": "waist"
133
+ },
134
+ {
135
+ "rep": "ABSOLUTE",
136
+ "type": "NON_EEF",
137
+ "format": "DEFAULT",
138
+ "state_key": "base_height_command"
139
+ },
140
+ {
141
+ "rep": "ABSOLUTE",
142
+ "type": "NON_EEF",
143
+ "format": "DEFAULT",
144
+ "state_key": "navigate_command"
145
+ }
146
+ ]
147
+ },
148
+ "language": {
149
+ "delta_indices": [
150
+ 0
151
+ ],
152
+ "modality_keys": [
153
+ "annotation.human.task_description"
154
+ ],
155
+ "sin_cos_embedding_keys": null,
156
+ "mean_std_embedding_keys": null,
157
+ "action_configs": null
158
+ }
159
+ },
160
+ "real_r1_pro_sharpa_relative_eef_mecka": {
161
+ "video": {
162
+ "delta_indices": [
163
+ -30,
164
+ 0
165
+ ],
166
+ "modality_keys": [
167
+ "ego_view_cropratio_res320x240_freq30"
168
+ ],
169
+ "sin_cos_embedding_keys": null,
170
+ "mean_std_embedding_keys": null,
171
+ "action_configs": null
172
+ },
173
+ "state": {
174
+ "delta_indices": [
175
+ 0
176
+ ],
177
+ "modality_keys": [
178
+ "left_wrist_eef",
179
+ "right_wrist_eef",
180
+ "left_hand_joints",
181
+ "right_hand_joints"
182
+ ],
183
+ "sin_cos_embedding_keys": null,
184
+ "mean_std_embedding_keys": null,
185
+ "action_configs": null
186
+ },
187
+ "action": {
188
+ "delta_indices": [
189
+ 0,
190
+ 1,
191
+ 2,
192
+ 3,
193
+ 4,
194
+ 5,
195
+ 6,
196
+ 7,
197
+ 8,
198
+ 9,
199
+ 10,
200
+ 11,
201
+ 12,
202
+ 13,
203
+ 14,
204
+ 15,
205
+ 16,
206
+ 17,
207
+ 18,
208
+ 19,
209
+ 20,
210
+ 21,
211
+ 22,
212
+ 23,
213
+ 24,
214
+ 25,
215
+ 26,
216
+ 27,
217
+ 28,
218
+ 29,
219
+ 30,
220
+ 31,
221
+ 32,
222
+ 33,
223
+ 34,
224
+ 35,
225
+ 36,
226
+ 37,
227
+ 38,
228
+ 39
229
+ ],
230
+ "modality_keys": [
231
+ "left_wrist_eef",
232
+ "right_wrist_eef",
233
+ "left_hand_joints",
234
+ "right_hand_joints"
235
+ ],
236
+ "sin_cos_embedding_keys": null,
237
+ "mean_std_embedding_keys": null,
238
+ "action_configs": [
239
+ {
240
+ "rep": "RELATIVE",
241
+ "type": "EEF",
242
+ "format": "XYZ_ROT6D",
243
+ "state_key": "left_wrist_eef"
244
+ },
245
+ {
246
+ "rep": "RELATIVE",
247
+ "type": "EEF",
248
+ "format": "XYZ_ROT6D",
249
+ "state_key": "right_wrist_eef"
250
+ },
251
+ {
252
+ "rep": "ABSOLUTE",
253
+ "type": "NON_EEF",
254
+ "format": "DEFAULT",
255
+ "state_key": "left_hand_joints"
256
+ },
257
+ {
258
+ "rep": "ABSOLUTE",
259
+ "type": "NON_EEF",
260
+ "format": "DEFAULT",
261
+ "state_key": "right_hand_joints"
262
+ }
263
+ ]
264
+ },
265
+ "language": {
266
+ "delta_indices": [
267
+ 0
268
+ ],
269
+ "modality_keys": [
270
+ "annotation.human.coarse_action"
271
+ ],
272
+ "sin_cos_embedding_keys": null,
273
+ "mean_std_embedding_keys": null,
274
+ "action_configs": null
275
+ }
276
+ },
277
+ "real_r1_pro_sharpa_relative_eef_human": {
278
+ "video": {
279
+ "delta_indices": [
280
+ -20,
281
+ 0
282
+ ],
283
+ "modality_keys": [
284
+ "ego_view_res320x240_freq20",
285
+ "left_wrist_view_res320x240_freq20",
286
+ "right_wrist_view_res320x240_freq20"
287
+ ],
288
+ "sin_cos_embedding_keys": null,
289
+ "mean_std_embedding_keys": null,
290
+ "action_configs": null
291
+ },
292
+ "state": {
293
+ "delta_indices": [
294
+ 0
295
+ ],
296
+ "modality_keys": [
297
+ "left_wrist_eef",
298
+ "right_wrist_eef",
299
+ "left_hand_joints",
300
+ "right_hand_joints"
301
+ ],
302
+ "sin_cos_embedding_keys": null,
303
+ "mean_std_embedding_keys": null,
304
+ "action_configs": null
305
+ },
306
+ "action": {
307
+ "delta_indices": [
308
+ 0,
309
+ 1,
310
+ 2,
311
+ 3,
312
+ 4,
313
+ 5,
314
+ 6,
315
+ 7,
316
+ 8,
317
+ 9,
318
+ 10,
319
+ 11,
320
+ 12,
321
+ 13,
322
+ 14,
323
+ 15,
324
+ 16,
325
+ 17,
326
+ 18,
327
+ 19,
328
+ 20,
329
+ 21,
330
+ 22,
331
+ 23,
332
+ 24,
333
+ 25,
334
+ 26,
335
+ 27,
336
+ 28,
337
+ 29,
338
+ 30,
339
+ 31,
340
+ 32,
341
+ 33,
342
+ 34,
343
+ 35,
344
+ 36,
345
+ 37,
346
+ 38,
347
+ 39
348
+ ],
349
+ "modality_keys": [
350
+ "left_wrist_eef",
351
+ "right_wrist_eef",
352
+ "left_hand_joints",
353
+ "right_hand_joints"
354
+ ],
355
+ "sin_cos_embedding_keys": null,
356
+ "mean_std_embedding_keys": null,
357
+ "action_configs": [
358
+ {
359
+ "rep": "RELATIVE",
360
+ "type": "EEF",
361
+ "format": "XYZ_ROT6D",
362
+ "state_key": "left_wrist_eef"
363
+ },
364
+ {
365
+ "rep": "RELATIVE",
366
+ "type": "EEF",
367
+ "format": "XYZ_ROT6D",
368
+ "state_key": "right_wrist_eef"
369
+ },
370
+ {
371
+ "rep": "ABSOLUTE",
372
+ "type": "NON_EEF",
373
+ "format": "DEFAULT",
374
+ "state_key": "left_hand_joints"
375
+ },
376
+ {
377
+ "rep": "ABSOLUTE",
378
+ "type": "NON_EEF",
379
+ "format": "DEFAULT",
380
+ "state_key": "right_hand_joints"
381
+ }
382
+ ]
383
+ },
384
+ "language": {
385
+ "delta_indices": [
386
+ 0
387
+ ],
388
+ "modality_keys": [
389
+ "annotation.human.coarse_action"
390
+ ],
391
+ "sin_cos_embedding_keys": null,
392
+ "mean_std_embedding_keys": null,
393
+ "action_configs": null
394
+ }
395
+ },
396
+ "real_r1_pro_sharpa_relative_eef": {
397
+ "video": {
398
+ "delta_indices": [
399
+ -20,
400
+ 0
401
+ ],
402
+ "modality_keys": [
403
+ "ego_view_res320x240_freq20",
404
+ "left_wrist_view_res320x240_freq20",
405
+ "right_wrist_view_res320x240_freq20"
406
+ ],
407
+ "sin_cos_embedding_keys": null,
408
+ "mean_std_embedding_keys": null,
409
+ "action_configs": null
410
+ },
411
+ "state": {
412
+ "delta_indices": [
413
+ 0
414
+ ],
415
+ "modality_keys": [
416
+ "left_wrist_eef",
417
+ "right_wrist_eef",
418
+ "left_hand_joints",
419
+ "right_hand_joints"
420
+ ],
421
+ "sin_cos_embedding_keys": null,
422
+ "mean_std_embedding_keys": null,
423
+ "action_configs": null
424
+ },
425
+ "action": {
426
+ "delta_indices": [
427
+ 0,
428
+ 1,
429
+ 2,
430
+ 3,
431
+ 4,
432
+ 5,
433
+ 6,
434
+ 7,
435
+ 8,
436
+ 9,
437
+ 10,
438
+ 11,
439
+ 12,
440
+ 13,
441
+ 14,
442
+ 15,
443
+ 16,
444
+ 17,
445
+ 18,
446
+ 19,
447
+ 20,
448
+ 21,
449
+ 22,
450
+ 23,
451
+ 24,
452
+ 25,
453
+ 26,
454
+ 27,
455
+ 28,
456
+ 29,
457
+ 30,
458
+ 31,
459
+ 32,
460
+ 33,
461
+ 34,
462
+ 35,
463
+ 36,
464
+ 37,
465
+ 38,
466
+ 39
467
+ ],
468
+ "modality_keys": [
469
+ "left_wrist_eef",
470
+ "right_wrist_eef",
471
+ "left_hand_joints",
472
+ "right_hand_joints"
473
+ ],
474
+ "sin_cos_embedding_keys": null,
475
+ "mean_std_embedding_keys": null,
476
+ "action_configs": [
477
+ {
478
+ "rep": "RELATIVE",
479
+ "type": "EEF",
480
+ "format": "XYZ_ROT6D",
481
+ "state_key": "left_wrist_eef"
482
+ },
483
+ {
484
+ "rep": "RELATIVE",
485
+ "type": "EEF",
486
+ "format": "XYZ_ROT6D",
487
+ "state_key": "right_wrist_eef"
488
+ },
489
+ {
490
+ "rep": "ABSOLUTE",
491
+ "type": "NON_EEF",
492
+ "format": "DEFAULT",
493
+ "state_key": "left_hand_joints"
494
+ },
495
+ {
496
+ "rep": "ABSOLUTE",
497
+ "type": "NON_EEF",
498
+ "format": "DEFAULT",
499
+ "state_key": "right_hand_joints"
500
+ }
501
+ ]
502
+ },
503
+ "language": {
504
+ "delta_indices": [
505
+ 0
506
+ ],
507
+ "modality_keys": [
508
+ "annotation.human.coarse_action"
509
+ ],
510
+ "sin_cos_embedding_keys": null,
511
+ "mean_std_embedding_keys": null,
512
+ "action_configs": null
513
+ }
514
+ },
515
+ "xdof_relative_eef_relative_joint": {
516
+ "video": {
517
+ "delta_indices": [
518
+ -30,
519
+ 0
520
+ ],
521
+ "modality_keys": [
522
+ "top_camera-images-rgb_320_240",
523
+ "left_camera-images-rgb_320_240",
524
+ "right_camera-images-rgb_320_240"
525
+ ],
526
+ "sin_cos_embedding_keys": null,
527
+ "mean_std_embedding_keys": null,
528
+ "action_configs": null
529
+ },
530
+ "state": {
531
+ "delta_indices": [
532
+ 0
533
+ ],
534
+ "modality_keys": [
535
+ "left_wrist_eef",
536
+ "right_wrist_eef",
537
+ "left_gripper_pos",
538
+ "right_gripper_pos",
539
+ "left_joint_pos",
540
+ "right_joint_pos"
541
+ ],
542
+ "sin_cos_embedding_keys": null,
543
+ "mean_std_embedding_keys": null,
544
+ "action_configs": null
545
+ },
546
+ "action": {
547
+ "delta_indices": [
548
+ 0,
549
+ 1,
550
+ 2,
551
+ 3,
552
+ 4,
553
+ 5,
554
+ 6,
555
+ 7,
556
+ 8,
557
+ 9,
558
+ 10,
559
+ 11,
560
+ 12,
561
+ 13,
562
+ 14,
563
+ 15,
564
+ 16,
565
+ 17,
566
+ 18,
567
+ 19,
568
+ 20,
569
+ 21,
570
+ 22,
571
+ 23,
572
+ 24,
573
+ 25,
574
+ 26,
575
+ 27,
576
+ 28,
577
+ 29,
578
+ 30,
579
+ 31,
580
+ 32,
581
+ 33,
582
+ 34,
583
+ 35,
584
+ 36,
585
+ 37,
586
+ 38,
587
+ 39
588
+ ],
589
+ "modality_keys": [
590
+ "left_wrist_eef",
591
+ "right_wrist_eef",
592
+ "left_gripper_pos",
593
+ "right_gripper_pos",
594
+ "left_joint_pos",
595
+ "right_joint_pos"
596
+ ],
597
+ "sin_cos_embedding_keys": null,
598
+ "mean_std_embedding_keys": null,
599
+ "action_configs": [
600
+ {
601
+ "rep": "RELATIVE",
602
+ "type": "EEF",
603
+ "format": "XYZ_ROT6D",
604
+ "state_key": "left_wrist_eef"
605
+ },
606
+ {
607
+ "rep": "RELATIVE",
608
+ "type": "EEF",
609
+ "format": "XYZ_ROT6D",
610
+ "state_key": "right_wrist_eef"
611
+ },
612
+ {
613
+ "rep": "ABSOLUTE",
614
+ "type": "NON_EEF",
615
+ "format": "DEFAULT",
616
+ "state_key": "left_gripper_pos"
617
+ },
618
+ {
619
+ "rep": "ABSOLUTE",
620
+ "type": "NON_EEF",
621
+ "format": "DEFAULT",
622
+ "state_key": "right_gripper_pos"
623
+ },
624
+ {
625
+ "rep": "RELATIVE",
626
+ "type": "NON_EEF",
627
+ "format": "DEFAULT",
628
+ "state_key": "left_joint_pos"
629
+ },
630
+ {
631
+ "rep": "RELATIVE",
632
+ "type": "NON_EEF",
633
+ "format": "DEFAULT",
634
+ "state_key": "right_joint_pos"
635
+ }
636
+ ]
637
+ },
638
+ "language": {
639
+ "delta_indices": [
640
+ 0
641
+ ],
642
+ "modality_keys": [
643
+ "annotation.task"
644
+ ],
645
+ "sin_cos_embedding_keys": null,
646
+ "mean_std_embedding_keys": null,
647
+ "action_configs": null
648
+ }
649
+ },
650
+ "real_r1_pro_sharpa_relative_eef_maxinsights": {
651
+ "video": {
652
+ "delta_indices": [
653
+ -30,
654
+ 0
655
+ ],
656
+ "modality_keys": [
657
+ "ego_view_cropratio_res320x240_freq30"
658
+ ],
659
+ "sin_cos_embedding_keys": null,
660
+ "mean_std_embedding_keys": null,
661
+ "action_configs": null
662
+ },
663
+ "state": {
664
+ "delta_indices": [
665
+ 0
666
+ ],
667
+ "modality_keys": [
668
+ "left_wrist_eef",
669
+ "right_wrist_eef",
670
+ "left_hand_joints",
671
+ "right_hand_joints"
672
+ ],
673
+ "sin_cos_embedding_keys": null,
674
+ "mean_std_embedding_keys": null,
675
+ "action_configs": null
676
+ },
677
+ "action": {
678
+ "delta_indices": [
679
+ 0,
680
+ 1,
681
+ 2,
682
+ 3,
683
+ 4,
684
+ 5,
685
+ 6,
686
+ 7,
687
+ 8,
688
+ 9,
689
+ 10,
690
+ 11,
691
+ 12,
692
+ 13,
693
+ 14,
694
+ 15,
695
+ 16,
696
+ 17,
697
+ 18,
698
+ 19,
699
+ 20,
700
+ 21,
701
+ 22,
702
+ 23,
703
+ 24,
704
+ 25,
705
+ 26,
706
+ 27,
707
+ 28,
708
+ 29,
709
+ 30,
710
+ 31,
711
+ 32,
712
+ 33,
713
+ 34,
714
+ 35,
715
+ 36,
716
+ 37,
717
+ 38,
718
+ 39
719
+ ],
720
+ "modality_keys": [
721
+ "left_wrist_eef",
722
+ "right_wrist_eef",
723
+ "left_hand_joints",
724
+ "right_hand_joints"
725
+ ],
726
+ "sin_cos_embedding_keys": null,
727
+ "mean_std_embedding_keys": null,
728
+ "action_configs": [
729
+ {
730
+ "rep": "RELATIVE",
731
+ "type": "EEF",
732
+ "format": "XYZ_ROT6D",
733
+ "state_key": "left_wrist_eef"
734
+ },
735
+ {
736
+ "rep": "RELATIVE",
737
+ "type": "EEF",
738
+ "format": "XYZ_ROT6D",
739
+ "state_key": "right_wrist_eef"
740
+ },
741
+ {
742
+ "rep": "ABSOLUTE",
743
+ "type": "NON_EEF",
744
+ "format": "DEFAULT",
745
+ "state_key": "left_hand_joints"
746
+ },
747
+ {
748
+ "rep": "ABSOLUTE",
749
+ "type": "NON_EEF",
750
+ "format": "DEFAULT",
751
+ "state_key": "right_hand_joints"
752
+ }
753
+ ]
754
+ },
755
+ "language": {
756
+ "delta_indices": [
757
+ 0
758
+ ],
759
+ "modality_keys": [
760
+ "annotation.human.coarse_action"
761
+ ],
762
+ "sin_cos_embedding_keys": null,
763
+ "mean_std_embedding_keys": null,
764
+ "action_configs": null
765
+ }
766
+ },
767
+ "xdof_relative_eef_relative_joint_subtask": {
768
+ "video": {
769
+ "delta_indices": [
770
+ -30,
771
+ 0
772
+ ],
773
+ "modality_keys": [
774
+ "top_camera-images-rgb_320_240",
775
+ "left_camera-images-rgb_320_240",
776
+ "right_camera-images-rgb_320_240"
777
+ ],
778
+ "sin_cos_embedding_keys": null,
779
+ "mean_std_embedding_keys": null,
780
+ "action_configs": null
781
+ },
782
+ "state": {
783
+ "delta_indices": [
784
+ 0
785
+ ],
786
+ "modality_keys": [
787
+ "left_wrist_eef",
788
+ "right_wrist_eef",
789
+ "left_gripper_pos",
790
+ "right_gripper_pos",
791
+ "left_joint_pos",
792
+ "right_joint_pos"
793
+ ],
794
+ "sin_cos_embedding_keys": null,
795
+ "mean_std_embedding_keys": null,
796
+ "action_configs": null
797
+ },
798
+ "action": {
799
+ "delta_indices": [
800
+ 0,
801
+ 1,
802
+ 2,
803
+ 3,
804
+ 4,
805
+ 5,
806
+ 6,
807
+ 7,
808
+ 8,
809
+ 9,
810
+ 10,
811
+ 11,
812
+ 12,
813
+ 13,
814
+ 14,
815
+ 15,
816
+ 16,
817
+ 17,
818
+ 18,
819
+ 19,
820
+ 20,
821
+ 21,
822
+ 22,
823
+ 23,
824
+ 24,
825
+ 25,
826
+ 26,
827
+ 27,
828
+ 28,
829
+ 29,
830
+ 30,
831
+ 31,
832
+ 32,
833
+ 33,
834
+ 34,
835
+ 35,
836
+ 36,
837
+ 37,
838
+ 38,
839
+ 39
840
+ ],
841
+ "modality_keys": [
842
+ "left_wrist_eef",
843
+ "right_wrist_eef",
844
+ "left_gripper_pos",
845
+ "right_gripper_pos",
846
+ "left_joint_pos",
847
+ "right_joint_pos"
848
+ ],
849
+ "sin_cos_embedding_keys": null,
850
+ "mean_std_embedding_keys": null,
851
+ "action_configs": [
852
+ {
853
+ "rep": "RELATIVE",
854
+ "type": "EEF",
855
+ "format": "XYZ_ROT6D",
856
+ "state_key": "left_wrist_eef"
857
+ },
858
+ {
859
+ "rep": "RELATIVE",
860
+ "type": "EEF",
861
+ "format": "XYZ_ROT6D",
862
+ "state_key": "right_wrist_eef"
863
+ },
864
+ {
865
+ "rep": "ABSOLUTE",
866
+ "type": "NON_EEF",
867
+ "format": "DEFAULT",
868
+ "state_key": "left_gripper_pos"
869
+ },
870
+ {
871
+ "rep": "ABSOLUTE",
872
+ "type": "NON_EEF",
873
+ "format": "DEFAULT",
874
+ "state_key": "right_gripper_pos"
875
+ },
876
+ {
877
+ "rep": "RELATIVE",
878
+ "type": "NON_EEF",
879
+ "format": "DEFAULT",
880
+ "state_key": "left_joint_pos"
881
+ },
882
+ {
883
+ "rep": "RELATIVE",
884
+ "type": "NON_EEF",
885
+ "format": "DEFAULT",
886
+ "state_key": "right_joint_pos"
887
+ }
888
+ ]
889
+ },
890
+ "language": {
891
+ "delta_indices": [
892
+ 0
893
+ ],
894
+ "modality_keys": [
895
+ "annotation.sub_task"
896
+ ],
897
+ "sin_cos_embedding_keys": null,
898
+ "mean_std_embedding_keys": null,
899
+ "action_configs": null
900
+ }
901
+ },
902
+ "oxe_droid_relative_eef_relative_joint": {
903
+ "video": {
904
+ "delta_indices": [
905
+ -15,
906
+ 0
907
+ ],
908
+ "modality_keys": [
909
+ "exterior_image_1_left",
910
+ "wrist_image_left"
911
+ ],
912
+ "sin_cos_embedding_keys": null,
913
+ "mean_std_embedding_keys": null,
914
+ "action_configs": null
915
+ },
916
+ "state": {
917
+ "delta_indices": [
918
+ 0
919
+ ],
920
+ "modality_keys": [
921
+ "eef_9d",
922
+ "gripper_position",
923
+ "joint_position"
924
+ ],
925
+ "sin_cos_embedding_keys": null,
926
+ "mean_std_embedding_keys": null,
927
+ "action_configs": null
928
+ },
929
+ "action": {
930
+ "delta_indices": [
931
+ 0,
932
+ 1,
933
+ 2,
934
+ 3,
935
+ 4,
936
+ 5,
937
+ 6,
938
+ 7,
939
+ 8,
940
+ 9,
941
+ 10,
942
+ 11,
943
+ 12,
944
+ 13,
945
+ 14,
946
+ 15,
947
+ 16,
948
+ 17,
949
+ 18,
950
+ 19,
951
+ 20,
952
+ 21,
953
+ 22,
954
+ 23,
955
+ 24,
956
+ 25,
957
+ 26,
958
+ 27,
959
+ 28,
960
+ 29,
961
+ 30,
962
+ 31,
963
+ 32,
964
+ 33,
965
+ 34,
966
+ 35,
967
+ 36,
968
+ 37,
969
+ 38,
970
+ 39
971
+ ],
972
+ "modality_keys": [
973
+ "eef_9d",
974
+ "gripper_position",
975
+ "joint_position"
976
+ ],
977
+ "sin_cos_embedding_keys": null,
978
+ "mean_std_embedding_keys": null,
979
+ "action_configs": [
980
+ {
981
+ "rep": "RELATIVE",
982
+ "type": "EEF",
983
+ "format": "XYZ_ROT6D",
984
+ "state_key": "eef_9d"
985
+ },
986
+ {
987
+ "rep": "ABSOLUTE",
988
+ "type": "NON_EEF",
989
+ "format": "DEFAULT",
990
+ "state_key": "gripper_position"
991
+ },
992
+ {
993
+ "rep": "RELATIVE",
994
+ "type": "NON_EEF",
995
+ "format": "DEFAULT",
996
+ "state_key": "joint_position"
997
+ }
998
+ ]
999
+ },
1000
+ "language": {
1001
+ "delta_indices": [
1002
+ 0
1003
+ ],
1004
+ "modality_keys": [
1005
+ "annotation.language.language_instruction"
1006
+ ],
1007
+ "sin_cos_embedding_keys": null,
1008
+ "mean_std_embedding_keys": null,
1009
+ "action_configs": null
1010
+ }
1011
+ },
1012
+ "new_embodiment": {
1013
+ "video": {
1014
+ "delta_indices": [
1015
+ 0
1016
+ ],
1017
+ "modality_keys": [
1018
+ "front",
1019
+ "wrist",
1020
+ "top"
1021
+ ],
1022
+ "sin_cos_embedding_keys": null,
1023
+ "mean_std_embedding_keys": null,
1024
+ "action_configs": null
1025
+ },
1026
+ "state": {
1027
+ "delta_indices": [
1028
+ 0
1029
+ ],
1030
+ "modality_keys": [
1031
+ "single_arm",
1032
+ "gripper"
1033
+ ],
1034
+ "sin_cos_embedding_keys": null,
1035
+ "mean_std_embedding_keys": null,
1036
+ "action_configs": null
1037
+ },
1038
+ "action": {
1039
+ "delta_indices": [
1040
+ 0,
1041
+ 1,
1042
+ 2,
1043
+ 3,
1044
+ 4,
1045
+ 5,
1046
+ 6,
1047
+ 7,
1048
+ 8,
1049
+ 9,
1050
+ 10,
1051
+ 11,
1052
+ 12,
1053
+ 13,
1054
+ 14,
1055
+ 15
1056
+ ],
1057
+ "modality_keys": [
1058
+ "single_arm",
1059
+ "gripper"
1060
+ ],
1061
+ "sin_cos_embedding_keys": null,
1062
+ "mean_std_embedding_keys": null,
1063
+ "action_configs": [
1064
+ {
1065
+ "rep": "RELATIVE",
1066
+ "type": "NON_EEF",
1067
+ "format": "DEFAULT",
1068
+ "state_key": null
1069
+ },
1070
+ {
1071
+ "rep": "ABSOLUTE",
1072
+ "type": "NON_EEF",
1073
+ "format": "DEFAULT",
1074
+ "state_key": null
1075
+ }
1076
+ ]
1077
+ },
1078
+ "language": {
1079
+ "delta_indices": [
1080
+ 0
1081
+ ],
1082
+ "modality_keys": [
1083
+ "annotation.human.task_description"
1084
+ ],
1085
+ "sin_cos_embedding_keys": null,
1086
+ "mean_std_embedding_keys": null,
1087
+ "action_configs": null
1088
+ }
1089
+ }
1090
+ },
1091
+ "image_crop_size": [
1092
+ 230,
1093
+ 230
1094
+ ],
1095
+ "image_target_size": [
1096
+ 256,
1097
+ 256
1098
+ ],
1099
+ "use_albumentations": true,
1100
+ "random_rotation_angle": 0,
1101
+ "color_jitter_params": {
1102
+ "brightness": 0.3,
1103
+ "contrast": 0.4,
1104
+ "saturation": 0.5,
1105
+ "hue": 0.08
1106
+ },
1107
+ "shortest_image_edge": 256,
1108
+ "crop_fraction": 0.95,
1109
+ "letter_box_transform": false,
1110
+ "model_name": "nvidia/Cosmos-Reason2-2B",
1111
+ "model_type": "qwen",
1112
+ "formalize_language": true,
1113
+ "max_state_dim": 132,
1114
+ "max_action_dim": 132,
1115
+ "max_action_horizon": 40,
1116
+ "use_percentiles": true,
1117
+ "use_mean_std": false,
1118
+ "clip_outliers": true,
1119
+ "apply_sincos_state_encoding": false,
1120
+ "use_relative_action": true,
1121
+ "exclude_state": false,
1122
+ "state_dropout_prob": 0.2
1123
+ }
1124
+ }
checkpoints/8000/pretrained_model/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoints/8000/pretrained_model/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "groot_eyedrops_basket", "run_id": "groot_eyedrops_basket"}