Hugging Face's logo

qjuu
/
phy_3D_unmovable_15000

Safetensors
gr00t_n1_5
Model card Files
xet
 Community
qjuu commited on
Commit
9cfdad9
·
verified ·
1 Parent(s): 28060cb

Upload folder using huggingface_hub

Browse files
Files changed (10) hide show
  1. config.json +64 -0
  2. experiment_cfg/metadata.json +871 -0
  3. model-00001-of-00002.safetensors +3 -0
  4. model-00002-of-00002.safetensors +3 -0
  5. model.safetensors.index.json +0 -0
  6. optimizer.pt +3 -0
  7. rng_state_0.pth +3 -0
  8. rng_state_1.pth +3 -0
  9. scheduler.pt +3 -0
  10. trainer_state.json +0 -0
config.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "diffusion_model_cfg": {
9
+ "attention_head_dim": 48,
10
+ "cross_attention_dim": 2048,
11
+ "dropout": 0.2,
12
+ "final_dropout": true,
13
+ "interleave_self_attention": true,
14
+ "norm_type": "ada_norm",
15
+ "num_attention_heads": 32,
16
+ "num_layers": 16,
17
+ "output_dim": 1024,
18
+ "positional_embeddings": null
19
+ },
20
+ "hidden_size": 1024,
21
+ "input_embedding_dim": 1536,
22
+ "max_action_dim": 32,
23
+ "max_state_dim": 64,
24
+ "model_dtype": "float32",
25
+ "noise_beta_alpha": 1.5,
26
+ "noise_beta_beta": 1.0,
27
+ "noise_s": 0.999,
28
+ "num_inference_timesteps": 4,
29
+ "num_target_vision_tokens": 32,
30
+ "num_timestep_buckets": 1000,
31
+ "tune_diffusion_model": true,
32
+ "tune_projector": true,
33
+ "use_vlln": true,
34
+ "vl_self_attention_cfg": {
35
+ "attention_head_dim": 64,
36
+ "dropout": 0.2,
37
+ "final_dropout": true,
38
+ "num_attention_heads": 32,
39
+ "num_layers": 4,
40
+ "positional_embeddings": null
41
+ }
42
+ },
43
+ "action_horizon": 16,
44
+ "architectures": [
45
+ "GR00T_N1_5"
46
+ ],
47
+ "attn_implementation": null,
48
+ "backbone_cfg": {
49
+ "eagle_path": "NVEagle/eagle_er-qwen3_1_7B-Siglip2_400M_stage1_5_128gpu_er_v7_1mlp_nops",
50
+ "load_bf16": false,
51
+ "project_to_dim": null,
52
+ "reproject_vision": false,
53
+ "select_layer": 12,
54
+ "tune_llm": false,
55
+ "tune_visual": true,
56
+ "use_flash_attention": true
57
+ },
58
+ "compute_dtype": "bfloat16",
59
+ "hidden_size": 2048,
60
+ "model_dtype": "float32",
61
+ "model_type": "gr00t_n1_5",
62
+ "torch_dtype": "bfloat16",
63
+ "transformers_version": "4.51.3"
64
+ }
experiment_cfg/metadata.json ADDED
@@ -0,0 +1,871 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "new_embodiment": {
3
+ "statistics": {
4
+ "state": {
5
+ "joint_0": {
6
+ "max": [
7
+ 0.3826597332954407
8
+ ],
9
+ "min": [
10
+ -0.3673768639564514
11
+ ],
12
+ "mean": [
13
+ -0.030417965725064278
14
+ ],
15
+ "std": [
16
+ 0.16984210908412933
17
+ ],
18
+ "q01": [
19
+ -0.30610814690589905
20
+ ],
21
+ "q99": [
22
+ 0.33524367213249207
23
+ ]
24
+ },
25
+ "joint_1": {
26
+ "max": [
27
+ 1.4679299592971802
28
+ ],
29
+ "min": [
30
+ -0.20288996398448944
31
+ ],
32
+ "mean": [
33
+ 0.5098796486854553
34
+ ],
35
+ "std": [
36
+ 0.3771459460258484
37
+ ],
38
+ "q01": [
39
+ -0.16899843513965607
40
+ ],
41
+ "q99": [
42
+ 1.344241738319397
43
+ ]
44
+ },
45
+ "joint_2": {
46
+ "max": [
47
+ 0.3117769658565521
48
+ ],
49
+ "min": [
50
+ -0.4048061668872833
51
+ ],
52
+ "mean": [
53
+ -0.0701102688908577
54
+ ],
55
+ "std": [
56
+ 0.1704546958208084
57
+ ],
58
+ "q01": [
59
+ -0.34388795495033264
60
+ ],
61
+ "q99": [
62
+ 0.24998249113559723
63
+ ]
64
+ },
65
+ "joint_3": {
66
+ "max": [
67
+ -0.3843629062175751
68
+ ],
69
+ "min": [
70
+ -2.5446574687957764
71
+ ],
72
+ "mean": [
73
+ -1.713296890258789
74
+ ],
75
+ "std": [
76
+ 0.5400778651237488
77
+ ],
78
+ "q01": [
79
+ -2.4729747772216797
80
+ ],
81
+ "q99": [
82
+ -0.4905872941017151
83
+ ]
84
+ },
85
+ "joint_4": {
86
+ "max": [
87
+ 0.6149448752403259
88
+ ],
89
+ "min": [
90
+ -0.44743144512176514
91
+ ],
92
+ "mean": [
93
+ -0.020573299378156662
94
+ ],
95
+ "std": [
96
+ 0.12101220339536667
97
+ ],
98
+ "q01": [
99
+ -0.26021847128868103
100
+ ],
101
+ "q99": [
102
+ 0.39621207118034363
103
+ ]
104
+ },
105
+ "joint_5": {
106
+ "max": [
107
+ 3.012094020843506
108
+ ],
109
+ "min": [
110
+ 1.16974675655365
111
+ ],
112
+ "mean": [
113
+ 2.1807122230529785
114
+ ],
115
+ "std": [
116
+ 0.32073017954826355
117
+ ],
118
+ "q01": [
119
+ 1.5115313529968262
120
+ ],
121
+ "q99": [
122
+ 2.812758684158325
123
+ ]
124
+ },
125
+ "joint_6": {
126
+ "max": [
127
+ 1.287500023841858
128
+ ],
129
+ "min": [
130
+ -0.5371313691139221
131
+ ],
132
+ "mean": [
133
+ 0.2976349890232086
134
+ ],
135
+ "std": [
136
+ 0.35320037603378296
137
+ ],
138
+ "q01": [
139
+ -0.4252602756023407
140
+ ],
141
+ "q99": [
142
+ 1.0496286153793335
143
+ ]
144
+ },
145
+ "eef_pos": {
146
+ "max": [
147
+ 0.1097615659236908,
148
+ 0.3386007845401764,
149
+ 0.4241220951080322
150
+ ],
151
+ "min": [
152
+ -0.24081063270568848,
153
+ -0.330623060464859,
154
+ 0.04072839021682739
155
+ ],
156
+ "mean": [
157
+ -0.060975126922130585,
158
+ -0.0604061633348465,
159
+ 0.22619883716106415
160
+ ],
161
+ "std": [
162
+ 0.09168554097414017,
163
+ 0.19939903914928436,
164
+ 0.08931832760572433
165
+ ],
166
+ "q01": [
167
+ -0.21294596791267395,
168
+ -0.31221282482147217,
169
+ 0.0481719970703125
170
+ ],
171
+ "q99": [
172
+ 0.09438377618789673,
173
+ 0.3007397949695587,
174
+ 0.3808176517486572
175
+ ]
176
+ },
177
+ "eef_linvel": {
178
+ "max": [
179
+ 0.04622548818588257,
180
+ 0.05074739456176758,
181
+ 0.05275845527648926
182
+ ],
183
+ "min": [
184
+ -0.03892272710800171,
185
+ -0.048753321170806885,
186
+ -0.05950927734375
187
+ ],
188
+ "mean": [
189
+ 0.00751082319766283,
190
+ 0.007268555928021669,
191
+ -0.005539536010473967
192
+ ],
193
+ "std": [
194
+ 0.010825020261108875,
195
+ 0.02849130518734455,
196
+ 0.026186544448137283
197
+ ],
198
+ "q01": [
199
+ -0.016816556453704834,
200
+ -0.048038363456726074,
201
+ -0.04946470260620117
202
+ ],
203
+ "q99": [
204
+ 0.037043746560811996,
205
+ 0.04871770739555359,
206
+ 0.04971742630004883
207
+ ]
208
+ },
209
+ "gripper_left": {
210
+ "max": [
211
+ 0.04000384360551834
212
+ ],
213
+ "min": [
214
+ 0.001576132606714964
215
+ ],
216
+ "mean": [
217
+ 0.01619638316333294
218
+ ],
219
+ "std": [
220
+ 0.017733288928866386
221
+ ],
222
+ "q01": [
223
+ 0.0016673837089911103
224
+ ],
225
+ "q99": [
226
+ 0.040000226348638535
227
+ ]
228
+ },
229
+ "gripper_right": {
230
+ "max": [
231
+ -0.0010853641433641315
232
+ ],
233
+ "min": [
234
+ -0.040344689041376114
235
+ ],
236
+ "mean": [
237
+ -0.01626543700695038
238
+ ],
239
+ "std": [
240
+ 0.017734628170728683
241
+ ],
242
+ "q01": [
243
+ -0.04000673070549965
244
+ ],
245
+ "q99": [
246
+ -0.0016642330447211862
247
+ ]
248
+ },
249
+ "bowl_pos": {
250
+ "max": [
251
+ 0.10772779583930969,
252
+ 0.3299146592617035,
253
+ 0.3862287998199463
254
+ ],
255
+ "min": [
256
+ -0.20778866112232208,
257
+ -0.28524330258369446,
258
+ 0.020979225635528564
259
+ ],
260
+ "mean": [
261
+ -0.04757988080382347,
262
+ -0.075005441904068,
263
+ 0.1378314197063446
264
+ ],
265
+ "std": [
266
+ 0.07683678716421127,
267
+ 0.21532154083251953,
268
+ 0.11046896874904633
269
+ ],
270
+ "q01": [
271
+ -0.19012489914894104,
272
+ -0.27511337399482727,
273
+ 0.023236870765686035
274
+ ],
275
+ "q99": [
276
+ 0.09168406575918198,
277
+ 0.31172406673431396,
278
+ 0.3371703624725342
279
+ ]
280
+ },
281
+ "milk_pos": {
282
+ "max": [
283
+ 0.01401741523295641,
284
+ 0.014115721918642521,
285
+ 0.09500002861022949
286
+ ],
287
+ "min": [
288
+ -0.014748563058674335,
289
+ -0.015016083605587482,
290
+ 0.01711934804916382
291
+ ],
292
+ "mean": [
293
+ -0.0011339024640619755,
294
+ -0.00014643957547377795,
295
+ 0.02061500772833824
296
+ ],
297
+ "std": [
298
+ 0.00749160535633564,
299
+ 0.009583552367985249,
300
+ 0.0036885940935462713
301
+ ],
302
+ "q01": [
303
+ -0.014748540706932545,
304
+ -0.015016083605587482,
305
+ 0.02040278911590576
306
+ ],
307
+ "q99": [
308
+ 0.01401741523295641,
309
+ 0.014092579483985901,
310
+ 0.020403146743774414
311
+ ]
312
+ },
313
+ "milk_linvel": {
314
+ "max": [
315
+ 3.245210677960131e-07,
316
+ 0.0001294492685701698,
317
+ 0.08653184026479721
318
+ ],
319
+ "min": [
320
+ -2.3190671605277657e-10,
321
+ -0.00046420079888775945,
322
+ -0.9737263917922974
323
+ ],
324
+ "mean": [
325
+ 4.961245303469752e-10,
326
+ -6.100921723373176e-07,
327
+ -0.0018320458475500345
328
+ ],
329
+ "std": [
330
+ 1.206004540676986e-08,
331
+ 1.816680196498055e-05,
332
+ 0.04032781720161438
333
+ ],
334
+ "q01": [
335
+ 0.0,
336
+ -3.2444091857541935e-08,
337
+ 2.354307070476236e-14
338
+ ],
339
+ "q99": [
340
+ 3.5361001778609624e-13,
341
+ 3.1892388990658166e-18,
342
+ 9.165248229692224e-06
343
+ ]
344
+ },
345
+ "orange_juice_pos": {
346
+ "max": [
347
+ -0.3766312301158905,
348
+ -0.5261257886886597,
349
+ 0.09500002861022949
350
+ ],
351
+ "min": [
352
+ -0.4233570694923401,
353
+ -0.574201762676239,
354
+ 0.0947427749633789
355
+ ],
356
+ "mean": [
357
+ -0.40074363350868225,
358
+ -0.5527758002281189,
359
+ 0.09473027288913727
360
+ ],
361
+ "std": [
362
+ 0.014323083683848381,
363
+ 0.014338647946715355,
364
+ 1.6110427168314345e-05
365
+ ],
366
+ "q01": [
367
+ -0.4233570694923401,
368
+ -0.574201762676239,
369
+ 0.0947427749633789
370
+ ],
371
+ "q99": [
372
+ -0.3766312301158905,
373
+ -0.5261257886886597,
374
+ 0.0947464108467102
375
+ ]
376
+ },
377
+ "orange_juice_linvel": {
378
+ "max": [
379
+ 2.8525255046929487e-08,
380
+ 4.213005922792945e-06,
381
+ 0.00023798071197234094
382
+ ],
383
+ "min": [
384
+ -4.649847618054537e-09,
385
+ -3.030943105386541e-07,
386
+ -1.3421995390672237e-06
387
+ ],
388
+ "mean": [
389
+ 3.356698946577197e-11,
390
+ -4.984679335962028e-11,
391
+ 8.814092211650859e-08
392
+ ],
393
+ "std": [
394
+ 1.0565230912362722e-09,
395
+ 3.305594020730496e-08,
396
+ 2.6860273010242963e-06
397
+ ],
398
+ "q01": [
399
+ -5.346723955370952e-15,
400
+ -7.023848863640936e-11,
401
+ -1.669434345809035e-14
402
+ ],
403
+ "q99": [
404
+ -5.005931366350617e-16,
405
+ 1.3776986651926459e-15,
406
+ 1.1035419675503935e-14
407
+ ]
408
+ },
409
+ "wine_bottle_pos": {
410
+ "max": [
411
+ -0.37526220083236694,
412
+ 0.5732696056365967,
413
+ 0.09500002861022949
414
+ ],
415
+ "min": [
416
+ -0.4237871766090393,
417
+ 0.5261248350143433,
418
+ 0.02306067943572998
419
+ ],
420
+ "mean": [
421
+ -0.3984052240848541,
422
+ 0.5496660470962524,
423
+ 0.023646052926778793
424
+ ],
425
+ "std": [
426
+ 0.014516881667077541,
427
+ 0.013536219485104084,
428
+ 0.003512189956381917
429
+ ],
430
+ "q01": [
431
+ -0.4237840175628662,
432
+ 0.5261284708976746,
433
+ 0.023441553115844727
434
+ ],
435
+ "q99": [
436
+ -0.37526220083236694,
437
+ 0.5732696056365967,
438
+ 0.023441553115844727
439
+ ]
440
+ },
441
+ "wine_bottle_linvel": {
442
+ "max": [
443
+ 4.129619992454536e-05,
444
+ 5.8732741308631375e-05,
445
+ 0.07300306856632233
446
+ ],
447
+ "min": [
448
+ -6.503009936098536e-11,
449
+ -2.0154789126536343e-06,
450
+ -0.9524779319763184
451
+ ],
452
+ "mean": [
453
+ 1.0968187780235894e-07,
454
+ 8.673613649534673e-08,
455
+ -0.0018581256736069918
456
+ ],
457
+ "std": [
458
+ 1.8827921621777932e-06,
459
+ 2.183039214287419e-06,
460
+ 0.039509475231170654
461
+ ],
462
+ "q01": [
463
+ -1.5614904524098533e-18,
464
+ -7.582908397905036e-19,
465
+ 2.6056893391557777e-15
466
+ ],
467
+ "q99": [
468
+ 1.9964556227026202e-18,
469
+ 3.718912216990419e-18,
470
+ 2.959924446928157e-15
471
+ ]
472
+ },
473
+ "timestamp": {
474
+ "max": [
475
+ 43.400001525878906
476
+ ],
477
+ "min": [
478
+ 0.0
479
+ ],
480
+ "mean": [
481
+ 18.45766258239746
482
+ ],
483
+ "std": [
484
+ 10.821619987487793
485
+ ],
486
+ "q01": [
487
+ 0.3499999940395355
488
+ ],
489
+ "q99": [
490
+ 39.400001525878906
491
+ ]
492
+ }
493
+ },
494
+ "action": {
495
+ "x": {
496
+ "max": [
497
+ 0.1875
498
+ ],
499
+ "min": [
500
+ -0.1703571379184723
501
+ ],
502
+ "mean": [
503
+ 0.032896749675273895
504
+ ],
505
+ "std": [
506
+ 0.04738948866724968
507
+ ],
508
+ "q01": [
509
+ -0.08517856895923615
510
+ ],
511
+ "q99": [
512
+ 0.15535713732242584
513
+ ]
514
+ },
515
+ "y": {
516
+ "max": [
517
+ 0.1875
518
+ ],
519
+ "min": [
520
+ -0.1875
521
+ ],
522
+ "mean": [
523
+ 0.02806279808282852
524
+ ],
525
+ "std": [
526
+ 0.11209765821695328
527
+ ],
528
+ "q01": [
529
+ -0.1875
530
+ ],
531
+ "q99": [
532
+ 0.1875
533
+ ]
534
+ },
535
+ "z": {
536
+ "max": [
537
+ 0.1875
538
+ ],
539
+ "min": [
540
+ -0.1875
541
+ ],
542
+ "mean": [
543
+ -0.017642466351389885
544
+ ],
545
+ "std": [
546
+ 0.0998985692858696
547
+ ],
548
+ "q01": [
549
+ -0.1875
550
+ ],
551
+ "q99": [
552
+ 0.1875
553
+ ]
554
+ },
555
+ "roll": {
556
+ "max": [
557
+ 0.07007142901420593
558
+ ],
559
+ "min": [
560
+ -0.033642858266830444
561
+ ],
562
+ "mean": [
563
+ -0.0005983010050840676
564
+ ],
565
+ "std": [
566
+ 0.008408918976783752
567
+ ],
568
+ "q01": [
569
+ -0.019500000402331352
570
+ ],
571
+ "q99": [
572
+ 0.03257142752408981
573
+ ]
574
+ },
575
+ "pitch": {
576
+ "max": [
577
+ 0.049714285880327225
578
+ ],
579
+ "min": [
580
+ -0.07157143205404282
581
+ ],
582
+ "mean": [
583
+ 0.002629962284117937
584
+ ],
585
+ "std": [
586
+ 0.008300474844872952
587
+ ],
588
+ "q01": [
589
+ -0.015642857179045677
590
+ ],
591
+ "q99": [
592
+ 0.027214284986257553
593
+ ]
594
+ },
595
+ "yaw": {
596
+ "max": [
597
+ 0.05057142674922943
598
+ ],
599
+ "min": [
600
+ -0.06921428442001343
601
+ ],
602
+ "mean": [
603
+ 0.005108598154038191
604
+ ],
605
+ "std": [
606
+ 0.012358150444924831
607
+ ],
608
+ "q01": [
609
+ -0.038357142359018326
610
+ ],
611
+ "q99": [
612
+ 0.03449999913573265
613
+ ]
614
+ },
615
+ "gripper": {
616
+ "max": [
617
+ 1.0
618
+ ],
619
+ "min": [
620
+ -1.0
621
+ ],
622
+ "mean": [
623
+ 0.2599111795425415
624
+ ],
625
+ "std": [
626
+ 0.9655139446258545
627
+ ],
628
+ "q01": [
629
+ -1.0
630
+ ],
631
+ "q99": [
632
+ 1.0
633
+ ]
634
+ }
635
+ }
636
+ },
637
+ "modalities": {
638
+ "video": {
639
+ "image": {
640
+ "resolution": [
641
+ 512,
642
+ 512
643
+ ],
644
+ "channels": 3,
645
+ "fps": 20.0
646
+ },
647
+ "wrist_image": {
648
+ "resolution": [
649
+ 512,
650
+ 512
651
+ ],
652
+ "channels": 3,
653
+ "fps": 20.0
654
+ }
655
+ },
656
+ "state": {
657
+ "joint_0": {
658
+ "absolute": true,
659
+ "rotation_type": null,
660
+ "shape": [
661
+ 1
662
+ ],
663
+ "continuous": true
664
+ },
665
+ "joint_1": {
666
+ "absolute": true,
667
+ "rotation_type": null,
668
+ "shape": [
669
+ 1
670
+ ],
671
+ "continuous": true
672
+ },
673
+ "joint_2": {
674
+ "absolute": true,
675
+ "rotation_type": null,
676
+ "shape": [
677
+ 1
678
+ ],
679
+ "continuous": true
680
+ },
681
+ "joint_3": {
682
+ "absolute": true,
683
+ "rotation_type": null,
684
+ "shape": [
685
+ 1
686
+ ],
687
+ "continuous": true
688
+ },
689
+ "joint_4": {
690
+ "absolute": true,
691
+ "rotation_type": null,
692
+ "shape": [
693
+ 1
694
+ ],
695
+ "continuous": true
696
+ },
697
+ "joint_5": {
698
+ "absolute": true,
699
+ "rotation_type": null,
700
+ "shape": [
701
+ 1
702
+ ],
703
+ "continuous": true
704
+ },
705
+ "joint_6": {
706
+ "absolute": true,
707
+ "rotation_type": null,
708
+ "shape": [
709
+ 1
710
+ ],
711
+ "continuous": true
712
+ },
713
+ "eef_pos": {
714
+ "absolute": true,
715
+ "rotation_type": null,
716
+ "shape": [
717
+ 3
718
+ ],
719
+ "continuous": true
720
+ },
721
+ "eef_linvel": {
722
+ "absolute": false,
723
+ "rotation_type": null,
724
+ "shape": [
725
+ 3
726
+ ],
727
+ "continuous": true
728
+ },
729
+ "gripper_left": {
730
+ "absolute": true,
731
+ "rotation_type": null,
732
+ "shape": [
733
+ 1
734
+ ],
735
+ "continuous": true
736
+ },
737
+ "gripper_right": {
738
+ "absolute": true,
739
+ "rotation_type": null,
740
+ "shape": [
741
+ 1
742
+ ],
743
+ "continuous": true
744
+ },
745
+ "bowl_pos": {
746
+ "absolute": true,
747
+ "rotation_type": null,
748
+ "shape": [
749
+ 3
750
+ ],
751
+ "continuous": true
752
+ },
753
+ "milk_pos": {
754
+ "absolute": true,
755
+ "rotation_type": null,
756
+ "shape": [
757
+ 3
758
+ ],
759
+ "continuous": true
760
+ },
761
+ "milk_linvel": {
762
+ "absolute": false,
763
+ "rotation_type": null,
764
+ "shape": [
765
+ 3
766
+ ],
767
+ "continuous": true
768
+ },
769
+ "orange_juice_pos": {
770
+ "absolute": true,
771
+ "rotation_type": null,
772
+ "shape": [
773
+ 3
774
+ ],
775
+ "continuous": true
776
+ },
777
+ "orange_juice_linvel": {
778
+ "absolute": false,
779
+ "rotation_type": null,
780
+ "shape": [
781
+ 3
782
+ ],
783
+ "continuous": true
784
+ },
785
+ "wine_bottle_pos": {
786
+ "absolute": true,
787
+ "rotation_type": null,
788
+ "shape": [
789
+ 3
790
+ ],
791
+ "continuous": true
792
+ },
793
+ "wine_bottle_linvel": {
794
+ "absolute": false,
795
+ "rotation_type": null,
796
+ "shape": [
797
+ 3
798
+ ],
799
+ "continuous": true
800
+ },
801
+ "timestamp": {
802
+ "absolute": true,
803
+ "rotation_type": null,
804
+ "shape": [
805
+ 1
806
+ ],
807
+ "continuous": true
808
+ }
809
+ },
810
+ "action": {
811
+ "x": {
812
+ "absolute": false,
813
+ "rotation_type": null,
814
+ "shape": [
815
+ 1
816
+ ],
817
+ "continuous": true
818
+ },
819
+ "y": {
820
+ "absolute": false,
821
+ "rotation_type": null,
822
+ "shape": [
823
+ 1
824
+ ],
825
+ "continuous": true
826
+ },
827
+ "z": {
828
+ "absolute": false,
829
+ "rotation_type": null,
830
+ "shape": [
831
+ 1
832
+ ],
833
+ "continuous": true
834
+ },
835
+ "roll": {
836
+ "absolute": false,
837
+ "rotation_type": null,
838
+ "shape": [
839
+ 1
840
+ ],
841
+ "continuous": true
842
+ },
843
+ "pitch": {
844
+ "absolute": false,
845
+ "rotation_type": null,
846
+ "shape": [
847
+ 1
848
+ ],
849
+ "continuous": true
850
+ },
851
+ "yaw": {
852
+ "absolute": false,
853
+ "rotation_type": null,
854
+ "shape": [
855
+ 1
856
+ ],
857
+ "continuous": true
858
+ },
859
+ "gripper": {
860
+ "absolute": false,
861
+ "rotation_type": null,
862
+ "shape": [
863
+ 1
864
+ ],
865
+ "continuous": true
866
+ }
867
+ }
868
+ },
869
+ "embodiment_tag": "new_embodiment"
870
+ }
871
+ }
model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba6e9fff1b570745626bdc1d71b00bbdc641eb61450a6cc7c1a74542a1a51c1b
3
+ size 4999367032
model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c5dcfb615816641dc6f2530def278851d94908a21cf1377c38416a3b4038040
3
+ size 2599313540
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cf22acaf0dcb26bc7c98003ced6850d5100662c14748a02048816e39126173c
3
+ size 8575938362
rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fd8a4539f49b2fcbf2a18aabe1270ddf2488a0f04aa3144a563092a87b88685
3
+ size 15984
rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d38a6d0a3ff5b5b5bcae6db20b86c37178c472e95fb1e978acf5f908b68c144
3
+ size 15984
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adf5bf452761c234e3daa85a999070bcf12b7ea957b65555b14a80fc069fa45d
3
+ size 1064
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff