Hugging Face's logo

LiFanxing
/
pack_agx

Safetensors
Model card Files
xet
 Community
LiFanxing commited on
Commit
bb0602c
·
verified ·
1 Parent(s): 6967635

Upload folder using huggingface_hub

Browse files
Files changed (13) hide show
  1. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/config.json +88 -0
  2. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/embodiment_id.json +57 -0
  3. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/conf.yaml +248 -0
  4. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/config.yaml +246 -0
  5. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/dataset_statistics.json +817 -0
  6. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/final_model_config.json +56 -0
  7. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/final_processor_config.json +0 -0
  8. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model-00001-of-00002.safetensors +3 -0
  9. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model-00002-of-00002.safetensors +3 -0
  10. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model.safetensors.index.json +0 -0
  11. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/processor_config.json +1156 -0
  12. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/provenance.json +9 -0
  13. rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/statistics.json +0 -0
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/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": "bfloat16",
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.0,
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": true,
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
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/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
+ "new_embodiment": 10,
55
+ "robocasa_gr1_tabletop": 10,
56
+ "unitree_g1_sonic": 11
57
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/conf.yaml ADDED
@@ -0,0 +1,248 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
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.0
69
+ exclude_state: false
70
+ use_mean_std: false
71
+ max_num_embodiments: 32
72
+ data:
73
+ datasets:
74
+ - dataset_paths:
75
+ - /workspace/datasets/osmo_sft/mimicGen_lerobot
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
+ - high
87
+ - left_wrist_view
88
+ - right_wrist_view
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
+ - left_arm
97
+ - right_arm
98
+ - left_hand
99
+ - right_hand
100
+ sin_cos_embedding_keys: null
101
+ mean_std_embedding_keys: null
102
+ action_configs: null
103
+ action:
104
+ delta_indices:
105
+ - 0
106
+ - 1
107
+ - 2
108
+ - 3
109
+ - 4
110
+ - 5
111
+ - 6
112
+ - 7
113
+ - 8
114
+ - 9
115
+ - 10
116
+ - 11
117
+ - 12
118
+ - 13
119
+ - 14
120
+ - 15
121
+ - 16
122
+ - 17
123
+ - 18
124
+ - 19
125
+ - 20
126
+ - 21
127
+ - 22
128
+ - 23
129
+ - 24
130
+ - 25
131
+ - 26
132
+ - 27
133
+ - 28
134
+ - 29
135
+ - 30
136
+ - 31
137
+ modality_keys:
138
+ - left_arm
139
+ - right_arm
140
+ - left_hand
141
+ - right_hand
142
+ sin_cos_embedding_keys: null
143
+ mean_std_embedding_keys: null
144
+ action_configs:
145
+ - rep: ABSOLUTE
146
+ type: NON_EEF
147
+ format: DEFAULT
148
+ state_key: null
149
+ - rep: ABSOLUTE
150
+ type: NON_EEF
151
+ format: DEFAULT
152
+ state_key: null
153
+ - rep: ABSOLUTE
154
+ type: NON_EEF
155
+ format: DEFAULT
156
+ state_key: null
157
+ - rep: ABSOLUTE
158
+ type: NON_EEF
159
+ format: DEFAULT
160
+ state_key: null
161
+ language:
162
+ delta_indices:
163
+ - 0
164
+ modality_keys:
165
+ - annotation.human.task_description
166
+ sin_cos_embedding_keys: null
167
+ mean_std_embedding_keys: null
168
+ action_configs: null
169
+ download_cache: false
170
+ shard_size: 1024
171
+ episode_sampling_rate: 0.1
172
+ num_shards_per_epoch: 100000
173
+ ds_weights_alpha: null
174
+ override_pretraining_statistics: true
175
+ mode: single_turn
176
+ random_chop: 0.0
177
+ mock_dataset_mode: false
178
+ shuffle: true
179
+ seed: 42
180
+ multiprocessing_context: fork
181
+ allow_padding: false
182
+ subsample_ratio: 1.0
183
+ image_crop_size:
184
+ - 244
185
+ - 244
186
+ image_target_size:
187
+ - 224
188
+ - 224
189
+ training:
190
+ output_dir: /osmo/data/output/sft
191
+ experiment_name: pack_agx_mimicgen_sft_l40s-1
192
+ max_steps: 40000
193
+ global_batch_size: 176
194
+ per_gpu_batch_size: null
195
+ gradient_accumulation_steps: 1
196
+ learning_rate: 0.0001
197
+ lr_scheduler_type: cosine
198
+ weight_decay: 1.0e-05
199
+ warmup_ratio: 0.05
200
+ warmup_steps: 0
201
+ max_grad_norm: 1.0
202
+ optim: adamw_torch
203
+ start_from_checkpoint: nvidia/GR00T-N1.7-3B
204
+ skip_weight_loading: false
205
+ tf32: true
206
+ fp16: false
207
+ bf16: true
208
+ eval_bf16: true
209
+ logging_steps: 10
210
+ save_steps: 10000
211
+ save_total_limit: 2
212
+ save_vl_model: false
213
+ save_only_model: false
214
+ resume_from_checkpoint: false
215
+ upload_checkpoints: false
216
+ upload_every: 1000
217
+ upload_last_n_checkpoints: 5
218
+ max_concurrent_uploads: 2
219
+ eval_strategy: 'no'
220
+ eval_steps: 500
221
+ eval_set_split_ratio: 0.1
222
+ eval_batch_size: 2
223
+ save_best_eval_metric_name: ''
224
+ save_best_eval_metric_greater_is_better: true
225
+ deepspeed_stage: 2
226
+ gradient_checkpointing: true
227
+ transformers_trust_remote_code: true
228
+ transformers_local_files_only: false
229
+ transformers_cache_dir: null
230
+ transformers_access_token: null
231
+ use_ddp: false
232
+ ddp_bucket_cap_mb: 100
233
+ num_gpus: 8
234
+ dataloader_num_workers: 8
235
+ remove_unused_columns: false
236
+ use_wandb: true
237
+ wandb_project: pack_agx_orin
238
+ enable_profiling: false
239
+ max_retries: 3
240
+ assert_loss_less_than: null
241
+ add_rl_callback: false
242
+ enable_open_loop_eval: false
243
+ open_loop_eval_traj_ids:
244
+ - 0
245
+ open_loop_eval_steps_per_traj: 100
246
+ open_loop_eval_plot_indices: null
247
+ max_steps: 40000
248
+ save_steps: 10000
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/config.yaml ADDED
@@ -0,0 +1,246 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
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.0
69
+ exclude_state: false
70
+ use_mean_std: false
71
+ max_num_embodiments: 32
72
+ data:
73
+ datasets:
74
+ - dataset_paths:
75
+ - /workspace/datasets/osmo_sft/mimicGen_lerobot
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
+ - high
87
+ - left_wrist_view
88
+ - right_wrist_view
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
+ - left_arm
97
+ - right_arm
98
+ - left_hand
99
+ - right_hand
100
+ sin_cos_embedding_keys: null
101
+ mean_std_embedding_keys: null
102
+ action_configs: null
103
+ action:
104
+ delta_indices:
105
+ - 0
106
+ - 1
107
+ - 2
108
+ - 3
109
+ - 4
110
+ - 5
111
+ - 6
112
+ - 7
113
+ - 8
114
+ - 9
115
+ - 10
116
+ - 11
117
+ - 12
118
+ - 13
119
+ - 14
120
+ - 15
121
+ - 16
122
+ - 17
123
+ - 18
124
+ - 19
125
+ - 20
126
+ - 21
127
+ - 22
128
+ - 23
129
+ - 24
130
+ - 25
131
+ - 26
132
+ - 27
133
+ - 28
134
+ - 29
135
+ - 30
136
+ - 31
137
+ modality_keys:
138
+ - left_arm
139
+ - right_arm
140
+ - left_hand
141
+ - right_hand
142
+ sin_cos_embedding_keys: null
143
+ mean_std_embedding_keys: null
144
+ action_configs:
145
+ - rep: absolute
146
+ type: non_eef
147
+ format: default
148
+ state_key: null
149
+ - rep: absolute
150
+ type: non_eef
151
+ format: default
152
+ state_key: null
153
+ - rep: absolute
154
+ type: non_eef
155
+ format: default
156
+ state_key: null
157
+ - rep: absolute
158
+ type: non_eef
159
+ format: default
160
+ state_key: null
161
+ language:
162
+ delta_indices:
163
+ - 0
164
+ modality_keys:
165
+ - annotation.human.task_description
166
+ sin_cos_embedding_keys: null
167
+ mean_std_embedding_keys: null
168
+ action_configs: null
169
+ download_cache: false
170
+ shard_size: 1024
171
+ episode_sampling_rate: 0.1
172
+ num_shards_per_epoch: 100000
173
+ ds_weights_alpha: null
174
+ override_pretraining_statistics: true
175
+ mode: single_turn
176
+ random_chop: 0.0
177
+ mock_dataset_mode: false
178
+ shuffle: true
179
+ seed: 42
180
+ multiprocessing_context: fork
181
+ allow_padding: false
182
+ subsample_ratio: 1.0
183
+ image_crop_size:
184
+ - 244
185
+ - 244
186
+ image_target_size:
187
+ - 224
188
+ - 224
189
+ training:
190
+ output_dir: /osmo/data/output/sft
191
+ experiment_name: pack_agx_mimicgen_sft_l40s-1
192
+ max_steps: 40000
193
+ global_batch_size: 176
194
+ per_gpu_batch_size: null
195
+ gradient_accumulation_steps: 1
196
+ learning_rate: 0.0001
197
+ lr_scheduler_type: cosine
198
+ weight_decay: 1.0e-05
199
+ warmup_ratio: 0.05
200
+ warmup_steps: 0
201
+ max_grad_norm: 1.0
202
+ optim: adamw_torch
203
+ start_from_checkpoint: nvidia/GR00T-N1.7-3B
204
+ skip_weight_loading: false
205
+ tf32: true
206
+ fp16: false
207
+ bf16: true
208
+ eval_bf16: true
209
+ logging_steps: 10
210
+ save_steps: 10000
211
+ save_total_limit: 2
212
+ save_vl_model: false
213
+ save_only_model: false
214
+ resume_from_checkpoint: false
215
+ upload_checkpoints: false
216
+ upload_every: 1000
217
+ upload_last_n_checkpoints: 5
218
+ max_concurrent_uploads: 2
219
+ eval_strategy: 'no'
220
+ eval_steps: 500
221
+ eval_set_split_ratio: 0.1
222
+ eval_batch_size: 2
223
+ save_best_eval_metric_name: ''
224
+ save_best_eval_metric_greater_is_better: true
225
+ deepspeed_stage: 2
226
+ gradient_checkpointing: true
227
+ transformers_trust_remote_code: true
228
+ transformers_local_files_only: false
229
+ transformers_cache_dir: null
230
+ transformers_access_token: null
231
+ use_ddp: false
232
+ ddp_bucket_cap_mb: 100
233
+ num_gpus: 8
234
+ dataloader_num_workers: 8
235
+ remove_unused_columns: false
236
+ use_wandb: true
237
+ wandb_project: pack_agx_orin
238
+ enable_profiling: false
239
+ max_retries: 3
240
+ assert_loss_less_than: null
241
+ add_rl_callback: false
242
+ enable_open_loop_eval: false
243
+ open_loop_eval_traj_ids:
244
+ - 0
245
+ open_loop_eval_steps_per_traj: 100
246
+ open_loop_eval_plot_indices: null
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,817 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "state": {
4
+ "left_arm": {
5
+ "min": [
6
+ -2.6180002689361572,
7
+ -0.5170007944107056,
8
+ -2.605445146560669,
9
+ -0.941068172454834,
10
+ -2.6180007457733154,
11
+ -0.5773232579231262,
12
+ -1.21999990940094
13
+ ],
14
+ "max": [
15
+ 1.8330000638961792,
16
+ 1.4686225652694702,
17
+ 2.618000030517578,
18
+ 2.893747568130493,
19
+ 2.618030071258545,
20
+ 0.5336107015609741,
21
+ 1.1208467483520508
22
+ ],
23
+ "mean": [
24
+ 0.03397461026906967,
25
+ 0.06421299278736115,
26
+ 0.05758204311132431,
27
+ 0.27775949239730835,
28
+ -0.13025632500648499,
29
+ 0.007583871018141508,
30
+ -0.07625222206115723
31
+ ],
32
+ "std": [
33
+ 0.2264140546321869,
34
+ 0.11888159066438675,
35
+ 0.3004203140735626,
36
+ 0.4782324433326721,
37
+ 0.38218921422958374,
38
+ 0.1262124478816986,
39
+ 0.19984598457813263
40
+ ],
41
+ "q01": [
42
+ -0.48701825737953186,
43
+ -0.16359113156795502,
44
+ -0.19557182490825653,
45
+ -0.36640194058418274,
46
+ -0.4887334406375885,
47
+ -0.3565746247768402,
48
+ -0.430766761302948
49
+ ],
50
+ "q99": [
51
+ 0.447535902261734,
52
+ 0.30491766333580017,
53
+ 1.7881178855895996,
54
+ 2.027198553085327,
55
+ 1.931532382965088,
56
+ 0.17279334366321564,
57
+ 0.3861079812049866
58
+ ]
59
+ },
60
+ "right_arm": {
61
+ "min": [
62
+ -0.5819464921951294,
63
+ -0.7679711580276489,
64
+ -0.5309500694274902,
65
+ -0.5488454103469849,
66
+ -1.7459508180618286,
67
+ -0.30524349212646484,
68
+ -0.3152182698249817
69
+ ],
70
+ "max": [
71
+ 0.4151162803173065,
72
+ 0.41699811816215515,
73
+ 1.109545350074768,
74
+ 1.1100053787231445,
75
+ 0.5287642478942871,
76
+ 0.5761488676071167,
77
+ 0.6478164196014404
78
+ ],
79
+ "mean": [
80
+ -0.08135099709033966,
81
+ -0.21508654952049255,
82
+ 0.36550894379615784,
83
+ 0.16707195341587067,
84
+ -0.917788565158844,
85
+ 0.35691720247268677,
86
+ 0.19647078216075897
87
+ ],
88
+ "std": [
89
+ 0.19866298139095306,
90
+ 0.23955583572387695,
91
+ 0.3104342520236969,
92
+ 0.27579522132873535,
93
+ 0.5590186715126038,
94
+ 0.17928668856620789,
95
+ 0.21194632351398468
96
+ ],
97
+ "q01": [
98
+ -0.47247207164764404,
99
+ -0.6479050517082214,
100
+ -0.3938380181789398,
101
+ -0.38564446568489075,
102
+ -1.6869972944259644,
103
+ -0.044392626732587814,
104
+ -0.16310521960258484
105
+ ],
106
+ "q99": [
107
+ 0.35601696372032166,
108
+ 0.31312745809555054,
109
+ 0.759691059589386,
110
+ 0.7420726418495178,
111
+ 0.44059351086616516,
112
+ 0.5760000944137573,
113
+ 0.6195146441459656
114
+ ]
115
+ },
116
+ "left_hand": {
117
+ "min": [
118
+ 0.3110215663909912,
119
+ -0.17287808656692505,
120
+ -0.34064748883247375,
121
+ -0.2714831531047821,
122
+ 0.04824570566415787,
123
+ -0.061133794486522675,
124
+ -0.19504642486572266,
125
+ 0.026379019021987915,
126
+ 0.09974434226751328,
127
+ -0.07047039270401001,
128
+ -0.21701787412166595,
129
+ 0.07029285281896591,
130
+ 0.06996158510446548,
131
+ -0.07870163023471832,
132
+ -0.3499046862125397,
133
+ 0.09658147394657135,
134
+ 0.1071639284491539,
135
+ 0.10751783102750778,
136
+ -0.10611534118652344,
137
+ -0.3493713438510895,
138
+ 0.11654524505138397,
139
+ 0.06668936461210251
140
+ ],
141
+ "max": [
142
+ 0.7544029951095581,
143
+ 0.25462716817855835,
144
+ 0.9876896739006042,
145
+ 0.14956827461719513,
146
+ 0.684272825717926,
147
+ 0.6161790490150452,
148
+ -0.05758664011955261,
149
+ 0.16711775958538055,
150
+ 0.21111251413822174,
151
+ 0.523128092288971,
152
+ 0.029198085889220238,
153
+ 0.29464149475097656,
154
+ 0.14509114623069763,
155
+ 0.42642611265182495,
156
+ 0.12231042981147766,
157
+ 0.4376564025878906,
158
+ 0.19374506175518036,
159
+ 0.2616366147994995,
160
+ 0.4910880923271179,
161
+ 0.21618697047233582,
162
+ 0.6632429361343384,
163
+ 0.12185268104076385
164
+ ],
165
+ "mean": [
166
+ 0.5027413368225098,
167
+ 0.0443955697119236,
168
+ 0.7812473773956299,
169
+ -0.24171330034732819,
170
+ 0.0780724510550499,
171
+ 0.4611330032348633,
172
+ -0.17361004650592804,
173
+ 0.08425454795360565,
174
+ 0.13860958814620972,
175
+ 0.30619052052497864,
176
+ -0.04741334170103073,
177
+ 0.17662231624126434,
178
+ 0.08682793378829956,
179
+ 0.19613352417945862,
180
+ -0.035619139671325684,
181
+ 0.28010082244873047,
182
+ 0.15392479300498962,
183
+ 0.13077884912490845,
184
+ 0.0721578449010849,
185
+ -0.017340725287795067,
186
+ 0.49609342217445374,
187
+ 0.07688833773136139
188
+ ],
189
+ "std": [
190
+ 0.13912734389305115,
191
+ 0.11440793424844742,
192
+ 0.1131726279854772,
193
+ 0.030352003872394562,
194
+ 0.04635119065642357,
195
+ 0.06501994282007217,
196
+ 0.0200747549533844,
197
+ 0.02039329893887043,
198
+ 0.020537154749035835,
199
+ 0.04183194041252136,
200
+ 0.009219580329954624,
201
+ 0.013295555487274993,
202
+ 0.004191492684185505,
203
+ 0.044321466237306595,
204
+ 0.011893567629158497,
205
+ 0.01682760939002037,
206
+ 0.022904932498931885,
207
+ 0.004340886604040836,
208
+ 0.05228414386510849,
209
+ 0.02184361405670643,
210
+ 0.043183695524930794,
211
+ 0.004720894619822502
212
+ ],
213
+ "q01": [
214
+ 0.37698373198509216,
215
+ -0.1696707159280777,
216
+ 0.4521143436431885,
217
+ -0.26398035883903503,
218
+ 0.04967504367232323,
219
+ 0.30734509229660034,
220
+ -0.19324928522109985,
221
+ 0.06023586913943291,
222
+ 0.1193559542298317,
223
+ 0.21226876974105835,
224
+ -0.07695815712213516,
225
+ 0.15077728033065796,
226
+ 0.08481554687023163,
227
+ 0.13395845890045166,
228
+ -0.040862325578927994,
229
+ 0.2493990659713745,
230
+ 0.1102863997220993,
231
+ 0.12562963366508484,
232
+ 0.0156185831874609,
233
+ -0.042573098093271255,
234
+ 0.38548898696899414,
235
+ 0.06685926765203476
236
+ ],
237
+ "q99": [
238
+ 0.7471737265586853,
239
+ 0.23414094746112823,
240
+ 0.9048880338668823,
241
+ -0.14035706222057343,
242
+ 0.22135336697101593,
243
+ 0.5865667462348938,
244
+ -0.11968348920345306,
245
+ 0.1341560333967209,
246
+ 0.17702925205230713,
247
+ 0.36039477586746216,
248
+ -0.028067301958799362,
249
+ 0.19317398965358734,
250
+ 0.10577443242073059,
251
+ 0.2631523013114929,
252
+ -0.002699794713407755,
253
+ 0.30015242099761963,
254
+ 0.17408394813537598,
255
+ 0.13388915359973907,
256
+ 0.17234452068805695,
257
+ 0.012402911670506,
258
+ 0.5556599497795105,
259
+ 0.08569242060184479
260
+ ]
261
+ },
262
+ "right_hand": {
263
+ "min": [
264
+ 0.04009956121444702,
265
+ -0.34934455156326294,
266
+ -0.13204336166381836,
267
+ -0.34144726395606995,
268
+ 0.08422064781188965,
269
+ -0.01858464814722538,
270
+ -0.30175480246543884,
271
+ 0.05265137925744057,
272
+ 0.11926719546318054,
273
+ -0.0959416851401329,
274
+ -0.11071047186851501,
275
+ 0.07874276489019394,
276
+ 0.16694247722625732,
277
+ -0.092088982462883,
278
+ -0.10332406312227249,
279
+ 0.05292117968201637,
280
+ 0.12555740773677826,
281
+ -7.358708535321057e-05,
282
+ -0.015641598030924797,
283
+ 0.015677252784371376,
284
+ 0.07094672322273254,
285
+ 0.05056879669427872
286
+ ],
287
+ "max": [
288
+ 0.6687175035476685,
289
+ 0.16571688652038574,
290
+ 0.6953725218772888,
291
+ 0.0049558840692043304,
292
+ 0.42503973841667175,
293
+ 0.643510103225708,
294
+ 0.05673333257436752,
295
+ 0.31882360577583313,
296
+ 0.4645529091358185,
297
+ 0.4012362062931061,
298
+ 0.08867213875055313,
299
+ 0.5379688143730164,
300
+ 0.6116448640823364,
301
+ 0.4138016998767853,
302
+ 0.08890783786773682,
303
+ 0.3182452321052551,
304
+ 0.563942015171051,
305
+ 0.13440823554992676,
306
+ 0.7590939998626709,
307
+ 0.2253386676311493,
308
+ 0.41105303168296814,
309
+ 0.2441820651292801
310
+ ],
311
+ "mean": [
312
+ 0.46442583203315735,
313
+ -0.22639799118041992,
314
+ 0.5469686388969421,
315
+ -0.26744481921195984,
316
+ 0.20058123767375946,
317
+ 0.4130410850048065,
318
+ -0.20870834589004517,
319
+ 0.21364019811153412,
320
+ 0.30965283513069153,
321
+ 0.25716859102249146,
322
+ -0.0806012898683548,
323
+ 0.3747219145298004,
324
+ 0.4548325836658478,
325
+ 0.2679595649242401,
326
+ -0.032462093979120255,
327
+ 0.22699682414531708,
328
+ 0.45014163851737976,
329
+ 0.05953042581677437,
330
+ 0.6587392687797546,
331
+ 0.11691971868276596,
332
+ 0.27107304334640503,
333
+ 0.1671062558889389
334
+ ],
335
+ "std": [
336
+ 0.18579652905464172,
337
+ 0.14615590870380402,
338
+ 0.21752281486988068,
339
+ 0.044575758278369904,
340
+ 0.07848440110683441,
341
+ 0.18576763570308685,
342
+ 0.049130793660879135,
343
+ 0.09474170953035355,
344
+ 0.09909486025571823,
345
+ 0.14756469428539276,
346
+ 0.016289111226797104,
347
+ 0.13882380723953247,
348
+ 0.1168961226940155,
349
+ 0.11799615621566772,
350
+ 0.053248677402734756,
351
+ 0.07415777444839478,
352
+ 0.10647093504667282,
353
+ 0.062276870012283325,
354
+ 0.11879948526620865,
355
+ 0.04300769791007042,
356
+ 0.10353684425354004,
357
+ 0.06488075107336044
358
+ ],
359
+ "q01": [
360
+ 0.07583548128604889,
361
+ -0.3491000235080719,
362
+ -0.12692509591579437,
363
+ -0.3310675621032715,
364
+ 0.091617651283741,
365
+ 0.004721063654869795,
366
+ -0.2879877984523773,
367
+ 0.05265168473124504,
368
+ 0.1209481805562973,
369
+ -0.08445091545581818,
370
+ -0.11070045083761215,
371
+ 0.08550947159528732,
372
+ 0.17066535353660583,
373
+ -0.07877332717180252,
374
+ -0.09169323742389679,
375
+ 0.058404043316841125,
376
+ 0.13074491918087006,
377
+ -1.3899413175977315e-08,
378
+ 0.2881202697753906,
379
+ 0.015767907723784447,
380
+ 0.07359182089567184,
381
+ 0.05076809972524643
382
+ ],
383
+ "q99": [
384
+ 0.6641287207603455,
385
+ 0.16195383667945862,
386
+ 0.6798381209373474,
387
+ -0.1688455492258072,
388
+ 0.4249716103076935,
389
+ 0.6336961984634399,
390
+ -0.12000460177659988,
391
+ 0.31848323345184326,
392
+ 0.46309179067611694,
393
+ 0.4007294178009033,
394
+ -0.05453607439994812,
395
+ 0.5376054644584656,
396
+ 0.6094125509262085,
397
+ 0.4007857143878937,
398
+ 0.08853020519018173,
399
+ 0.3166349530220032,
400
+ 0.5605353713035583,
401
+ 0.13426423072814941,
402
+ 0.7479431629180908,
403
+ 0.2228250950574875,
404
+ 0.40399208664894104,
405
+ 0.24417828023433685
406
+ ]
407
+ }
408
+ },
409
+ "action": {
410
+ "left_arm": {
411
+ "min": [
412
+ -2.435500144958496,
413
+ -0.5170005559921265,
414
+ -2.6074514389038086,
415
+ -0.960104763507843,
416
+ -2.6180002689361572,
417
+ -0.5761909484863281,
418
+ -1.21999990940094
419
+ ],
420
+ "max": [
421
+ 1.650720238685608,
422
+ 1.5118794441223145,
423
+ 2.618000030517578,
424
+ 2.941423177719116,
425
+ 2.618009567260742,
426
+ 0.554316520690918,
427
+ 1.150658130645752
428
+ ],
429
+ "mean": [
430
+ 0.030979428440332413,
431
+ 0.059244874864816666,
432
+ 0.055112969130277634,
433
+ 0.2781330645084381,
434
+ -0.12910927832126617,
435
+ 0.003901271615177393,
436
+ -0.0813087597489357
437
+ ],
438
+ "std": [
439
+ 0.22147201001644135,
440
+ 0.11688368022441864,
441
+ 0.3068068027496338,
442
+ 0.4805169403553009,
443
+ 0.38486242294311523,
444
+ 0.1303543895483017,
445
+ 0.19689610600471497
446
+ ],
447
+ "q01": [
448
+ -0.5192533135414124,
449
+ -0.1744977980852127,
450
+ -0.2101220339536667,
451
+ -0.36943936347961426,
452
+ -0.4900490343570709,
453
+ -0.36965081095695496,
454
+ -0.43037477135658264
455
+ ],
456
+ "q99": [
457
+ 0.44855770468711853,
458
+ 0.2825288474559784,
459
+ 1.8330429792404175,
460
+ 2.0240910053253174,
461
+ 1.9558473825454712,
462
+ 0.17112790048122406,
463
+ 0.3625663220882416
464
+ ]
465
+ },
466
+ "right_arm": {
467
+ "min": [
468
+ -0.6008480191230774,
469
+ -0.7692788243293762,
470
+ -0.5288291573524475,
471
+ -0.5501343011856079,
472
+ -1.7618770599365234,
473
+ -0.2748262882232666,
474
+ -0.3167533874511719
475
+ ],
476
+ "max": [
477
+ 0.41644051671028137,
478
+ 0.414732962846756,
479
+ 1.192404866218567,
480
+ 1.162872314453125,
481
+ 0.5300719141960144,
482
+ 0.5760312080383301,
483
+ 0.6244572401046753
484
+ ],
485
+ "mean": [
486
+ -0.0862913578748703,
487
+ -0.21357138454914093,
488
+ 0.36508581042289734,
489
+ 0.16368941962718964,
490
+ -0.9202964305877686,
491
+ 0.3586640954017639,
492
+ 0.19258983433246613
493
+ ],
494
+ "std": [
495
+ 0.20573680102825165,
496
+ 0.2410082370042801,
497
+ 0.31232622265815735,
498
+ 0.28223201632499695,
499
+ 0.5672284960746765,
500
+ 0.17856332659721375,
501
+ 0.19678549468517303
502
+ ],
503
+ "q01": [
504
+ -0.4935360252857208,
505
+ -0.6506670713424683,
506
+ -0.39635327458381653,
507
+ -0.3884526193141937,
508
+ -1.7035306692123413,
509
+ -0.035909153521060944,
510
+ -0.1532890796661377
511
+ ],
512
+ "q99": [
513
+ 0.3578270971775055,
514
+ 0.3123546540737152,
515
+ 0.7585879564285278,
516
+ 0.75949627161026,
517
+ 0.44765618443489075,
518
+ 0.5759999752044678,
519
+ 0.5937375426292419
520
+ ]
521
+ },
522
+ "left_hand": {
523
+ "min": [
524
+ 0.3918694257736206,
525
+ -0.17544984817504883,
526
+ 0.5742565393447876,
527
+ -0.26477256417274475,
528
+ 0.047149769961833954,
529
+ 0.41272443532943726,
530
+ -0.1980220228433609,
531
+ 0.0594881996512413,
532
+ 0.1185695081949234,
533
+ 0.27439790964126587,
534
+ -0.08423010259866714,
535
+ 0.1644398719072342,
536
+ 0.0840228721499443,
537
+ 0.15344059467315674,
538
+ -0.04130871221423149,
539
+ 0.25055503845214844,
540
+ 0.11074753105640411,
541
+ 0.13054612278938293,
542
+ 0.029649898409843445,
543
+ -0.03919065743684769,
544
+ 0.4713686406612396,
545
+ 0.0660797655582428
546
+ ],
547
+ "max": [
548
+ 0.7500954270362854,
549
+ 0.23579378426074982,
550
+ 0.9043861627578735,
551
+ -0.19977086782455444,
552
+ 0.10726554691791534,
553
+ 0.5934039950370789,
554
+ -0.11681664735078812,
555
+ 0.13553768396377563,
556
+ 0.177893728017807,
557
+ 0.36133188009262085,
558
+ -0.036417875438928604,
559
+ 0.19358991086483002,
560
+ 0.08884952962398529,
561
+ 0.26334381103515625,
562
+ -0.03527446091175079,
563
+ 0.3009660840034485,
564
+ 0.17489220201969147,
565
+ 0.13134565949440002,
566
+ 0.17368391156196594,
567
+ 0.007294907234609127,
568
+ 0.554821789264679,
569
+ 0.08312493562698364
570
+ ],
571
+ "mean": [
572
+ 0.502416729927063,
573
+ 0.04052925854921341,
574
+ 0.7890733480453491,
575
+ -0.24464645981788635,
576
+ 0.07288830727338791,
577
+ 0.46554625034332275,
578
+ -0.17521795630455017,
579
+ 0.08465934544801712,
580
+ 0.13840633630752563,
581
+ 0.3096136152744293,
582
+ -0.04855429753661156,
583
+ 0.17727497220039368,
584
+ 0.08652010560035706,
585
+ 0.20092320442199707,
586
+ -0.0386708527803421,
587
+ 0.28234758973121643,
588
+ 0.15446864068508148,
589
+ 0.1308107227087021,
590
+ 0.07289990037679672,
591
+ -0.01834801398217678,
592
+ 0.49866196513175964,
593
+ 0.0763162299990654
594
+ ],
595
+ "std": [
596
+ 0.13876068592071533,
597
+ 0.11674730479717255,
598
+ 0.0806949064135548,
599
+ 0.014196805655956268,
600
+ 0.021397452801465988,
601
+ 0.05359002575278269,
602
+ 0.019477881491184235,
603
+ 0.0198651310056448,
604
+ 0.020826028659939766,
605
+ 0.032544609159231186,
606
+ 0.008071118034422398,
607
+ 0.011100209318101366,
608
+ 0.0013812492834403276,
609
+ 0.04264937341213226,
610
+ 0.0012516834540292566,
611
+ 0.011456096544861643,
612
+ 0.02276504971086979,
613
+ 0.00011685315985412359,
614
+ 0.048566266894340515,
615
+ 0.015242626890540123,
616
+ 0.0327131040394304,
617
+ 0.003983195405453417
618
+ ],
619
+ "q01": [
620
+ 0.3929607570171356,
621
+ -0.17535732686519623,
622
+ 0.5742722749710083,
623
+ -0.2647572159767151,
624
+ 0.047164663672447205,
625
+ 0.4127298593521118,
626
+ -0.19425080716609955,
627
+ 0.059504516422748566,
628
+ 0.11858192831277847,
629
+ 0.274588018655777,
630
+ -0.07925879955291748,
631
+ 0.16444341838359833,
632
+ 0.0840383917093277,
633
+ 0.15349750220775604,
634
+ -0.04087769240140915,
635
+ 0.250562846660614,
636
+ 0.11118252575397491,
637
+ 0.1305932104587555,
638
+ 0.030105693265795708,
639
+ -0.039142828434705734,
640
+ 0.4713757038116455,
641
+ 0.06610637158155441
642
+ ],
643
+ "q99": [
644
+ 0.746482253074646,
645
+ 0.23534412682056427,
646
+ 0.9043689370155334,
647
+ -0.19979244470596313,
648
+ 0.10725811123847961,
649
+ 0.5912734270095825,
650
+ -0.11782712489366531,
651
+ 0.13544833660125732,
652
+ 0.17776291072368622,
653
+ 0.3613196015357971,
654
+ -0.03777790069580078,
655
+ 0.19358032941818237,
656
+ 0.08880164474248886,
657
+ 0.26332810521125793,
658
+ -0.0365343913435936,
659
+ 0.30095863342285156,
660
+ 0.17485184967517853,
661
+ 0.1311883181333542,
662
+ 0.17366233468055725,
663
+ 0.0039198752492666245,
664
+ 0.5548086166381836,
665
+ 0.08310144394636154
666
+ ]
667
+ },
668
+ "right_hand": {
669
+ "min": [
670
+ 0.05656527355313301,
671
+ -0.31667423248291016,
672
+ -0.14167125523090363,
673
+ -0.3035319447517395,
674
+ 0.10522574931383133,
675
+ -0.03028453327715397,
676
+ -0.28060656785964966,
677
+ 0.051752351224422455,
678
+ 0.11661845445632935,
679
+ -0.10934512317180634,
680
+ -0.11242455244064331,
681
+ 0.06948084384202957,
682
+ 0.158926323056221,
683
+ -0.10716329514980316,
684
+ -0.08241944760084152,
685
+ 0.048943858593702316,
686
+ 0.1167350709438324,
687
+ 0.02247517742216587,
688
+ 0.27274423837661743,
689
+ 0.014368311502039433,
690
+ 0.06936880946159363,
691
+ 0.04981590062379837
692
+ ],
693
+ "max": [
694
+ 0.7023979425430298,
695
+ 0.16711464524269104,
696
+ 0.7318638563156128,
697
+ -0.20320987701416016,
698
+ 0.5249636173248291,
699
+ 0.643631100654602,
700
+ -0.151748389005661,
701
+ 0.31930866837501526,
702
+ 0.4663243591785431,
703
+ 0.40068960189819336,
704
+ -0.05086810886859894,
705
+ 0.5383633375167847,
706
+ 0.6140285134315491,
707
+ 0.4134733974933624,
708
+ 0.09306606650352478,
709
+ 0.32129883766174316,
710
+ 0.5657610297203064,
711
+ 0.1346166580915451,
712
+ 0.7374468445777893,
713
+ 0.22712412476539612,
714
+ 0.4027511477470398,
715
+ 0.24494193494319916
716
+ ],
717
+ "mean": [
718
+ 0.48479023575782776,
719
+ -0.19300033152103424,
720
+ 0.5662065148353577,
721
+ -0.2588498294353485,
722
+ 0.20845599472522736,
723
+ 0.42852354049682617,
724
+ -0.2047835737466812,
725
+ 0.21921195089817047,
726
+ 0.30921033024787903,
727
+ 0.26010358333587646,
728
+ -0.08231329917907715,
729
+ 0.3772484362125397,
730
+ 0.45590940117836,
731
+ 0.2839793264865875,
732
+ -0.03281934931874275,
733
+ 0.23261258006095886,
734
+ 0.4511491656303406,
735
+ 0.07160188257694244,
736
+ 0.6632956266403198,
737
+ 0.08630979061126709,
738
+ 0.2724231779575348,
739
+ 0.16675138473510742
740
+ ],
741
+ "std": [
742
+ 0.2009241133928299,
743
+ 0.12601718306541443,
744
+ 0.23686330020427704,
745
+ 0.029346652328968048,
746
+ 0.08104277402162552,
747
+ 0.19370298087596893,
748
+ 0.04986842721700668,
749
+ 0.09878876060247421,
750
+ 0.10072118788957596,
751
+ 0.14988134801387787,
752
+ 0.013688120990991592,
753
+ 0.13981981575489044,
754
+ 0.11708086729049683,
755
+ 0.12859278917312622,
756
+ 0.052582766860723495,
757
+ 0.07852862030267715,
758
+ 0.10716310143470764,
759
+ 0.052044063806533813,
760
+ 0.10737286508083344,
761
+ 0.05219470337033272,
762
+ 0.1040814146399498,
763
+ 0.06587822735309601
764
+ ],
765
+ "q01": [
766
+ 0.06780627369880676,
767
+ -0.3166660964488983,
768
+ -0.13796158134937286,
769
+ -0.30219724774360657,
770
+ 0.10594586282968521,
771
+ -0.013602913357317448,
772
+ -0.2792985439300537,
773
+ 0.05178355425596237,
774
+ 0.11790037900209427,
775
+ -0.10453744977712631,
776
+ -0.1124066486954689,
777
+ 0.07422985881567001,
778
+ 0.16193227469921112,
779
+ -0.09863691031932831,
780
+ -0.0823676809668541,
781
+ 0.051408782601356506,
782
+ 0.12347953021526337,
783
+ 0.02248801849782467,
784
+ 0.27782076597213745,
785
+ 0.014379582367837429,
786
+ 0.06937909126281738,
787
+ 0.04985225573182106
788
+ ],
789
+ "q99": [
790
+ 0.702386736869812,
791
+ 0.16283226013183594,
792
+ 0.7304694652557373,
793
+ -0.20540979504585266,
794
+ 0.42579561471939087,
795
+ 0.6435886025428772,
796
+ -0.1518232524394989,
797
+ 0.3192494511604309,
798
+ 0.4635205864906311,
799
+ 0.4006670415401459,
800
+ -0.053987760096788406,
801
+ 0.538344144821167,
802
+ 0.6100354790687561,
803
+ 0.4134240746498108,
804
+ 0.08971802145242691,
805
+ 0.3174830377101898,
806
+ 0.5614163875579834,
807
+ 0.13442933559417725,
808
+ 0.7374396920204163,
809
+ 0.2251783162355423,
810
+ 0.4027312994003296,
811
+ 0.24489758908748627
812
+ ]
813
+ }
814
+ },
815
+ "relative_action": {}
816
+ }
817
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/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": true,
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.0,
53
+ "exclude_state": false,
54
+ "use_mean_std": false,
55
+ "max_num_embodiments": 32
56
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ff11128357c8ca2694ca3e749578aa02afc6d3a7f73d283afc34b4a253da950
3
+ size 4990519200
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce1a0c1b4549b4d0c6adf6fe021f53b2425e7d5d4f0d7b1133615c5778ff9093
3
+ size 1919980152
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/processor_config.json ADDED
@@ -0,0 +1,1156 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "high",
1019
+ "left_wrist_view",
1020
+ "right_wrist_view"
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
+ "left_arm",
1032
+ "right_arm",
1033
+ "left_hand",
1034
+ "right_hand"
1035
+ ],
1036
+ "sin_cos_embedding_keys": null,
1037
+ "mean_std_embedding_keys": null,
1038
+ "action_configs": null
1039
+ },
1040
+ "action": {
1041
+ "delta_indices": [
1042
+ 0,
1043
+ 1,
1044
+ 2,
1045
+ 3,
1046
+ 4,
1047
+ 5,
1048
+ 6,
1049
+ 7,
1050
+ 8,
1051
+ 9,
1052
+ 10,
1053
+ 11,
1054
+ 12,
1055
+ 13,
1056
+ 14,
1057
+ 15,
1058
+ 16,
1059
+ 17,
1060
+ 18,
1061
+ 19,
1062
+ 20,
1063
+ 21,
1064
+ 22,
1065
+ 23,
1066
+ 24,
1067
+ 25,
1068
+ 26,
1069
+ 27,
1070
+ 28,
1071
+ 29,
1072
+ 30,
1073
+ 31
1074
+ ],
1075
+ "modality_keys": [
1076
+ "left_arm",
1077
+ "right_arm",
1078
+ "left_hand",
1079
+ "right_hand"
1080
+ ],
1081
+ "sin_cos_embedding_keys": null,
1082
+ "mean_std_embedding_keys": null,
1083
+ "action_configs": [
1084
+ {
1085
+ "rep": "ABSOLUTE",
1086
+ "type": "NON_EEF",
1087
+ "format": "DEFAULT",
1088
+ "state_key": null
1089
+ },
1090
+ {
1091
+ "rep": "ABSOLUTE",
1092
+ "type": "NON_EEF",
1093
+ "format": "DEFAULT",
1094
+ "state_key": null
1095
+ },
1096
+ {
1097
+ "rep": "ABSOLUTE",
1098
+ "type": "NON_EEF",
1099
+ "format": "DEFAULT",
1100
+ "state_key": null
1101
+ },
1102
+ {
1103
+ "rep": "ABSOLUTE",
1104
+ "type": "NON_EEF",
1105
+ "format": "DEFAULT",
1106
+ "state_key": null
1107
+ }
1108
+ ]
1109
+ },
1110
+ "language": {
1111
+ "delta_indices": [
1112
+ 0
1113
+ ],
1114
+ "modality_keys": [
1115
+ "annotation.human.task_description"
1116
+ ],
1117
+ "sin_cos_embedding_keys": null,
1118
+ "mean_std_embedding_keys": null,
1119
+ "action_configs": null
1120
+ }
1121
+ }
1122
+ },
1123
+ "image_crop_size": [
1124
+ 230,
1125
+ 230
1126
+ ],
1127
+ "image_target_size": [
1128
+ 256,
1129
+ 256
1130
+ ],
1131
+ "use_albumentations": true,
1132
+ "random_rotation_angle": 0,
1133
+ "color_jitter_params": {
1134
+ "brightness": 0.3,
1135
+ "contrast": 0.4,
1136
+ "saturation": 0.5,
1137
+ "hue": 0.08
1138
+ },
1139
+ "shortest_image_edge": 256,
1140
+ "crop_fraction": 0.95,
1141
+ "letter_box_transform": false,
1142
+ "model_name": "nvidia/Cosmos-Reason2-2B",
1143
+ "model_type": "qwen",
1144
+ "formalize_language": true,
1145
+ "max_state_dim": 132,
1146
+ "max_action_dim": 132,
1147
+ "max_action_horizon": 40,
1148
+ "use_percentiles": true,
1149
+ "use_mean_std": false,
1150
+ "clip_outliers": true,
1151
+ "apply_sincos_state_encoding": false,
1152
+ "use_relative_action": true,
1153
+ "exclude_state": false,
1154
+ "state_dropout_prob": 0.0
1155
+ }
1156
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/provenance.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "checkpoint": "global_step_800",
3
+ "global_step": 800,
4
+ "source_weights": "/home/fansing/pack_agx/sim2real/downloads/osmo_checkpoints/rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/actor/model_state_dict/full_weights.pt",
5
+ "reference": "/home/fansing/pack_agx/sim2real/downloads/osmo_checkpoints/rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800-reference",
6
+ "policy_tensors": 1031,
7
+ "dropped_rl_tensors": 8,
8
+ "tensor_bytes": 6910361856
9
+ }
rl_pack_agx_orin_contR_004pos_10ang-1/global_step_800/statistics.json ADDED
The diff for this file is too large to render. See raw diff