Hugging Face's logo

EbotsVLA
/
cable_harness_gr00t_models

Safetensors
Model card Files
xet
 Community
hossein-ebots commited on
Commit
49be034
·
verified ·
1 Parent(s): 92e67ea

Uploaded a new model

Browse files
Files changed (18) hide show
  1. checkpoint-25000/config.json +70 -0
  2. checkpoint-25000/embodiment_id.json +11 -0
  3. checkpoint-25000/experiment_cfg/conf.yaml +210 -0
  4. checkpoint-25000/experiment_cfg/config.yaml +244 -0
  5. checkpoint-25000/experiment_cfg/dataset_statistics.json +932 -0
  6. checkpoint-25000/experiment_cfg/final_model_config.json +53 -0
  7. checkpoint-25000/experiment_cfg/final_processor_config.json +0 -0
  8. checkpoint-25000/model-00001-of-00002.safetensors +3 -0
  9. checkpoint-25000/model-00002-of-00002.safetensors +3 -0
  10. checkpoint-25000/model.safetensors.index.json +0 -0
  11. checkpoint-25000/optimizer.pt +3 -0
  12. checkpoint-25000/processor_config.json +456 -0
  13. checkpoint-25000/rng_state.pth +3 -0
  14. checkpoint-25000/scheduler.pt +3 -0
  15. checkpoint-25000/statistics.json +0 -0
  16. checkpoint-25000/trainer_state.json +0 -0
  17. checkpoint-25000/training_args.bin +3 -0
  18. checkpoint-25000/wandb_config.json +1 -0
checkpoint-25000/config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_horizon": 50,
3
+ "add_pos_embed": true,
4
+ "apply_sincos_state_encoding": true,
5
+ "architectures": [
6
+ "Gr00tN1d6"
7
+ ],
8
+ "attn_dropout": 0.2,
9
+ "attn_implementation": null,
10
+ "backbone_embedding_dim": 2048,
11
+ "backbone_model_type": "eagle",
12
+ "backbone_trainable_params_fp32": true,
13
+ "collator_overwrite_image_inputs": false,
14
+ "color_jitter_params": {
15
+ "brightness": 0.1,
16
+ "contrast": 0.1,
17
+ "hue": 0.1,
18
+ "saturation": 0.1
19
+ },
20
+ "crop_fraction": 0.95,
21
+ "diffusion_model_cfg": {
22
+ "attention_head_dim": 48,
23
+ "dropout": 0.2,
24
+ "final_dropout": true,
25
+ "interleave_self_attention": true,
26
+ "norm_type": "ada_norm",
27
+ "num_attention_heads": 32,
28
+ "num_layers": 32,
29
+ "output_dim": 1024,
30
+ "positional_embeddings": null
31
+ },
32
+ "eagle_collator": true,
33
+ "formalize_language": true,
34
+ "gemma_collator": false,
35
+ "hidden_size": 1024,
36
+ "image_crop_size": null,
37
+ "image_target_size": null,
38
+ "input_embedding_dim": 1536,
39
+ "load_bf16": true,
40
+ "max_action_dim": 128,
41
+ "max_num_embodiments": 32,
42
+ "max_seq_len": 1024,
43
+ "max_state_dim": 128,
44
+ "model_dtype": "bfloat16",
45
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
46
+ "model_type": "Gr00tN1d6",
47
+ "noise_beta_alpha": 1.5,
48
+ "noise_beta_beta": 1.0,
49
+ "noise_s": 0.999,
50
+ "num_inference_timesteps": 4,
51
+ "num_timestep_buckets": 1000,
52
+ "random_rotation_angle": null,
53
+ "reproject_vision": false,
54
+ "select_layer": 16,
55
+ "shortest_image_edge": 256,
56
+ "state_dropout_prob": 0.0,
57
+ "torch_dtype": "bfloat16",
58
+ "transformers_version": "4.51.3",
59
+ "tune_diffusion_model": true,
60
+ "tune_llm": false,
61
+ "tune_projector": true,
62
+ "tune_top_llm_layers": 4,
63
+ "tune_visual": false,
64
+ "tune_vlln": true,
65
+ "use_albumentations_transforms": true,
66
+ "use_alternate_vl_dit": true,
67
+ "use_flash_attention": true,
68
+ "use_relative_action": true,
69
+ "use_vlln": true
70
+ }
checkpoint-25000/embodiment_id.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "robocasa_panda_omron": 13,
3
+ "gr1": 20,
4
+ "behavior_r1_pro": 24,
5
+ "unitree_g1": 8,
6
+ "oxe_google": 0,
7
+ "oxe_widowx": 1,
8
+ "libero_panda": 2,
9
+ "oxe_droid": 16,
10
+ "new_embodiment": 10
11
+ }
checkpoint-25000/experiment_cfg/conf.yaml ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ load_config_path: null
2
+ model:
3
+ model_type: Gr00tN1d6
4
+ model_dtype: bfloat16
5
+ model_name: nvidia/Eagle-Block2A-2B-v2
6
+ backbone_model_type: eagle
7
+ model_revision: null
8
+ tune_top_llm_layers: 4
9
+ backbone_embedding_dim: 2048
10
+ tune_llm: false
11
+ tune_visual: false
12
+ select_layer: 16
13
+ reproject_vision: false
14
+ use_flash_attention: true
15
+ load_bf16: false
16
+ collator_overwrite_image_inputs: false
17
+ eagle_collator: true
18
+ backbone_trainable_params_fp32: true
19
+ image_crop_size: null
20
+ image_target_size: null
21
+ shortest_image_edge: 256
22
+ crop_fraction: 0.95
23
+ random_rotation_angle: 5
24
+ color_jitter_params:
25
+ brightness: 0.3
26
+ contrast: 0.4
27
+ saturation: 0.5
28
+ hue: 0.08
29
+ use_albumentations_transforms: true
30
+ formalize_language: true
31
+ apply_sincos_state_encoding: false
32
+ use_relative_action: true
33
+ max_state_dim: 29
34
+ max_action_dim: 29
35
+ action_horizon: 16
36
+ hidden_size: 1024
37
+ input_embedding_dim: 1536
38
+ add_pos_embed: true
39
+ attn_dropout: 0.2
40
+ use_vlln: true
41
+ max_seq_len: 1024
42
+ use_alternate_vl_dit: true
43
+ attend_text_every_n_blocks: 2
44
+ diffusion_model_cfg:
45
+ positional_embeddings: null
46
+ num_layers: 32
47
+ num_attention_heads: 32
48
+ attention_head_dim: 48
49
+ norm_type: ada_norm
50
+ dropout: 0.2
51
+ final_dropout: true
52
+ output_dim: 1024
53
+ interleave_self_attention: true
54
+ num_inference_timesteps: 4
55
+ noise_beta_alpha: 1.5
56
+ noise_beta_beta: 1.0
57
+ noise_s: 0.999
58
+ num_timestep_buckets: 1000
59
+ tune_projector: true
60
+ tune_diffusion_model: true
61
+ tune_vlln: true
62
+ state_dropout_prob: 0.0
63
+ state_additive_noise_scale: 0.0
64
+ max_num_embodiments: 32
65
+ data:
66
+ datasets:
67
+ - dataset_paths:
68
+ - /home/hsaadatzi/set_1
69
+ embodiment_tag: new_embodiment
70
+ mix_ratio: 1.0
71
+ dataset_type: physical_embodiment
72
+ val_dataset_path: null
73
+ modality_configs:
74
+ new_embodiment:
75
+ video:
76
+ delta_indices:
77
+ - 0
78
+ modality_keys:
79
+ - cam_high
80
+ - cam_left_wrist
81
+ - cam_side
82
+ sin_cos_embedding_keys: null
83
+ mean_std_embedding_keys: null
84
+ action_configs: null
85
+ state:
86
+ delta_indices:
87
+ - 0
88
+ modality_keys:
89
+ - arm_l
90
+ - arm_l_finger
91
+ sin_cos_embedding_keys: null
92
+ mean_std_embedding_keys: null
93
+ action_configs: null
94
+ action:
95
+ delta_indices:
96
+ - 0
97
+ - 1
98
+ - 2
99
+ - 3
100
+ - 4
101
+ - 5
102
+ - 6
103
+ - 7
104
+ - 8
105
+ - 9
106
+ - 10
107
+ - 11
108
+ - 12
109
+ - 13
110
+ - 14
111
+ - 15
112
+ modality_keys:
113
+ - arm_l
114
+ - arm_l_finger
115
+ sin_cos_embedding_keys: null
116
+ mean_std_embedding_keys: null
117
+ action_configs:
118
+ - rep: RELATIVE
119
+ type: NON_EEF
120
+ format: DEFAULT
121
+ state_key: null
122
+ - rep: ABSOLUTE
123
+ type: NON_EEF
124
+ format: DEFAULT
125
+ state_key: null
126
+ language:
127
+ delta_indices:
128
+ - 0
129
+ modality_keys:
130
+ - annotation.human.task_description
131
+ sin_cos_embedding_keys: null
132
+ mean_std_embedding_keys: null
133
+ action_configs: null
134
+ download_cache: false
135
+ shard_size: 1024
136
+ episode_sampling_rate: 0.1
137
+ num_shards_per_epoch: 100000
138
+ override_pretraining_statistics: false
139
+ mode: single_turn
140
+ random_chop: 0.0
141
+ mock_dataset_mode: false
142
+ shuffle: true
143
+ seed: 42
144
+ multiprocessing_context: fork
145
+ allow_padding: false
146
+ subsample_ratio: 1.0
147
+ image_crop_size:
148
+ - 244
149
+ - 244
150
+ image_target_size:
151
+ - 224
152
+ - 224
153
+ video_backend: torchcodec
154
+ training:
155
+ output_dir: /home/hsaadatzi/checkpoints/run_001
156
+ experiment_name: null
157
+ max_steps: 30000
158
+ global_batch_size: 32
159
+ batch_size: null
160
+ gradient_accumulation_steps: 1
161
+ learning_rate: 0.0001
162
+ lr_scheduler_type: cosine
163
+ weight_decay: 1.0e-05
164
+ warmup_ratio: 0.05
165
+ warmup_steps: 0
166
+ max_grad_norm: 1.0
167
+ optim: adamw_torch
168
+ start_from_checkpoint: nvidia/GR00T-N1.6-3B
169
+ tf32: true
170
+ fp16: false
171
+ bf16: true
172
+ eval_bf16: true
173
+ logging_steps: 10
174
+ save_steps: 5000
175
+ save_total_limit: 5
176
+ save_vl_model: false
177
+ upload_checkpoints: false
178
+ upload_every: 1000
179
+ upload_last_n_checkpoints: 5
180
+ max_concurrent_uploads: 2
181
+ eval_strategy: 'no'
182
+ eval_steps: 500
183
+ eval_set_split_ratio: 0.1
184
+ eval_batch_size: 2
185
+ save_best_eval_metric_name: ''
186
+ save_best_eval_metric_greater_is_better: true
187
+ deepspeed_stage: 2
188
+ gradient_checkpointing: false
189
+ transformers_trust_remote_code: true
190
+ transformers_local_files_only: false
191
+ transformers_cache_dir: null
192
+ transformers_access_token: null
193
+ use_ddp: false
194
+ ddp_bucket_cap_mb: 100
195
+ num_gpus: 1
196
+ dataloader_num_workers: 8
197
+ remove_unused_columns: false
198
+ use_wandb: true
199
+ wandb_project: finetune-gr00t-n1d6
200
+ enable_profiling: false
201
+ max_retries: 3
202
+ assert_loss_less_than: null
203
+ add_rl_callback: false
204
+ enable_open_loop_eval: false
205
+ open_loop_eval_traj_ids:
206
+ - 0
207
+ open_loop_eval_steps_per_traj: 100
208
+ open_loop_eval_plot_indices: null
209
+ max_steps: 30000
210
+ save_steps: 5000
checkpoint-25000/experiment_cfg/config.yaml ADDED
@@ -0,0 +1,244 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object:gr00t.configs.base_config.Config
2
+ data: !!python/object:gr00t.configs.data.data_config.DataConfig
3
+ allow_padding: false
4
+ datasets:
5
+ - !!python/object:gr00t.configs.data.data_config.SingleDatasetConfig
6
+ dataset_paths:
7
+ - /home/hsaadatzi/set_1
8
+ dataset_type: physical_embodiment
9
+ embodiment_tag: new_embodiment
10
+ mix_ratio: 1.0
11
+ val_dataset_path: null
12
+ download_cache: false
13
+ episode_sampling_rate: 0.1
14
+ image_crop_size:
15
+ - 244
16
+ - 244
17
+ image_target_size:
18
+ - 224
19
+ - 224
20
+ mock_dataset_mode: false
21
+ modality_configs:
22
+ new_embodiment:
23
+ action: !!python/object:gr00t.data.types.ModalityConfig
24
+ action_configs:
25
+ - !!python/object:gr00t.data.types.ActionConfig
26
+ format: &id001 !!python/object/apply:gr00t.data.types.ActionFormat
27
+ - default
28
+ rep: !!python/object/apply:gr00t.data.types.ActionRepresentation
29
+ - relative
30
+ state_key: null
31
+ type: &id002 !!python/object/apply:gr00t.data.types.ActionType
32
+ - non_eef
33
+ - !!python/object:gr00t.data.types.ActionConfig
34
+ format: *id001
35
+ rep: !!python/object/apply:gr00t.data.types.ActionRepresentation
36
+ - absolute
37
+ state_key: null
38
+ type: *id002
39
+ delta_indices:
40
+ - 0
41
+ - 1
42
+ - 2
43
+ - 3
44
+ - 4
45
+ - 5
46
+ - 6
47
+ - 7
48
+ - 8
49
+ - 9
50
+ - 10
51
+ - 11
52
+ - 12
53
+ - 13
54
+ - 14
55
+ - 15
56
+ mean_std_embedding_keys: null
57
+ modality_keys:
58
+ - arm_l
59
+ - arm_l_finger
60
+ sin_cos_embedding_keys: null
61
+ language: !!python/object:gr00t.data.types.ModalityConfig
62
+ action_configs: null
63
+ delta_indices:
64
+ - 0
65
+ mean_std_embedding_keys: null
66
+ modality_keys:
67
+ - annotation.human.task_description
68
+ sin_cos_embedding_keys: null
69
+ state: !!python/object:gr00t.data.types.ModalityConfig
70
+ action_configs: null
71
+ delta_indices:
72
+ - 0
73
+ mean_std_embedding_keys: null
74
+ modality_keys:
75
+ - arm_l
76
+ - arm_l_finger
77
+ sin_cos_embedding_keys: null
78
+ video: !!python/object:gr00t.data.types.ModalityConfig
79
+ action_configs: null
80
+ delta_indices:
81
+ - 0
82
+ mean_std_embedding_keys: null
83
+ modality_keys:
84
+ - cam_high
85
+ - cam_left_wrist
86
+ - cam_side
87
+ sin_cos_embedding_keys: null
88
+ mode: single_turn
89
+ multiprocessing_context: fork
90
+ num_shards_per_epoch: 100000
91
+ override_pretraining_statistics: false
92
+ random_chop: 0.0
93
+ seed: 42
94
+ shard_size: 1024
95
+ shuffle: true
96
+ subsample_ratio: 1.0
97
+ video_backend: torchcodec
98
+ load_config_path: null
99
+ model: !!python/object:gr00t.configs.model.gr00t_n1d6.Gr00tN1d6Config
100
+ _attn_implementation_autoset: false
101
+ _attn_implementation_internal: null
102
+ _commit_hash: null
103
+ _name_or_path: ''
104
+ add_cross_attention: false
105
+ architectures: null
106
+ backbone_model_type: eagle
107
+ backbone_trainable_params_fp32: true
108
+ bad_words_ids: null
109
+ begin_suppress_tokens: null
110
+ bos_token_id: null
111
+ chunk_size_feed_forward: 0
112
+ color_jitter_params:
113
+ brightness: 0.3
114
+ contrast: 0.4
115
+ hue: 0.08
116
+ saturation: 0.5
117
+ cross_attention_hidden_size: null
118
+ decoder_start_token_id: null
119
+ diffusion_model_cfg:
120
+ attention_head_dim: 48
121
+ dropout: 0.2
122
+ final_dropout: true
123
+ interleave_self_attention: true
124
+ norm_type: ada_norm
125
+ num_attention_heads: 32
126
+ num_layers: 32
127
+ output_dim: 1024
128
+ positional_embeddings: null
129
+ diversity_penalty: 0.0
130
+ do_sample: false
131
+ eagle_collator: true
132
+ early_stopping: false
133
+ encoder_no_repeat_ngram_size: 0
134
+ eos_token_id: null
135
+ exponential_decay_length_penalty: null
136
+ finetuning_task: null
137
+ forced_bos_token_id: null
138
+ forced_eos_token_id: null
139
+ id2label:
140
+ 0: LABEL_0
141
+ 1: LABEL_1
142
+ is_decoder: false
143
+ is_encoder_decoder: false
144
+ label2id:
145
+ LABEL_0: 0
146
+ LABEL_1: 1
147
+ length_penalty: 1.0
148
+ load_bf16: false
149
+ max_length: 20
150
+ min_length: 0
151
+ model_name: nvidia/Eagle-Block2A-2B-v2
152
+ no_repeat_ngram_size: 0
153
+ num_beam_groups: 1
154
+ num_beams: 1
155
+ num_return_sequences: 1
156
+ output_attentions: false
157
+ output_hidden_states: false
158
+ output_scores: false
159
+ pad_token_id: null
160
+ prefix: null
161
+ problem_type: null
162
+ pruned_heads: {}
163
+ random_rotation_angle: 5
164
+ remove_invalid_values: false
165
+ repetition_penalty: 1.0
166
+ reproject_vision: false
167
+ return_dict: true
168
+ return_dict_in_generate: false
169
+ sep_token_id: null
170
+ state_dropout_prob: 0.0
171
+ suppress_tokens: null
172
+ task_specific_params: null
173
+ temperature: 1.0
174
+ tf_legacy_loss: false
175
+ tie_encoder_decoder: false
176
+ tie_word_embeddings: true
177
+ tokenizer_class: null
178
+ top_k: 50
179
+ top_p: 1.0
180
+ torch_dtype: null
181
+ torchscript: false
182
+ transformers_version: null
183
+ tune_diffusion_model: true
184
+ tune_llm: false
185
+ tune_projector: true
186
+ tune_visual: false
187
+ typical_p: 1.0
188
+ use_bfloat16: false
189
+ use_relative_action: true
190
+ training: !!python/object:gr00t.configs.training.training_config.TrainingConfig
191
+ add_rl_callback: false
192
+ assert_loss_less_than: null
193
+ batch_size: null
194
+ bf16: true
195
+ dataloader_num_workers: 8
196
+ ddp_bucket_cap_mb: 100
197
+ deepspeed_stage: 2
198
+ enable_open_loop_eval: false
199
+ enable_profiling: false
200
+ eval_batch_size: 2
201
+ eval_bf16: true
202
+ eval_set_split_ratio: 0.1
203
+ eval_steps: 500
204
+ eval_strategy: 'no'
205
+ experiment_name: null
206
+ fp16: false
207
+ global_batch_size: 32
208
+ gradient_accumulation_steps: 1
209
+ gradient_checkpointing: false
210
+ learning_rate: 0.0001
211
+ logging_steps: 10
212
+ lr_scheduler_type: cosine
213
+ max_concurrent_uploads: 2
214
+ max_grad_norm: 1.0
215
+ max_retries: 3
216
+ max_steps: 30000
217
+ num_gpus: 1
218
+ open_loop_eval_plot_indices: null
219
+ open_loop_eval_steps_per_traj: 100
220
+ open_loop_eval_traj_ids:
221
+ - 0
222
+ optim: adamw_torch
223
+ output_dir: /home/hsaadatzi/checkpoints/run_001
224
+ remove_unused_columns: false
225
+ save_best_eval_metric_greater_is_better: true
226
+ save_best_eval_metric_name: ''
227
+ save_steps: 5000
228
+ save_total_limit: 5
229
+ save_vl_model: false
230
+ start_from_checkpoint: nvidia/GR00T-N1.6-3B
231
+ tf32: true
232
+ transformers_access_token: null
233
+ transformers_cache_dir: null
234
+ transformers_local_files_only: false
235
+ transformers_trust_remote_code: true
236
+ upload_checkpoints: false
237
+ upload_every: 1000
238
+ upload_last_n_checkpoints: 5
239
+ use_ddp: false
240
+ use_wandb: true
241
+ wandb_project: finetune-gr00t-n1d6
242
+ warmup_ratio: 0.05
243
+ warmup_steps: 0
244
+ weight_decay: 1.0e-05
checkpoint-25000/experiment_cfg/dataset_statistics.json ADDED
@@ -0,0 +1,932 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "state": {
4
+ "arm_l": {
5
+ "min": [
6
+ -0.1006389930844307,
7
+ -0.3946418762207031,
8
+ -0.08725712448358536,
9
+ -3.053096294403076,
10
+ -2.5050132274627686,
11
+ -2.2086756229400635
12
+ ],
13
+ "max": [
14
+ 0.3803256154060364,
15
+ -0.026149729266762733,
16
+ 0.11409957706928253,
17
+ 3.0605881214141846,
18
+ 1.1357500553131104,
19
+ 2.0965044498443604
20
+ ],
21
+ "mean": [
22
+ 0.22020021080970764,
23
+ -0.18635892868041992,
24
+ 0.004393074195832014,
25
+ 0.5827129483222961,
26
+ -0.07390148937702179,
27
+ 0.26686400175094604
28
+ ],
29
+ "std": [
30
+ 0.13740186393260956,
31
+ 0.10274863988161087,
32
+ 0.03899266943335533,
33
+ 2.263795852661133,
34
+ 0.5816270112991333,
35
+ 1.5642648935317993
36
+ ],
37
+ "q01": [
38
+ -0.05724964052438736,
39
+ -0.3691798377037048,
40
+ -0.07054037600755692,
41
+ -2.571592035293579,
42
+ -1.9074896192550659,
43
+ -1.9545634984970093
44
+ ],
45
+ "q99": [
46
+ 0.3662006115913391,
47
+ -0.04128728926181793,
48
+ 0.0825900062918663,
49
+ 2.7419415187835687,
50
+ 0.9838142991065979,
51
+ 1.9541316032409668
52
+ ]
53
+ },
54
+ "arm_l_finger": {
55
+ "min": [
56
+ 0.0
57
+ ],
58
+ "max": [
59
+ 1.0
60
+ ],
61
+ "mean": [
62
+ 0.40531498193740845
63
+ ],
64
+ "std": [
65
+ 0.4911738634109497
66
+ ],
67
+ "q01": [
68
+ 0.0
69
+ ],
70
+ "q99": [
71
+ 1.0
72
+ ]
73
+ }
74
+ },
75
+ "action": {
76
+ "arm_l": {
77
+ "min": [
78
+ -0.1001979187130928,
79
+ -0.3946418762207031,
80
+ -0.08725712448358536,
81
+ -3.053096294403076,
82
+ -2.5050132274627686,
83
+ -2.2086756229400635
84
+ ],
85
+ "max": [
86
+ 0.3803256154060364,
87
+ -0.026149729266762733,
88
+ 0.11409957706928253,
89
+ 3.0605881214141846,
90
+ 1.1357500553131104,
91
+ 2.0965044498443604
92
+ ],
93
+ "mean": [
94
+ 0.2204194962978363,
95
+ -0.1862933337688446,
96
+ 0.004337905440479517,
97
+ 0.5859783291816711,
98
+ -0.07178063690662384,
99
+ 0.2697632610797882
100
+ ],
101
+ "std": [
102
+ 0.13729918003082275,
103
+ 0.10269364714622498,
104
+ 0.03899139538407326,
105
+ 2.2634758949279785,
106
+ 0.5796284675598145,
107
+ 1.5644787549972534
108
+ ],
109
+ "q01": [
110
+ -0.05701250210404396,
111
+ -0.36919168710708616,
112
+ -0.07054037600755692,
113
+ -2.5716392993927,
114
+ -1.9065781831741333,
115
+ -1.9546537256240846
116
+ ],
117
+ "q99": [
118
+ 0.3665055799484253,
119
+ -0.04128721356391907,
120
+ 0.08243336915969844,
121
+ 2.7419415187835687,
122
+ 0.9838142991065979,
123
+ 1.954293155670166
124
+ ]
125
+ },
126
+ "arm_l_finger": {
127
+ "min": [
128
+ 0.0
129
+ ],
130
+ "max": [
131
+ 1.0
132
+ ],
133
+ "mean": [
134
+ 0.4041811227798462
135
+ ],
136
+ "std": [
137
+ 0.4912165403366089
138
+ ],
139
+ "q01": [
140
+ 0.0
141
+ ],
142
+ "q99": [
143
+ 1.0
144
+ ]
145
+ }
146
+ },
147
+ "relative_action": {
148
+ "arm_l": {
149
+ "min": [
150
+ [
151
+ -0.004235490225255489,
152
+ -0.0022258460521698,
153
+ -0.0029641762375831604,
154
+ -6.11368465423584,
155
+ -1.0007593631744385,
156
+ -4.141870498657227
157
+ ],
158
+ [
159
+ -0.007333104033023119,
160
+ -0.0038002878427505493,
161
+ -0.005079630762338638,
162
+ -6.109963417053223,
163
+ -1.0420904159545898,
164
+ -4.141870498657227
165
+ ],
166
+ [
167
+ -0.010181523859500885,
168
+ -0.005613610148429871,
169
+ -0.007330911699682474,
170
+ -6.105661869049072,
171
+ -1.0420892238616943,
172
+ -4.141870498657227
173
+ ],
174
+ [
175
+ -0.013596647419035435,
176
+ -0.007086321711540222,
177
+ -0.009969491511583328,
178
+ -6.101454734802246,
179
+ -1.0416295528411865,
180
+ -4.141870498657227
181
+ ],
182
+ [
183
+ -0.016582434996962547,
184
+ -0.008699402213096619,
185
+ -0.012323632836341858,
186
+ -6.097495079040527,
187
+ -1.0410830974578857,
188
+ -4.141870498657227
189
+ ],
190
+ [
191
+ -0.01968994550406933,
192
+ -0.010646328330039978,
193
+ -0.014502190053462982,
194
+ -6.093442916870117,
195
+ -1.0447039604187012,
196
+ -4.141870498657227
197
+ ],
198
+ [
199
+ -0.02321077696979046,
200
+ -0.012145861983299255,
201
+ -0.01675165817141533,
202
+ -6.090091705322266,
203
+ -1.0529625415802002,
204
+ -4.141870498657227
205
+ ],
206
+ [
207
+ -0.026251966133713722,
208
+ -0.013671144843101501,
209
+ -0.018930215388536453,
210
+ -6.086215019226074,
211
+ -1.062105417251587,
212
+ -4.094995021820068
213
+ ],
214
+ [
215
+ -0.029011275619268417,
216
+ -0.015162065625190735,
217
+ -0.021075669676065445,
218
+ -6.0817484855651855,
219
+ -1.0701377391815186,
220
+ -4.094995021820068
221
+ ],
222
+ [
223
+ -0.030858267098665237,
224
+ -0.016241326928138733,
225
+ -0.022946612909436226,
226
+ -6.076840877532959,
227
+ -1.079089879989624,
228
+ -4.094995021820068
229
+ ],
230
+ [
231
+ -0.030858267098665237,
232
+ -0.01686413586139679,
233
+ -0.024742793291807175,
234
+ -6.073434352874756,
235
+ -1.0620386600494385,
236
+ -4.094995021820068
237
+ ],
238
+ [
239
+ -0.030665498226881027,
240
+ -0.01831965148448944,
241
+ -0.027076484635472298,
242
+ -6.071956634521484,
243
+ -1.03946053981781,
244
+ -4.094995021820068
245
+ ],
246
+ [
247
+ -0.030776353552937508,
248
+ -0.019868552684783936,
249
+ -0.029044806957244873,
250
+ -6.076470375061035,
251
+ -1.0394726991653442,
252
+ -4.094995021820068
253
+ ],
254
+ [
255
+ -0.030368803068995476,
256
+ -0.021434292197227478,
257
+ -0.030379850417375565,
258
+ -6.072749137878418,
259
+ -1.0644713640213013,
260
+ -4.094995021820068
261
+ ],
262
+ [
263
+ -0.028749702498316765,
264
+ -0.022919461131095886,
265
+ -0.03225233405828476,
266
+ -6.068447589874268,
267
+ -1.0686957836151123,
268
+ -4.094995021820068
269
+ ],
270
+ [
271
+ -0.028726020827889442,
272
+ -0.024467021226882935,
273
+ -0.03399868309497833,
274
+ -6.064240455627441,
275
+ -1.0724029541015625,
276
+ -4.094995021820068
277
+ ]
278
+ ],
279
+ "max": [
280
+ [
281
+ 0.004725892096757889,
282
+ 0.0027886033058166504,
283
+ 0.0022737495601177216,
284
+ 6.035301685333252,
285
+ 1.474894642829895,
286
+ 4.174610614776611
287
+ ],
288
+ [
289
+ 0.008718788623809814,
290
+ 0.004971235990524292,
291
+ 0.003996998071670532,
292
+ 6.030788421630859,
293
+ 1.477924108505249,
294
+ 4.174603462219238
295
+ ],
296
+ [
297
+ 0.010449156165122986,
298
+ 0.007192403078079224,
299
+ 0.005943775177001953,
300
+ 6.026132106781006,
301
+ 1.4809396266937256,
302
+ 4.1746063232421875
303
+ ],
304
+ [
305
+ 0.012407854199409485,
306
+ 0.00953701138496399,
307
+ 0.00783834233880043,
308
+ 6.024982452392578,
309
+ 1.4844553470611572,
310
+ 4.17458963394165
311
+ ],
312
+ [
313
+ 0.013578694313764572,
314
+ 0.01195332407951355,
315
+ 0.009777691215276718,
316
+ 6.027261734008789,
317
+ 1.4884464740753174,
318
+ 4.174583435058594
319
+ ],
320
+ [
321
+ 0.015893515199422836,
322
+ 0.014174491167068481,
323
+ 0.011414855718612671,
324
+ 6.0297393798828125,
325
+ 1.4888871908187866,
326
+ 4.174566268920898
327
+ ],
328
+ [
329
+ 0.018296536058187485,
330
+ 0.016482263803482056,
331
+ 0.01325889304280281,
332
+ 6.031765460968018,
333
+ 1.5922274589538574,
334
+ 4.174569129943848
335
+ ],
336
+ [
337
+ 0.020938966423273087,
338
+ 0.018898576498031616,
339
+ 0.014180425554513931,
340
+ 6.03474760055542,
341
+ 1.669804573059082,
342
+ 4.174562454223633
343
+ ],
344
+ [
345
+ 0.023059602826833725,
346
+ 0.021119743585586548,
347
+ 0.016026385128498077,
348
+ 6.036848068237305,
349
+ 1.6720974445343018,
350
+ 4.174544334411621
351
+ ],
352
+ [
353
+ 0.02528684213757515,
354
+ 0.023333728313446045,
355
+ 0.01709432154893875,
356
+ 6.045999526977539,
357
+ 1.6746065616607666,
358
+ 4.174551963806152
359
+ ],
360
+ [
361
+ 0.027646947652101517,
362
+ 0.02493080496788025,
363
+ 0.017966773360967636,
364
+ 6.054683685302734,
365
+ 1.6769334077835083,
366
+ 4.1745405197143555
367
+ ],
368
+ [
369
+ 0.02964061126112938,
370
+ 0.026324838399887085,
371
+ 0.018863672390580177,
372
+ 6.063432693481445,
373
+ 1.6792851686477661,
374
+ 4.174548149108887
375
+ ],
376
+ [
377
+ 0.03165138140320778,
378
+ 0.02792191505432129,
379
+ 0.020340880379080772,
380
+ 6.072515964508057,
381
+ 1.6815807819366455,
382
+ 4.174520492553711
383
+ ],
384
+ [
385
+ 0.033690545707941055,
386
+ 0.029246509075164795,
387
+ 0.021087855100631714,
388
+ 6.050999641418457,
389
+ 1.6830500364303589,
390
+ 4.17452335357666
391
+ ],
392
+ [
393
+ 0.03488786518573761,
394
+ 0.030564427375793457,
395
+ 0.02272876352071762,
396
+ 6.043996334075928,
397
+ 1.6830500364303589,
398
+ 4.174544334411621
399
+ ],
400
+ [
401
+ 0.03697727620601654,
402
+ 0.03216150403022766,
403
+ 0.023145724087953568,
404
+ 6.033823490142822,
405
+ 1.6830500364303589,
406
+ 4.174530029296875
407
+ ]
408
+ ],
409
+ "mean": [
410
+ [
411
+ 0.0002163744647987187,
412
+ 7.641994307050481e-05,
413
+ -8.130486094160005e-05,
414
+ 0.0034378347918391228,
415
+ 0.0022275091614574194,
416
+ 0.0030457961838692427
417
+ ],
418
+ [
419
+ 0.00043370219646021724,
420
+ 0.00015253714809659868,
421
+ -0.00016153481556102633,
422
+ 0.006807205732911825,
423
+ 0.004453099798411131,
424
+ 0.006040936801582575
425
+ ],
426
+ [
427
+ 0.0006517759175039828,
428
+ 0.00022835968411527574,
429
+ -0.00024060020223259926,
430
+ 0.009926194325089455,
431
+ 0.006668432150036097,
432
+ 0.008849866688251495
433
+ ],
434
+ [
435
+ 0.0008706587250344455,
436
+ 0.0003037923015654087,
437
+ -0.0003184233501087874,
438
+ 0.012570429593324661,
439
+ 0.008867476135492325,
440
+ 0.011301635764539242
441
+ ],
442
+ [
443
+ 0.001090367790311575,
444
+ 0.00037885637721046805,
445
+ -0.00039493260555900633,
446
+ 0.014831420965492725,
447
+ 0.011045495979487896,
448
+ 0.013464882038533688
449
+ ],
450
+ [
451
+ 0.001310946187004447,
452
+ 0.0004533863684628159,
453
+ -0.0004699206037912518,
454
+ 0.016954883933067322,
455
+ 0.013207071460783482,
456
+ 0.015529079362750053
457
+ ],
458
+ [
459
+ 0.001532378257252276,
460
+ 0.000527279160451144,
461
+ -0.0005432537291198969,
462
+ 0.01904982328414917,
463
+ 0.015351218171417713,
464
+ 0.01757732592523098
465
+ ],
466
+ [
467
+ 0.0017547266324982047,
468
+ 0.0006005460163578391,
469
+ -0.0006147850654087961,
470
+ 0.02100413106381893,
471
+ 0.01747729443013668,
472
+ 0.01952529326081276
473
+ ],
474
+ [
475
+ 0.001977890497073531,
476
+ 0.000673260074108839,
477
+ -0.000684502359945327,
478
+ 0.022886790335178375,
479
+ 0.019582683220505714,
480
+ 0.021423732861876488
481
+ ],
482
+ [
483
+ 0.0022018246818333864,
484
+ 0.0007454183069057763,
485
+ -0.000752423657104373,
486
+ 0.024584155529737473,
487
+ 0.021664537489414215,
488
+ 0.023186663165688515
489
+ ],
490
+ [
491
+ 0.0024263961240649223,
492
+ 0.0008170936489477754,
493
+ -0.0008186146151274443,
494
+ 0.026230355724692345,
495
+ 0.023726701736450195,
496
+ 0.02491825819015503
497
+ ],
498
+ [
499
+ 0.0026514939963817596,
500
+ 0.0008883167756721377,
501
+ -0.0008830438018776476,
502
+ 0.027802716940641403,
503
+ 0.025767916813492775,
504
+ 0.026602279394865036
505
+ ],
506
+ [
507
+ 0.0028768533375114202,
508
+ 0.000959044205956161,
509
+ -0.0009457518463023007,
510
+ 0.029322097077965736,
511
+ 0.027788635343313217,
512
+ 0.028254840523004532
513
+ ],
514
+ [
515
+ 0.003102144692093134,
516
+ 0.0010293236700817943,
517
+ -0.0010069032432511449,
518
+ 0.030722489580512047,
519
+ 0.029789531603455544,
520
+ 0.029825381934642792
521
+ ],
522
+ [
523
+ 0.00332716410048306,
524
+ 0.0010992122115567327,
525
+ -0.0010667059104889631,
526
+ 0.03202369809150696,
527
+ 0.03176870942115784,
528
+ 0.031330857425928116
529
+ ],
530
+ [
531
+ 0.003551631234586239,
532
+ 0.0011689665261656046,
533
+ -0.0011258666636422276,
534
+ 0.03329562768340111,
535
+ 0.033727407455444336,
536
+ 0.03282223269343376
537
+ ]
538
+ ],
539
+ "std": [
540
+ [
541
+ 0.0004819085879717022,
542
+ 0.0004134059709031135,
543
+ 0.0004326316702645272,
544
+ 0.2175336480140686,
545
+ 0.019367270171642303,
546
+ 0.15662790834903717
547
+ ],
548
+ [
549
+ 0.0009274919866584241,
550
+ 0.0007888803957030177,
551
+ 0.0008277753950096667,
552
+ 0.30294761061668396,
553
+ 0.029970292001962662,
554
+ 0.21806159615516663
555
+ ],
556
+ [
557
+ 0.0013585215201601386,
558
+ 0.0011487336596474051,
559
+ 0.0012072516838088632,
560
+ 0.36816227436065674,
561
+ 0.039736825972795486,
562
+ 0.26499849557876587
563
+ ],
564
+ [
565
+ 0.0017761035123839974,
566
+ 0.0014954146463423967,
567
+ 0.0015731505118310452,
568
+ 0.42029252648353577,
569
+ 0.04882282763719559,
570
+ 0.3023085296154022
571
+ ],
572
+ [
573
+ 0.0021811346523463726,
574
+ 0.0018304316326975822,
575
+ 0.0019274166552349925,
576
+ 0.46372589468955994,
577
+ 0.05711359530687332,
578
+ 0.3334437310695648
579
+ ],
580
+ [
581
+ 0.002574501559138298,
582
+ 0.002156357280910015,
583
+ 0.0022718843538314104,
584
+ 0.5006685853004456,
585
+ 0.06485568732023239,
586
+ 0.35988283157348633
587
+ ],
588
+ [
589
+ 0.0029594062361866236,
590
+ 0.0024752845056355,
591
+ 0.0026077041402459145,
592
+ 0.5341380834579468,
593
+ 0.07204389572143555,
594
+ 0.3841642439365387
595
+ ],
596
+ [
597
+ 0.003335252869874239,
598
+ 0.002786999335512519,
599
+ 0.0029365187510848045,
600
+ 0.5646325349807739,
601
+ 0.07874979823827744,
602
+ 0.4061633050441742
603
+ ],
604
+ [
605
+ 0.003702078713104129,
606
+ 0.0030922533478587866,
607
+ 0.0032592019997537136,
608
+ 0.5932672023773193,
609
+ 0.08501800149679184,
610
+ 0.4268217086791992
611
+ ],
612
+ [
613
+ 0.004062052350491285,
614
+ 0.003391882171854377,
615
+ 0.0035759746097028255,
616
+ 0.619925320148468,
617
+ 0.09101448208093643,
618
+ 0.4459671378135681
619
+ ],
620
+ [
621
+ 0.004415089264512062,
622
+ 0.0036871489137411118,
623
+ 0.003888334147632122,
624
+ 0.6443029642105103,
625
+ 0.09655258059501648,
626
+ 0.46360480785369873
627
+ ],
628
+ [
629
+ 0.004760424606502056,
630
+ 0.003976963460445404,
631
+ 0.004195548128336668,
632
+ 0.6667561531066895,
633
+ 0.10180620849132538,
634
+ 0.48002365231513977
635
+ ],
636
+ [
637
+ 0.005099463276565075,
638
+ 0.004263482987880707,
639
+ 0.004500573500990868,
640
+ 0.6881596446037292,
641
+ 0.10687198489904404,
642
+ 0.4957028329372406
643
+ ],
644
+ [
645
+ 0.005432277452200651,
646
+ 0.004545154515653849,
647
+ 0.004801253788173199,
648
+ 0.7077308893203735,
649
+ 0.1117762103676796,
650
+ 0.5103345513343811
651
+ ],
652
+ [
653
+ 0.005758755374699831,
654
+ 0.0048241023905575275,
655
+ 0.0050991130992770195,
656
+ 0.7262118458747864,
657
+ 0.11642031371593475,
658
+ 0.5241769552230835
659
+ ],
660
+ [
661
+ 0.00607927655801177,
662
+ 0.005098994355648756,
663
+ 0.005391774699091911,
664
+ 0.7436231374740601,
665
+ 0.12094634771347046,
666
+ 0.5373213291168213
667
+ ]
668
+ ],
669
+ "q01": [
670
+ [
671
+ -0.000783190131187439,
672
+ -0.0010995802283287049,
673
+ -0.0017063701525330544,
674
+ -0.011952216625213622,
675
+ -0.014866047129034997,
676
+ -0.026526288986206056
677
+ ],
678
+ [
679
+ -0.00144214928150177,
680
+ -0.002055450975894928,
681
+ -0.0032650306820869446,
682
+ -0.02315974235534668,
683
+ -0.03005633022636175,
684
+ -0.0502545902132988
685
+ ],
686
+ [
687
+ -0.002066373825073242,
688
+ -0.0029022130370140076,
689
+ -0.00477173924446106,
690
+ -0.03507861375808716,
691
+ -0.05649155378341675,
692
+ -0.07169771492481232
693
+ ],
694
+ [
695
+ -0.0025957997888326643,
696
+ -0.0036856690049171447,
697
+ -0.006174442358314991,
698
+ -0.04656525373458862,
699
+ -0.09204670242965221,
700
+ -0.09361163184046745
701
+ ],
702
+ [
703
+ -0.0031623896677047013,
704
+ -0.004378363490104675,
705
+ -0.0075454264879226685,
706
+ -0.057169198989868164,
707
+ -0.11983355432748795,
708
+ -0.11725302934646606
709
+ ],
710
+ [
711
+ -0.003610817492008209,
712
+ -0.005054646879434585,
713
+ -0.00880739651620388,
714
+ -0.06597569465637207,
715
+ -0.13928682267665862,
716
+ -0.1416412591934204
717
+ ],
718
+ [
719
+ -0.004038579761981964,
720
+ -0.005723421573638916,
721
+ -0.010088354349136353,
722
+ -0.07570052146911621,
723
+ -0.1531088799238205,
724
+ -0.16520346313714981
725
+ ],
726
+ [
727
+ -0.004418597668409348,
728
+ -0.006335234344005585,
729
+ -0.011348597705364227,
730
+ -0.08547067642211914,
731
+ -0.16345036029815674,
732
+ -0.1875450313091278
733
+ ],
734
+ [
735
+ -0.004836916923522949,
736
+ -0.006971227824687958,
737
+ -0.01253506913781166,
738
+ -0.09581494331359863,
739
+ -0.17206013202667236,
740
+ -0.2141501738131046
741
+ ],
742
+ [
743
+ -0.00521811842918396,
744
+ -0.007605612277984619,
745
+ -0.013763915747404099,
746
+ -0.1060959768295288,
747
+ -0.17913395166397095,
748
+ -0.24052307724952698
749
+ ],
750
+ [
751
+ -0.0055686719715595245,
752
+ -0.008255749940872192,
753
+ -0.014931758865714073,
754
+ -0.11532807350158691,
755
+ -0.1853969693183899,
756
+ -0.26605337858200073
757
+ ],
758
+ [
759
+ -0.005876349881291389,
760
+ -0.00880393385887146,
761
+ -0.01607256755232811,
762
+ -0.1264019012451172,
763
+ -0.19008098155260086,
764
+ -0.2968493103981018
765
+ ],
766
+ [
767
+ -0.006243829131126403,
768
+ -0.009374499320983887,
769
+ -0.017257198691368103,
770
+ -0.13980839014053345,
771
+ -0.1949293029308319,
772
+ -0.33139215797185895
773
+ ],
774
+ [
775
+ -0.0065585472621023655,
776
+ -0.009976940751075745,
777
+ -0.01833145134150982,
778
+ -0.15565848350524902,
779
+ -0.1987837851047516,
780
+ -0.3682512640953064
781
+ ],
782
+ [
783
+ -0.006869843453168868,
784
+ -0.010600230693817138,
785
+ -0.019422590732574463,
786
+ -0.17178049325942993,
787
+ -0.20442944765090942,
788
+ -0.41198515355587
789
+ ],
790
+ [
791
+ -0.007178685069084168,
792
+ -0.011201828718185425,
793
+ -0.020491862669587135,
794
+ -0.2114121913909912,
795
+ -0.20778784155845642,
796
+ -0.4583313864469528
797
+ ]
798
+ ],
799
+ "q99": [
800
+ [
801
+ 0.001854635775089264,
802
+ 0.001488762497901917,
803
+ 0.0011989539489150047,
804
+ 0.02952023029327404,
805
+ 0.07106825612485485,
806
+ 0.02365352749824526
807
+ ],
808
+ [
809
+ 0.0035716816782951355,
810
+ 0.0028239190578460693,
811
+ 0.0022778572514653215,
812
+ 0.057523131370544434,
813
+ 0.1271457362174989,
814
+ 0.04680264353752142
815
+ ],
816
+ [
817
+ 0.0052725705318152905,
818
+ 0.004127025604248047,
819
+ 0.0033534459513612147,
820
+ 0.084359149932862,
821
+ 0.174667456150055,
822
+ 0.06934742867946629
823
+ ],
824
+ [
825
+ 0.006923403590917587,
826
+ 0.005444197356700898,
827
+ 0.004340956285595921,
828
+ 0.11210823059082031,
829
+ 0.2171077996492391,
830
+ 0.09296035766601562
831
+ ],
832
+ [
833
+ 0.008570060133934021,
834
+ 0.006746745705604558,
835
+ 0.005261417478322983,
836
+ 0.1455139183998114,
837
+ 0.2617990970611572,
838
+ 0.11702019870281223
839
+ ],
840
+ [
841
+ 0.010160937905311584,
842
+ 0.008042958378791812,
843
+ 0.006161484867334366,
844
+ 0.1761331558227539,
845
+ 0.30157041549682617,
846
+ 0.1421416997909546
847
+ ],
848
+ [
849
+ 0.01173381507396698,
850
+ 0.009308264851570136,
851
+ 0.007075637578964233,
852
+ 0.21187005758285526,
853
+ 0.33680654495954615,
854
+ 0.1688853681087502
855
+ ],
856
+ [
857
+ 0.013314731419086456,
858
+ 0.010565415024757385,
859
+ 0.007903368212282658,
860
+ 0.25347017288208307,
861
+ 0.3655771374702461,
862
+ 0.19965922832489014
863
+ ],
864
+ [
865
+ 0.014810927212238312,
866
+ 0.011783044934272771,
867
+ 0.008784984759986413,
868
+ 0.30216019630432533,
869
+ 0.3884774807095528,
870
+ 0.237341046333313
871
+ ],
872
+ [
873
+ 0.01633507013320923,
874
+ 0.013005378395319012,
875
+ 0.00967448204755783,
876
+ 0.3683969974517822,
877
+ 0.41362667083740234,
878
+ 0.2768973708152771
879
+ ],
880
+ [
881
+ 0.017793752253055573,
882
+ 0.014202341437339783,
883
+ 0.01056361198425293,
884
+ 0.4881906509399414,
885
+ 0.43941307067871094,
886
+ 0.33497917652130127
887
+ ],
888
+ [
889
+ 0.019259430468082428,
890
+ 0.015403673946857453,
891
+ 0.011512245163321576,
892
+ 4.7528596687316895,
893
+ 0.4597476124763489,
894
+ 3.104358286857725
895
+ ],
896
+ [
897
+ 0.020647648721933365,
898
+ 0.016567468643188477,
899
+ 0.012318935245275497,
900
+ 4.813448896408085,
901
+ 0.47831791639328003,
902
+ 3.4248428344726562
903
+ ],
904
+ [
905
+ 0.02205244079232216,
906
+ 0.017773187160491963,
907
+ 0.013244111053645667,
908
+ 4.869495391845703,
909
+ 0.4969540590047836,
910
+ 3.5356242561340334
911
+ ],
912
+ [
913
+ 0.023444674909114838,
914
+ 0.01893170177936554,
915
+ 0.014308696687221597,
916
+ 4.900356526374819,
917
+ 0.5136902332305908,
918
+ 3.5841041922569286
919
+ ],
920
+ [
921
+ 0.02482401207089424,
922
+ 0.02013346850872042,
923
+ 0.015128909088671212,
924
+ 4.909006128311161,
925
+ 0.5350964790582667,
926
+ 3.6080309534072916
927
+ ]
928
+ ]
929
+ }
930
+ }
931
+ }
932
+ }
checkpoint-25000/experiment_cfg/final_model_config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "Gr00tN1d6",
3
+ "model_dtype": "bfloat16",
4
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
5
+ "backbone_model_type": "eagle",
6
+ "model_revision": null,
7
+ "tune_top_llm_layers": 4,
8
+ "backbone_embedding_dim": 2048,
9
+ "tune_llm": false,
10
+ "tune_visual": false,
11
+ "select_layer": 16,
12
+ "reproject_vision": false,
13
+ "use_flash_attention": true,
14
+ "load_bf16": true,
15
+ "collator_overwrite_image_inputs": false,
16
+ "eagle_collator": true,
17
+ "backbone_trainable_params_fp32": true,
18
+ "apply_sincos_state_encoding": true,
19
+ "use_relative_action": true,
20
+ "max_state_dim": 128,
21
+ "max_action_dim": 128,
22
+ "action_horizon": 50,
23
+ "hidden_size": 1024,
24
+ "input_embedding_dim": 1536,
25
+ "add_pos_embed": true,
26
+ "attn_dropout": 0.2,
27
+ "use_vlln": true,
28
+ "max_seq_len": 1024,
29
+ "use_alternate_vl_dit": true,
30
+ "attend_text_every_n_blocks": 2,
31
+ "diffusion_model_cfg": {
32
+ "attention_head_dim": 48,
33
+ "dropout": 0.2,
34
+ "final_dropout": true,
35
+ "interleave_self_attention": true,
36
+ "norm_type": "ada_norm",
37
+ "num_attention_heads": 32,
38
+ "num_layers": 32,
39
+ "output_dim": 1024,
40
+ "positional_embeddings": null
41
+ },
42
+ "num_inference_timesteps": 4,
43
+ "noise_beta_alpha": 1.5,
44
+ "noise_beta_beta": 1.0,
45
+ "noise_s": 0.999,
46
+ "num_timestep_buckets": 1000,
47
+ "tune_projector": true,
48
+ "tune_diffusion_model": true,
49
+ "tune_vlln": true,
50
+ "state_dropout_prob": 0.0,
51
+ "state_additive_noise_scale": 0.0,
52
+ "max_num_embodiments": 32
53
+ }
checkpoint-25000/experiment_cfg/final_processor_config.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-25000/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:205899fc2ae50691a3193c26df4ae342057745a93e9ba39e7ca73d1787ff146d
3
+ size 4990120184
checkpoint-25000/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:820d9f492f5061a2f151691f8bbccfec359c907c26b3be0b24f13539af2d612f
3
+ size 4823190320
checkpoint-25000/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-25000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98277a13a21638a13fa864fcbd321ee81b4cde38f8a44552c0c16ed22a2b76d1
3
+ size 12960193762
checkpoint-25000/processor_config.json ADDED
@@ -0,0 +1,456 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "processor_class": "Gr00tN1d6Processor",
3
+ "processor_kwargs": {
4
+ "modality_configs": {
5
+ "behavior_r1_pro": {
6
+ "video": {
7
+ "delta_indices": [
8
+ 0
9
+ ],
10
+ "modality_keys": [
11
+ "observation.images.rgb.head_256_256",
12
+ "observation.images.rgb.left_wrist_256_256",
13
+ "observation.images.rgb.right_wrist_256_256"
14
+ ],
15
+ "sin_cos_embedding_keys": null,
16
+ "mean_std_embedding_keys": null,
17
+ "action_configs": null
18
+ },
19
+ "state": {
20
+ "delta_indices": [
21
+ 0
22
+ ],
23
+ "modality_keys": [
24
+ "robot_pos",
25
+ "robot_ori_cos",
26
+ "robot_ori_sin",
27
+ "robot_2d_ori",
28
+ "robot_2d_ori_cos",
29
+ "robot_2d_ori_sin",
30
+ "robot_lin_vel",
31
+ "robot_ang_vel",
32
+ "arm_left_qpos",
33
+ "arm_left_qpos_sin",
34
+ "arm_left_qpos_cos",
35
+ "eef_left_pos",
36
+ "eef_left_quat",
37
+ "gripper_left_qpos",
38
+ "arm_right_qpos",
39
+ "arm_right_qpos_sin",
40
+ "arm_right_qpos_cos",
41
+ "eef_right_pos",
42
+ "eef_right_quat",
43
+ "gripper_right_qpos",
44
+ "trunk_qpos"
45
+ ],
46
+ "sin_cos_embedding_keys": null,
47
+ "mean_std_embedding_keys": null,
48
+ "action_configs": null
49
+ },
50
+ "action": {
51
+ "delta_indices": [
52
+ 0,
53
+ 1,
54
+ 2,
55
+ 3,
56
+ 4,
57
+ 5,
58
+ 6,
59
+ 7,
60
+ 8,
61
+ 9,
62
+ 10,
63
+ 11,
64
+ 12,
65
+ 13,
66
+ 14,
67
+ 15,
68
+ 16,
69
+ 17,
70
+ 18,
71
+ 19,
72
+ 20,
73
+ 21,
74
+ 22,
75
+ 23,
76
+ 24,
77
+ 25,
78
+ 26,
79
+ 27,
80
+ 28,
81
+ 29,
82
+ 30,
83
+ 31
84
+ ],
85
+ "modality_keys": [
86
+ "base",
87
+ "torso",
88
+ "left_arm",
89
+ "left_gripper",
90
+ "right_arm",
91
+ "right_gripper"
92
+ ],
93
+ "sin_cos_embedding_keys": null,
94
+ "mean_std_embedding_keys": null,
95
+ "action_configs": [
96
+ {
97
+ "rep": "ABSOLUTE",
98
+ "type": "NON_EEF",
99
+ "format": "DEFAULT",
100
+ "state_key": null
101
+ },
102
+ {
103
+ "rep": "RELATIVE",
104
+ "type": "NON_EEF",
105
+ "format": "DEFAULT",
106
+ "state_key": "trunk_qpos"
107
+ },
108
+ {
109
+ "rep": "RELATIVE",
110
+ "type": "NON_EEF",
111
+ "format": "DEFAULT",
112
+ "state_key": "arm_left_qpos"
113
+ },
114
+ {
115
+ "rep": "ABSOLUTE",
116
+ "type": "NON_EEF",
117
+ "format": "DEFAULT",
118
+ "state_key": null
119
+ },
120
+ {
121
+ "rep": "RELATIVE",
122
+ "type": "NON_EEF",
123
+ "format": "DEFAULT",
124
+ "state_key": "arm_right_qpos"
125
+ },
126
+ {
127
+ "rep": "ABSOLUTE",
128
+ "type": "NON_EEF",
129
+ "format": "DEFAULT",
130
+ "state_key": null
131
+ }
132
+ ]
133
+ },
134
+ "language": {
135
+ "delta_indices": [
136
+ 0
137
+ ],
138
+ "modality_keys": [
139
+ "annotation.human.coarse_action"
140
+ ],
141
+ "sin_cos_embedding_keys": null,
142
+ "mean_std_embedding_keys": null,
143
+ "action_configs": null
144
+ }
145
+ },
146
+ "gr1": {
147
+ "video": {
148
+ "delta_indices": [
149
+ 0
150
+ ],
151
+ "modality_keys": [
152
+ "ego_view_bg_crop_pad_res256_freq20"
153
+ ],
154
+ "sin_cos_embedding_keys": null,
155
+ "mean_std_embedding_keys": null,
156
+ "action_configs": null
157
+ },
158
+ "state": {
159
+ "delta_indices": [
160
+ 0
161
+ ],
162
+ "modality_keys": [
163
+ "left_arm",
164
+ "right_arm",
165
+ "left_hand",
166
+ "right_hand",
167
+ "waist"
168
+ ],
169
+ "sin_cos_embedding_keys": [
170
+ "left_arm",
171
+ "right_arm",
172
+ "left_hand",
173
+ "right_hand",
174
+ "waist"
175
+ ],
176
+ "mean_std_embedding_keys": null,
177
+ "action_configs": null
178
+ },
179
+ "action": {
180
+ "delta_indices": [
181
+ 0,
182
+ 1,
183
+ 2,
184
+ 3,
185
+ 4,
186
+ 5,
187
+ 6,
188
+ 7,
189
+ 8,
190
+ 9,
191
+ 10,
192
+ 11,
193
+ 12,
194
+ 13,
195
+ 14,
196
+ 15
197
+ ],
198
+ "modality_keys": [
199
+ "left_arm",
200
+ "right_arm",
201
+ "left_hand",
202
+ "right_hand",
203
+ "waist"
204
+ ],
205
+ "sin_cos_embedding_keys": null,
206
+ "mean_std_embedding_keys": null,
207
+ "action_configs": [
208
+ {
209
+ "rep": "RELATIVE",
210
+ "type": "NON_EEF",
211
+ "format": "DEFAULT",
212
+ "state_key": null
213
+ },
214
+ {
215
+ "rep": "RELATIVE",
216
+ "type": "NON_EEF",
217
+ "format": "DEFAULT",
218
+ "state_key": null
219
+ },
220
+ {
221
+ "rep": "RELATIVE",
222
+ "type": "NON_EEF",
223
+ "format": "DEFAULT",
224
+ "state_key": null
225
+ },
226
+ {
227
+ "rep": "RELATIVE",
228
+ "type": "NON_EEF",
229
+ "format": "DEFAULT",
230
+ "state_key": null
231
+ },
232
+ {
233
+ "rep": "ABSOLUTE",
234
+ "type": "NON_EEF",
235
+ "format": "DEFAULT",
236
+ "state_key": null
237
+ }
238
+ ]
239
+ },
240
+ "language": {
241
+ "delta_indices": [
242
+ 0
243
+ ],
244
+ "modality_keys": [
245
+ "task"
246
+ ],
247
+ "sin_cos_embedding_keys": null,
248
+ "mean_std_embedding_keys": null,
249
+ "action_configs": null
250
+ }
251
+ },
252
+ "robocasa_panda_omron": {
253
+ "video": {
254
+ "delta_indices": [
255
+ 0
256
+ ],
257
+ "modality_keys": [
258
+ "res256_image_side_0",
259
+ "res256_image_side_1",
260
+ "res256_image_wrist_0"
261
+ ],
262
+ "sin_cos_embedding_keys": null,
263
+ "mean_std_embedding_keys": null,
264
+ "action_configs": null
265
+ },
266
+ "state": {
267
+ "delta_indices": [
268
+ 0
269
+ ],
270
+ "modality_keys": [
271
+ "end_effector_position_relative",
272
+ "end_effector_rotation_relative",
273
+ "gripper_qpos",
274
+ "base_position",
275
+ "base_rotation"
276
+ ],
277
+ "sin_cos_embedding_keys": null,
278
+ "mean_std_embedding_keys": null,
279
+ "action_configs": null
280
+ },
281
+ "action": {
282
+ "delta_indices": [
283
+ 0,
284
+ 1,
285
+ 2,
286
+ 3,
287
+ 4,
288
+ 5,
289
+ 6,
290
+ 7,
291
+ 8,
292
+ 9,
293
+ 10,
294
+ 11,
295
+ 12,
296
+ 13,
297
+ 14,
298
+ 15
299
+ ],
300
+ "modality_keys": [
301
+ "end_effector_position",
302
+ "end_effector_rotation",
303
+ "gripper_close",
304
+ "base_motion",
305
+ "control_mode"
306
+ ],
307
+ "sin_cos_embedding_keys": null,
308
+ "mean_std_embedding_keys": null,
309
+ "action_configs": [
310
+ {
311
+ "rep": "ABSOLUTE",
312
+ "type": "NON_EEF",
313
+ "format": "DEFAULT",
314
+ "state_key": null
315
+ },
316
+ {
317
+ "rep": "ABSOLUTE",
318
+ "type": "NON_EEF",
319
+ "format": "DEFAULT",
320
+ "state_key": null
321
+ },
322
+ {
323
+ "rep": "ABSOLUTE",
324
+ "type": "NON_EEF",
325
+ "format": "DEFAULT",
326
+ "state_key": null
327
+ },
328
+ {
329
+ "rep": "ABSOLUTE",
330
+ "type": "NON_EEF",
331
+ "format": "DEFAULT",
332
+ "state_key": null
333
+ },
334
+ {
335
+ "rep": "ABSOLUTE",
336
+ "type": "NON_EEF",
337
+ "format": "DEFAULT",
338
+ "state_key": null
339
+ }
340
+ ]
341
+ },
342
+ "language": {
343
+ "delta_indices": [
344
+ 0
345
+ ],
346
+ "modality_keys": [
347
+ "annotation.human.action.task_description"
348
+ ],
349
+ "sin_cos_embedding_keys": null,
350
+ "mean_std_embedding_keys": null,
351
+ "action_configs": null
352
+ }
353
+ },
354
+ "new_embodiment": {
355
+ "video": {
356
+ "delta_indices": [
357
+ 0
358
+ ],
359
+ "modality_keys": [
360
+ "cam_high",
361
+ "cam_left_wrist",
362
+ "cam_side"
363
+ ],
364
+ "sin_cos_embedding_keys": null,
365
+ "mean_std_embedding_keys": null,
366
+ "action_configs": null
367
+ },
368
+ "state": {
369
+ "delta_indices": [
370
+ 0
371
+ ],
372
+ "modality_keys": [
373
+ "arm_l",
374
+ "arm_l_finger"
375
+ ],
376
+ "sin_cos_embedding_keys": null,
377
+ "mean_std_embedding_keys": null,
378
+ "action_configs": null
379
+ },
380
+ "action": {
381
+ "delta_indices": [
382
+ 0,
383
+ 1,
384
+ 2,
385
+ 3,
386
+ 4,
387
+ 5,
388
+ 6,
389
+ 7,
390
+ 8,
391
+ 9,
392
+ 10,
393
+ 11,
394
+ 12,
395
+ 13,
396
+ 14,
397
+ 15
398
+ ],
399
+ "modality_keys": [
400
+ "arm_l",
401
+ "arm_l_finger"
402
+ ],
403
+ "sin_cos_embedding_keys": null,
404
+ "mean_std_embedding_keys": null,
405
+ "action_configs": [
406
+ {
407
+ "rep": "RELATIVE",
408
+ "type": "NON_EEF",
409
+ "format": "DEFAULT",
410
+ "state_key": null
411
+ },
412
+ {
413
+ "rep": "ABSOLUTE",
414
+ "type": "NON_EEF",
415
+ "format": "DEFAULT",
416
+ "state_key": null
417
+ }
418
+ ]
419
+ },
420
+ "language": {
421
+ "delta_indices": [
422
+ 0
423
+ ],
424
+ "modality_keys": [
425
+ "annotation.human.task_description"
426
+ ],
427
+ "sin_cos_embedding_keys": null,
428
+ "mean_std_embedding_keys": null,
429
+ "action_configs": null
430
+ }
431
+ }
432
+ },
433
+ "image_crop_size": null,
434
+ "image_target_size": null,
435
+ "use_albumentations": true,
436
+ "random_rotation_angle": 5,
437
+ "color_jitter_params": {
438
+ "brightness": 0.3,
439
+ "contrast": 0.4,
440
+ "saturation": 0.5,
441
+ "hue": 0.08
442
+ },
443
+ "shortest_image_edge": 256,
444
+ "crop_fraction": 0.95,
445
+ "model_name": "nvidia/Eagle-Block2A-2B-v2",
446
+ "model_type": "eagle",
447
+ "formalize_language": true,
448
+ "max_state_dim": 128,
449
+ "max_action_dim": 128,
450
+ "max_action_horizon": 50,
451
+ "use_percentiles": false,
452
+ "clip_outliers": true,
453
+ "apply_sincos_state_encoding": true,
454
+ "use_relative_action": true
455
+ }
456
+ }
checkpoint-25000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8619b0f2660fc43d003ca55b206476057084e3ecdd73cf73f7f7ca76e6472b0b
3
+ size 14645
checkpoint-25000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a8e58e6861042066ffa018eec7f22ecc54f49d1057e922f8859691d3672674b
3
+ size 1465
checkpoint-25000/statistics.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-25000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-25000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cb687c258d2b54116bf800440037af536578cf4483214c2e9513583b8cef6a8
3
+ size 5777
checkpoint-25000/wandb_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"project": "finetune-gr00t-n1d6", "run_id": "run_001"}