Hugging Face's logo

TTaekwan
/
simul_ATQ

Safetensors
Model card Files
xet
 Community
TTaekwan commited on
Commit
efd2b81
·
verified ·
1 Parent(s): 527bffc

Upload folder using huggingface_hub

Browse files
Files changed (10) hide show
  1. s2_contact/config.json +137 -0
  2. s2_contact/experiment_cfg/metadata.json +291 -0
  3. s2_contact/model-00001-of-00002.safetensors +3 -0
  4. s2_contact/model-00002-of-00002.safetensors +3 -0
  5. s2_contact/model.safetensors.index.json +0 -0
  6. s2_contact/optimizer.pt +3 -0
  7. s2_contact/rng_state_0.pth +3 -0
  8. s2_contact/rng_state_1.pth +3 -0
  9. s2_contact/scheduler.pt +3 -0
  10. s2_contact/trainer_state.json +0 -0
s2_contact/config.json ADDED
@@ -0,0 +1,137 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "action_dim": 32,
3
+ "action_head_cfg": {
4
+ "action_dim": 32,
5
+ "action_horizon": 16,
6
+ "add_pos_embed": true,
7
+ "backbone_embedding_dim": 2048,
8
+ "conf_threshold": 0.5,
9
+ "diffusion_model_cfg": {
10
+ "attention_head_dim": 48,
11
+ "cross_attention_dim": 2048,
12
+ "dropout": 0.2,
13
+ "final_dropout": true,
14
+ "interleave_self_attention": true,
15
+ "norm_type": "ada_norm",
16
+ "num_attention_heads": 32,
17
+ "num_layers": 16,
18
+ "output_dim": 1024,
19
+ "positional_embeddings": null
20
+ },
21
+ "flare_align_layers": 12,
22
+ "flare_cos_dim": 1,
23
+ "flare_loss_lambda": 0.0,
24
+ "flevel_contact_pos_weight": null,
25
+ "flevel_level_class_weight": null,
26
+ "flevel_loss_coef": 0.1,
27
+ "flevel_readout_detach": false,
28
+ "flevel_readout_dropout": 0.1,
29
+ "flevel_readout_hidden": 512,
30
+ "grade_loss_coef": 0.0,
31
+ "grade_num_classes": 5,
32
+ "grade_num_questions": 5,
33
+ "hidden_size": 1024,
34
+ "input_embedding_dim": 1536,
35
+ "level_action_weights": [
36
+ 0.5,
37
+ 0.5
38
+ ],
39
+ "level_hold_dims": [
40
+ 6
41
+ ],
42
+ "level_ks": [
43
+ 1,
44
+ 2.5
45
+ ],
46
+ "level_sample_weight": "none",
47
+ "level_sample_weight_floor": 0.1,
48
+ "level_sample_weight_reverse": false,
49
+ "level_sum_offset": [
50
+ 0.0,
51
+ 0.0,
52
+ 0.0,
53
+ 0.15881325386741035,
54
+ 0.0,
55
+ 0.010101000370632515,
56
+ 0.0,
57
+ 0.0,
58
+ 0.0,
59
+ 0.0,
60
+ 0.0,
61
+ 0.0,
62
+ 0.0,
63
+ 0.0,
64
+ 0.0,
65
+ 0.0,
66
+ 0.0,
67
+ 0.0,
68
+ 0.0,
69
+ 0.0,
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0,
77
+ 0.0,
78
+ 0.0,
79
+ 0.0,
80
+ 0.0,
81
+ 0.0
82
+ ],
83
+ "max_action_dim": 32,
84
+ "max_state_dim": 64,
85
+ "model_dtype": "float32",
86
+ "noise_beta_alpha": 1.5,
87
+ "noise_beta_beta": 1.0,
88
+ "noise_s": 0.999,
89
+ "num_inference_timesteps": 4,
90
+ "num_target_vision_tokens": 64,
91
+ "num_timestep_buckets": 1000,
92
+ "ratio_grid": [
93
+ 1,
94
+ 2.5
95
+ ],
96
+ "ratio_loss_coef": 0.1,
97
+ "ratio_readout_detach": true,
98
+ "ratio_readout_dropout": 0.1,
99
+ "ratio_readout_hidden": 256,
100
+ "ratio_readout_mode": "ratio_current",
101
+ "ratio_target_kind": "conf",
102
+ "respect_vl_padding": true,
103
+ "tune_diffusion_model": true,
104
+ "tune_projector": true,
105
+ "use_vlln": true,
106
+ "vl_self_attention_cfg": {
107
+ "attention_head_dim": 64,
108
+ "dropout": 0.2,
109
+ "final_dropout": true,
110
+ "num_attention_heads": 32,
111
+ "num_layers": 4,
112
+ "positional_embeddings": null
113
+ }
114
+ },
115
+ "action_horizon": 16,
116
+ "architectures": [
117
+ "GR00T_N1_5_FLEVEL"
118
+ ],
119
+ "attn_implementation": null,
120
+ "backbone_cfg": {
121
+ "eagle_path": "NVEagle/eagle_er-qwen3_1_7B-Siglip2_400M_stage1_5_128gpu_er_v7_1mlp_nops",
122
+ "features_only": true,
123
+ "load_bf16": false,
124
+ "project_to_dim": null,
125
+ "reproject_vision": false,
126
+ "select_layer": 12,
127
+ "tune_llm": false,
128
+ "tune_visual": true,
129
+ "use_flash_attention": true
130
+ },
131
+ "compute_dtype": "bfloat16",
132
+ "hidden_size": 2048,
133
+ "model_dtype": "float32",
134
+ "model_type": "gr00t_n1_5_flevel",
135
+ "torch_dtype": "bfloat16",
136
+ "transformers_version": "4.51.3"
137
+ }
s2_contact/experiment_cfg/metadata.json ADDED
@@ -0,0 +1,291 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "libero": {
3
+ "statistics": {
4
+ "state": {
5
+ "eef_pos_absolute": {
6
+ "max": [
7
+ 0.21031762659549713,
8
+ 0.39128610491752625,
9
+ 1.3660105466842651
10
+ ],
11
+ "min": [
12
+ -0.4828203022480011,
13
+ -0.3255046010017395,
14
+ 0.008128180168569088
15
+ ],
16
+ "mean": [
17
+ -0.046518828719854355,
18
+ 0.034408919513225555,
19
+ 0.7645694613456726
20
+ ],
21
+ "std": [
22
+ 0.10494082421064377,
23
+ 0.1517619788646698,
24
+ 0.3785194456577301
25
+ ],
26
+ "q01": [
27
+ -0.39912487387657164,
28
+ -0.26883512616157534,
29
+ 0.03826696485280991
30
+ ],
31
+ "q99": [
32
+ 0.13556506335735302,
33
+ 0.33566484451293943,
34
+ 1.2706659078598022
35
+ ]
36
+ },
37
+ "eef_rot_absolute": {
38
+ "max": [
39
+ 3.6714255809783936,
40
+ 3.560650587081909,
41
+ 1.386339545249939
42
+ ],
43
+ "min": [
44
+ 0.35277295112609863,
45
+ -3.641430377960205,
46
+ -1.842738389968872
47
+ ],
48
+ "mean": [
49
+ 2.9716713428497314,
50
+ -0.2204727977514267,
51
+ -0.12557993829250336
52
+ ],
53
+ "std": [
54
+ 0.3442671298980713,
55
+ 0.9068173766136169,
56
+ 0.32538288831710815
57
+ ],
58
+ "q01": [
59
+ 1.5089580535888671,
60
+ -2.719791135787964,
61
+ -1.08050856590271
62
+ ],
63
+ "q99": [
64
+ 3.2773460674285886,
65
+ 2.406109695434569,
66
+ 0.59776972055435
67
+ ]
68
+ },
69
+ "gripper_close": {
70
+ "max": [
71
+ 0.04233968257904053,
72
+ 0.0013633022317662835
73
+ ],
74
+ "min": [
75
+ -0.0013586411951109767,
76
+ -0.042040832340717316
77
+ ],
78
+ "mean": [
79
+ 0.026915358379483223,
80
+ -0.027191326022148132
81
+ ],
82
+ "std": [
83
+ 0.014175750315189362,
84
+ 0.014058776199817657
85
+ ],
86
+ "q01": [
87
+ 0.0017423711018636821,
88
+ -0.04002561450004578
89
+ ],
90
+ "q99": [
91
+ 0.04031316146254539,
92
+ -0.0017781084170565016
93
+ ]
94
+ }
95
+ },
96
+ "action": {
97
+ "eef_pos_delta": {
98
+ "max": [
99
+ 0.9375,
100
+ 0.9375,
101
+ 0.9375
102
+ ],
103
+ "min": [
104
+ -0.9375,
105
+ -0.9375,
106
+ -0.9375
107
+ ],
108
+ "mean": [
109
+ 0.06278152763843536,
110
+ 0.08684158325195312,
111
+ -0.0903734639286995
112
+ ],
113
+ "std": [
114
+ 0.33551836013793945,
115
+ 0.37847793102264404,
116
+ 0.4446770250797272
117
+ ],
118
+ "q01": [
119
+ -0.7044642567634583,
120
+ -0.8008928298950195,
121
+ -0.9375
122
+ ],
123
+ "q99": [
124
+ 0.9375,
125
+ 0.8678571581840515,
126
+ 0.9375
127
+ ]
128
+ },
129
+ "eef_rot_delta": {
130
+ "max": [
131
+ 0.3557142913341522,
132
+ 0.375,
133
+ 0.375
134
+ ],
135
+ "min": [
136
+ -0.2582142949104309,
137
+ -0.375,
138
+ -0.3675000071525574
139
+ ],
140
+ "mean": [
141
+ 0.0005407554563134909,
142
+ 0.005643464159220457,
143
+ -0.005229106638580561
144
+ ],
145
+ "std": [
146
+ 0.03924214467406273,
147
+ 0.06341660022735596,
148
+ 0.07792268693447113
149
+ ],
150
+ "q01": [
151
+ -0.11464285850524902,
152
+ -0.1639285683631897,
153
+ -0.2239285707473755
154
+ ],
155
+ "q99": [
156
+ 0.13178572058677673,
157
+ 0.19285714626312256,
158
+ 0.335357129573822
159
+ ]
160
+ },
161
+ "gripper_close": {
162
+ "max": [
163
+ 1.0
164
+ ],
165
+ "min": [
166
+ -1.0
167
+ ],
168
+ "mean": [
169
+ -0.0496407225728035
170
+ ],
171
+ "std": [
172
+ 1.000144362449646
173
+ ],
174
+ "q01": [
175
+ -1.0
176
+ ],
177
+ "q99": [
178
+ 1.0
179
+ ]
180
+ },
181
+ "ratio_label": {
182
+ "max": [
183
+ 1.0,
184
+ 1.0
185
+ ],
186
+ "min": [
187
+ 0.0,
188
+ 0.0
189
+ ],
190
+ "mean": [
191
+ 0.0,
192
+ 0.0
193
+ ],
194
+ "std": [
195
+ 1.0,
196
+ 1.0
197
+ ],
198
+ "q01": [
199
+ 0.0,
200
+ 0.0
201
+ ],
202
+ "q99": [
203
+ 1.0,
204
+ 1.0
205
+ ]
206
+ }
207
+ }
208
+ },
209
+ "modalities": {
210
+ "video": {
211
+ "front_view": {
212
+ "resolution": [
213
+ 256,
214
+ 256
215
+ ],
216
+ "channels": 3,
217
+ "fps": 10.0
218
+ },
219
+ "left_wrist_view": {
220
+ "resolution": [
221
+ 256,
222
+ 256
223
+ ],
224
+ "channels": 3,
225
+ "fps": 10.0
226
+ }
227
+ },
228
+ "state": {
229
+ "eef_pos_absolute": {
230
+ "absolute": true,
231
+ "rotation_type": null,
232
+ "shape": [
233
+ 3
234
+ ],
235
+ "continuous": true
236
+ },
237
+ "eef_rot_absolute": {
238
+ "absolute": true,
239
+ "rotation_type": "euler_angles_rpy",
240
+ "shape": [
241
+ 3
242
+ ],
243
+ "continuous": true
244
+ },
245
+ "gripper_close": {
246
+ "absolute": true,
247
+ "rotation_type": null,
248
+ "shape": [
249
+ 2
250
+ ],
251
+ "continuous": true
252
+ }
253
+ },
254
+ "action": {
255
+ "eef_pos_delta": {
256
+ "absolute": false,
257
+ "rotation_type": null,
258
+ "shape": [
259
+ 3
260
+ ],
261
+ "continuous": true
262
+ },
263
+ "eef_rot_delta": {
264
+ "absolute": false,
265
+ "rotation_type": "euler_angles_rpy",
266
+ "shape": [
267
+ 3
268
+ ],
269
+ "continuous": true
270
+ },
271
+ "gripper_close": {
272
+ "absolute": true,
273
+ "rotation_type": null,
274
+ "shape": [
275
+ 1
276
+ ],
277
+ "continuous": true
278
+ },
279
+ "ratio_label": {
280
+ "absolute": true,
281
+ "rotation_type": null,
282
+ "shape": [
283
+ 2
284
+ ],
285
+ "continuous": true
286
+ }
287
+ }
288
+ },
289
+ "embodiment_tag": "libero"
290
+ }
291
+ }
s2_contact/model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4349343715239806637a0c2d90c3f23564ea103704c2aea53e6e40f08bbbea1a
3
+ size 4999367032
s2_contact/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8f7561b866b2ec42baa22d1af664e2f12f716150d1ada6bc2014c767cec587c
3
+ size 3641628572
s2_contact/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
s2_contact/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b986dfd0d95581a55474d9a2961775cf8eae392285f3c8049ca459fa986a3bbd
3
+ size 10660580663
s2_contact/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f17e8e9dbb14791b7812de40847a8367d4ec0f36585f306d9d088d768ac975a0
3
+ size 14917
s2_contact/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bed3f09df06015f48262d6b32e8212bd26caacdd3e2308075d00083900128091
3
+ size 14917
s2_contact/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96b418fbb509f96d95ad52dd99d48eeb782a306175bb55d2661eb9b3585613ba
3
+ size 1465
s2_contact/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff