pzal commited on
Commit
1b1906c
·
verified ·
1 Parent(s): 5d6792a

Add files using upload-large-folder tool

Browse files
Files changed (31) hide show
  1. .gitattributes +18 -0
  2. vl__30_0__jerk_limited__ee__fixed/5000/_CHECKPOINT_METADATA +1 -0
  3. vl__30_0__jerk_limited__ee__fixed/5000/assets/ur10e/norm_stats.json +491 -0
  4. vl__30_0__jerk_limited__ee__fixed/5000/params/_METADATA +1 -0
  5. vl__30_0__jerk_limited__ee__fixed/5000/params/_sharding +1 -0
  6. vl__30_0__jerk_limited__ee__fixed/5000/params/d/738dce1060f2bc55eb0a51719117e3ef +0 -0
  7. vl__30_0__jerk_limited__ee__fixed/5000/params/manifest.ocdbt +0 -0
  8. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/04c89c7981ea360eafaa131b1b9222d6 +3 -0
  9. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/0f66dbca9f049d6bdb86ac23e14aea68 +3 -0
  10. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/26446c0de72a3845eba758f16641646f +3 -0
  11. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/2783f87d6436d7852bb79d2bcbf0f27b +3 -0
  12. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/3699c2406efc6e079b7509e9501ba056 +0 -0
  13. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/387fce33b254ea3d84f8b48d8e7f4076 +3 -0
  14. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/4c205c221643396363a8b5cb02735f9e +3 -0
  15. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/542858950707a854db59edc23f0ea173 +3 -0
  16. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/549b378ec695cedab7b45cf52a6e8368 +3 -0
  17. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/6f8ea509c991c127780f603d9a4bd074 +3 -0
  18. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/795bf0c88d72bbe505df8f18145d7f8a +3 -0
  19. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/7ee35eb20e1d0424c594c1dac75c550f +3 -0
  20. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/82133cc4974bb608681fc2f04729a44d +3 -0
  21. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/99f20e6033de2f272444a7a7bbf45eab +3 -0
  22. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/a197b1c7302ef78cdb81e701df35f1ef +3 -0
  23. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/ab620365459ad785774e6ca1c70186bf +0 -0
  24. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/b6fd90504a67bc0e14d2698c5ef74b2b +3 -0
  25. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/bc25c3b762ba42ca0ffa5a12defce179 +3 -0
  26. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/cbba0567628d092367ca59c9696424c9 +3 -0
  27. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/ee6016c7ae952cfa8558b1bcc8a9e8bc +0 -0
  28. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/f1c77bc54705f740062e47aba530bc3d +3 -0
  29. vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/manifest.ocdbt +0 -0
  30. vl__30_0__jerk_limited__ee__fixed/config.yaml +95 -0
  31. vl__30_0__jerk_limited__ee__fixed/wandb_id.txt +1 -0
.gitattributes CHANGED
@@ -271,3 +271,21 @@ vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/
271
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/0a5c4ed72598888634d910ed0dbe926e filter=lfs diff=lfs merge=lfs -text
272
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/e78e1e05989c27f35d2393f269068368 filter=lfs diff=lfs merge=lfs -text
273
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/09f9800bf5d27f7f59cc0e2a7983dfd8 filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
271
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/0a5c4ed72598888634d910ed0dbe926e filter=lfs diff=lfs merge=lfs -text
272
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/e78e1e05989c27f35d2393f269068368 filter=lfs diff=lfs merge=lfs -text
273
  vl__30_0__jerk_limited__j_loss_fix_and_pred_pose_based_corrective_actions/10000/params/ocdbt.process_0/d/09f9800bf5d27f7f59cc0e2a7983dfd8 filter=lfs diff=lfs merge=lfs -text
274
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/26446c0de72a3845eba758f16641646f filter=lfs diff=lfs merge=lfs -text
275
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/7ee35eb20e1d0424c594c1dac75c550f filter=lfs diff=lfs merge=lfs -text
276
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/2783f87d6436d7852bb79d2bcbf0f27b filter=lfs diff=lfs merge=lfs -text
277
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/99f20e6033de2f272444a7a7bbf45eab filter=lfs diff=lfs merge=lfs -text
278
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/387fce33b254ea3d84f8b48d8e7f4076 filter=lfs diff=lfs merge=lfs -text
279
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/4c205c221643396363a8b5cb02735f9e filter=lfs diff=lfs merge=lfs -text
280
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/cbba0567628d092367ca59c9696424c9 filter=lfs diff=lfs merge=lfs -text
281
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/0f66dbca9f049d6bdb86ac23e14aea68 filter=lfs diff=lfs merge=lfs -text
282
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/04c89c7981ea360eafaa131b1b9222d6 filter=lfs diff=lfs merge=lfs -text
283
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/542858950707a854db59edc23f0ea173 filter=lfs diff=lfs merge=lfs -text
284
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/f1c77bc54705f740062e47aba530bc3d filter=lfs diff=lfs merge=lfs -text
285
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/bc25c3b762ba42ca0ffa5a12defce179 filter=lfs diff=lfs merge=lfs -text
286
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/549b378ec695cedab7b45cf52a6e8368 filter=lfs diff=lfs merge=lfs -text
287
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/795bf0c88d72bbe505df8f18145d7f8a filter=lfs diff=lfs merge=lfs -text
288
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/a197b1c7302ef78cdb81e701df35f1ef filter=lfs diff=lfs merge=lfs -text
289
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/82133cc4974bb608681fc2f04729a44d filter=lfs diff=lfs merge=lfs -text
290
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/6f8ea509c991c127780f603d9a4bd074 filter=lfs diff=lfs merge=lfs -text
291
+ vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/b6fd90504a67bc0e14d2698c5ef74b2b filter=lfs diff=lfs merge=lfs -text
vl__30_0__jerk_limited__ee__fixed/5000/_CHECKPOINT_METADATA ADDED
@@ -0,0 +1 @@
 
 
1
+ {"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1764870994803489442, "commit_timestamp_nsecs": 1764871319624898077, "custom": {}}
vl__30_0__jerk_limited__ee__fixed/5000/assets/ur10e/norm_stats.json ADDED
@@ -0,0 +1,491 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "norm_stats": {
3
+ "state": {
4
+ "mean": [
5
+ -0.0003960459434892982,
6
+ -0.000397581810830161,
7
+ -0.00007269233901752159,
8
+ -0.00011631726374616846,
9
+ 0.00005929979306529276,
10
+ 0.0004051154828630388,
11
+ -0.00019493294530548155,
12
+ -0.00019923386571463197,
13
+ -0.000042968251364072785,
14
+ -0.00005530511771212332,
15
+ 0.000030398547096410766,
16
+ 0.00020429094729479402,
17
+ 0.5986685752868652,
18
+ 0.5981546640396118,
19
+ 0.05039028078317642,
20
+ 0.0004953373572789133,
21
+ 0.10711561888456345,
22
+ -0.0005407895660027862,
23
+ -0.006484619807451963,
24
+ 0.003617143025621772,
25
+ 0.5981546640396118,
26
+ 0.0,
27
+ 0.0,
28
+ 0.0,
29
+ 0.0,
30
+ 0.0,
31
+ 0.0,
32
+ 0.0,
33
+ 0.0,
34
+ 0.0,
35
+ 0.0,
36
+ 0.0
37
+ ],
38
+ "std": [
39
+ 0.004681008402258158,
40
+ 0.006901760585606098,
41
+ 0.005719676148146391,
42
+ 0.0025849423836916685,
43
+ 0.003988918848335743,
44
+ 0.02001541666686535,
45
+ 0.002355140633881092,
46
+ 0.003466826630756259,
47
+ 0.002882360015064478,
48
+ 0.0012979020830243826,
49
+ 0.002016529208049178,
50
+ 0.010074124671518803,
51
+ 0.4901678264141083,
52
+ 0.4902710020542145,
53
+ 0.11618480086326599,
54
+ 0.1681893765926361,
55
+ 0.07788223028182983,
56
+ 0.04501405730843544,
57
+ 0.0661778524518013,
58
+ 0.5958529710769653,
59
+ 0.4902710020542145,
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
+ ],
72
+ "q01": [
73
+ -0.014896168034523725,
74
+ -0.023782781732827423,
75
+ -0.018495610508322716,
76
+ -0.00842591795325279,
77
+ -0.011514233987033367,
78
+ -0.06873272716999054,
79
+ -0.007511040891706944,
80
+ -0.011889379095286132,
81
+ -0.009328192451037467,
82
+ -0.004211092096008361,
83
+ -0.005837117517367006,
84
+ -0.03425580984652043,
85
+ 0.0,
86
+ 0.0,
87
+ -0.19741441082954406,
88
+ -0.3133266278743744,
89
+ 0.00863822932690382,
90
+ -0.11799316820502281,
91
+ -0.1673899977207184,
92
+ -1.4435412939071655,
93
+ 0.0,
94
+ 0.0,
95
+ 0.0,
96
+ 0.0,
97
+ 0.0,
98
+ 0.0,
99
+ 0.0,
100
+ 0.0,
101
+ 0.0,
102
+ 0.0,
103
+ 0.0,
104
+ 0.0
105
+ ],
106
+ "q99": [
107
+ 0.013969538396224383,
108
+ 0.023224324332177637,
109
+ 0.010958568666875362,
110
+ 0.007709948647767306,
111
+ 0.014396399548649783,
112
+ 0.06840321898460389,
113
+ 0.007093064415082333,
114
+ 0.011769884042814378,
115
+ 0.005536517574451862,
116
+ 0.0038669928954914203,
117
+ 0.007195224316045641,
118
+ 0.03456086775064468,
119
+ 0.9998,
120
+ 0.9998,
121
+ 0.29779768657684336,
122
+ 0.406406463944912,
123
+ 0.3194635294646025,
124
+ 0.1383301957249642,
125
+ 0.1970333372354507,
126
+ 1.6112462955951692,
127
+ 0.9998,
128
+ 0.0,
129
+ 0.0,
130
+ 0.0,
131
+ 0.0,
132
+ 0.0,
133
+ 0.0,
134
+ 0.0,
135
+ 0.0,
136
+ 0.0,
137
+ 0.0,
138
+ 0.0
139
+ ]
140
+ },
141
+ "actions": {
142
+ "mean": [
143
+ 0.0013973810710012913,
144
+ 0.004460461903363466,
145
+ 0.005434031598269939,
146
+ -0.0003450494259595871,
147
+ -0.0009035700932145119,
148
+ -0.005387853365391493,
149
+ 0.5995351076126099,
150
+ 0.0,
151
+ 0.0,
152
+ 0.0,
153
+ 0.0,
154
+ 0.0,
155
+ 0.0,
156
+ 0.0,
157
+ 0.0,
158
+ 0.0,
159
+ 0.0,
160
+ 0.0,
161
+ 0.0,
162
+ 0.0,
163
+ 0.0,
164
+ 0.0,
165
+ 0.0,
166
+ 0.0,
167
+ 0.0,
168
+ 0.0,
169
+ 0.0,
170
+ 0.0,
171
+ 0.0,
172
+ 0.0,
173
+ 0.0,
174
+ 0.0
175
+ ],
176
+ "std": [
177
+ 0.06290271878242493,
178
+ 0.08913861215114594,
179
+ 0.07710187137126923,
180
+ 0.027364596724510193,
181
+ 0.04435565695166588,
182
+ 0.2560824155807495,
183
+ 0.489992618560791,
184
+ 0.0,
185
+ 0.0,
186
+ 0.0,
187
+ 0.0,
188
+ 0.0,
189
+ 0.0,
190
+ 0.0,
191
+ 0.0,
192
+ 0.0,
193
+ 0.0,
194
+ 0.0,
195
+ 0.0,
196
+ 0.0,
197
+ 0.0,
198
+ 0.0,
199
+ 0.0,
200
+ 0.0,
201
+ 0.0,
202
+ 0.0,
203
+ 0.0,
204
+ 0.0,
205
+ 0.0,
206
+ 0.0,
207
+ 0.0,
208
+ 0.0
209
+ ],
210
+ "q01": [
211
+ -0.20575400571823121,
212
+ -0.29649169664382935,
213
+ -0.16343017610907554,
214
+ -0.09328187179565428,
215
+ -0.1677116035461426,
216
+ -0.8966748435020446,
217
+ 0.0,
218
+ 0.0,
219
+ 0.0,
220
+ 0.0,
221
+ 0.0,
222
+ 0.0,
223
+ 0.0,
224
+ 0.0,
225
+ 0.0,
226
+ 0.0,
227
+ 0.0,
228
+ 0.0,
229
+ 0.0,
230
+ 0.0,
231
+ 0.0,
232
+ 0.0,
233
+ 0.0,
234
+ 0.0,
235
+ 0.0,
236
+ 0.0,
237
+ 0.0,
238
+ 0.0,
239
+ 0.0,
240
+ 0.0,
241
+ 0.0,
242
+ 0.0
243
+ ],
244
+ "q99": [
245
+ 0.20699647083282469,
246
+ 0.3072968993186951,
247
+ 0.2638481133878231,
248
+ 0.07787252345681195,
249
+ 0.13428698064684863,
250
+ 0.8701046673774719,
251
+ 0.9998,
252
+ 0.0,
253
+ 0.0,
254
+ 0.0,
255
+ 0.0,
256
+ 0.0,
257
+ 0.0,
258
+ 0.0,
259
+ 0.0,
260
+ 0.0,
261
+ 0.0,
262
+ 0.0,
263
+ 0.0,
264
+ 0.0,
265
+ 0.0,
266
+ 0.0,
267
+ 0.0,
268
+ 0.0,
269
+ 0.0,
270
+ 0.0,
271
+ 0.0,
272
+ 0.0,
273
+ 0.0,
274
+ 0.0,
275
+ 0.0,
276
+ 0.0
277
+ ]
278
+ },
279
+ "previous_actions": {
280
+ "mean": [
281
+ 0.0013973810710012913,
282
+ 0.004460461903363466,
283
+ 0.005434031598269939,
284
+ -0.0003450494259595871,
285
+ -0.0009035700932145119,
286
+ -0.005387853365391493,
287
+ 0.5995351076126099,
288
+ 0.0,
289
+ 0.0,
290
+ 0.0,
291
+ 0.0,
292
+ 0.0,
293
+ 0.0,
294
+ 0.0,
295
+ 0.0,
296
+ 0.0,
297
+ 0.0,
298
+ 0.0,
299
+ 0.0,
300
+ 0.0,
301
+ 0.0,
302
+ 0.0,
303
+ 0.0,
304
+ 0.0,
305
+ 0.0,
306
+ 0.0,
307
+ 0.0,
308
+ 0.0,
309
+ 0.0,
310
+ 0.0,
311
+ 0.0,
312
+ 0.0
313
+ ],
314
+ "std": [
315
+ 0.06290271878242493,
316
+ 0.08913861215114594,
317
+ 0.07710187137126923,
318
+ 0.027364596724510193,
319
+ 0.04435565695166588,
320
+ 0.2560824155807495,
321
+ 0.489992618560791,
322
+ 0.0,
323
+ 0.0,
324
+ 0.0,
325
+ 0.0,
326
+ 0.0,
327
+ 0.0,
328
+ 0.0,
329
+ 0.0,
330
+ 0.0,
331
+ 0.0,
332
+ 0.0,
333
+ 0.0,
334
+ 0.0,
335
+ 0.0,
336
+ 0.0,
337
+ 0.0,
338
+ 0.0,
339
+ 0.0,
340
+ 0.0,
341
+ 0.0,
342
+ 0.0,
343
+ 0.0,
344
+ 0.0,
345
+ 0.0,
346
+ 0.0
347
+ ],
348
+ "q01": [
349
+ -0.20575400571823121,
350
+ -0.29649169664382935,
351
+ -0.16343017610907554,
352
+ -0.09328187179565428,
353
+ -0.1677116035461426,
354
+ -0.8966748435020446,
355
+ 0.0,
356
+ 0.0,
357
+ 0.0,
358
+ 0.0,
359
+ 0.0,
360
+ 0.0,
361
+ 0.0,
362
+ 0.0,
363
+ 0.0,
364
+ 0.0,
365
+ 0.0,
366
+ 0.0,
367
+ 0.0,
368
+ 0.0,
369
+ 0.0,
370
+ 0.0,
371
+ 0.0,
372
+ 0.0,
373
+ 0.0,
374
+ 0.0,
375
+ 0.0,
376
+ 0.0,
377
+ 0.0,
378
+ 0.0,
379
+ 0.0,
380
+ 0.0
381
+ ],
382
+ "q99": [
383
+ 0.20699647083282469,
384
+ 0.3072968993186951,
385
+ 0.2638481133878231,
386
+ 0.07787252345681195,
387
+ 0.13428698064684863,
388
+ 0.8701046673774719,
389
+ 0.9998,
390
+ 0.0,
391
+ 0.0,
392
+ 0.0,
393
+ 0.0,
394
+ 0.0,
395
+ 0.0,
396
+ 0.0,
397
+ 0.0,
398
+ 0.0,
399
+ 0.0,
400
+ 0.0,
401
+ 0.0,
402
+ 0.0,
403
+ 0.0,
404
+ 0.0,
405
+ 0.0,
406
+ 0.0,
407
+ 0.0,
408
+ 0.0,
409
+ 0.0,
410
+ 0.0,
411
+ 0.0,
412
+ 0.0,
413
+ 0.0,
414
+ 0.0
415
+ ]
416
+ }
417
+ },
418
+ "sample_ratio": 0.05,
419
+ "dataset_infos": {
420
+ "robotgeneralist/pp__1__train": {
421
+ "commit_hash": "b8158e3f7eb1eb4fcad944d3dfbcdba2796a214b",
422
+ "num_episodes": 166,
423
+ "num_frames": 69615
424
+ },
425
+ "robotgeneralist/pp__1__adversarial": {
426
+ "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
427
+ "num_episodes": 64,
428
+ "num_frames": 17906
429
+ },
430
+ "robotgeneralist/pp__2": {
431
+ "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
432
+ "num_episodes": 92,
433
+ "num_frames": 37458
434
+ },
435
+ "robotgeneralist/pp__2__rotations": {
436
+ "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
437
+ "num_episodes": 102,
438
+ "num_frames": 13935
439
+ },
440
+ "robotgeneralist/pp__3__with_ft": {
441
+ "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
442
+ "num_episodes": 99,
443
+ "num_frames": 38424
444
+ },
445
+ "robotgeneralist/pp__3__with_ft__rotations": {
446
+ "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
447
+ "num_episodes": 49,
448
+ "num_frames": 7151
449
+ },
450
+ "robotgeneralist/pp__3__with_ft__adversarial_start": {
451
+ "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
452
+ "num_episodes": 147,
453
+ "num_frames": 21313
454
+ },
455
+ "robotgeneralist/pp__4__with_ft": {
456
+ "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
457
+ "num_episodes": 94,
458
+ "num_frames": 33046
459
+ },
460
+ "robotgeneralist/pp__5__with_ft__adversarial_target": {
461
+ "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
462
+ "num_episodes": 48,
463
+ "num_frames": 10615
464
+ },
465
+ "robotgeneralist/pp__6__with_ft": {
466
+ "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
467
+ "num_episodes": 379,
468
+ "num_frames": 123611
469
+ },
470
+ "robotgeneralist/precise-packing-2025-10-22-part-1": {
471
+ "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
472
+ "num_episodes": 98,
473
+ "num_frames": 41410
474
+ },
475
+ "robotgeneralist/precise-packing-2025-10-22-part-2": {
476
+ "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
477
+ "num_episodes": 94,
478
+ "num_frames": 37859
479
+ },
480
+ "robotgeneralist/precise-packing-2025-10-22-part-3": {
481
+ "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
482
+ "num_episodes": 127,
483
+ "num_frames": 53341
484
+ },
485
+ "robotgeneralist/precise-packing-2025-10-22-part-4": {
486
+ "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
487
+ "num_episodes": 102,
488
+ "num_frames": 38263
489
+ }
490
+ }
491
+ }
vl__30_0__jerk_limited__ee__fixed/5000/params/_METADATA ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'action_time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'state_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('params', 'state_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
vl__30_0__jerk_limited__ee__fixed/5000/params/_sharding ADDED
@@ -0,0 +1 @@
 
 
1
+ {"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}"}
vl__30_0__jerk_limited__ee__fixed/5000/params/d/738dce1060f2bc55eb0a51719117e3ef ADDED
Binary file (2.26 kB). View file
 
vl__30_0__jerk_limited__ee__fixed/5000/params/manifest.ocdbt ADDED
Binary file (117 Bytes). View file
 
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/04c89c7981ea360eafaa131b1b9222d6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8601ce1be6734aa1d289bb62f466b26a03278e77fcd0507371b66ff525625e8b
3
+ size 139915264
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/0f66dbca9f049d6bdb86ac23e14aea68 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:663ec666638972c1248c81c705b5267ae72706ad32e43ecf9c216743500d91d1
3
+ size 34975744
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/26446c0de72a3845eba758f16641646f ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ff295cd872ecdf9b2e2a3085ad94bcdb45e773e304a3d04f4528b1b28fc5a12
3
+ size 3077554176
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/2783f87d6436d7852bb79d2bcbf0f27b ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92cf007ebf74139fdbfca4453e6e3060da463675b5b238ac3d6f5295e30959e3
3
+ size 133597
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/3699c2406efc6e079b7509e9501ba056 ADDED
Binary file (872 Bytes). View file
 
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/387fce33b254ea3d84f8b48d8e7f4076 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc4b7c579ed21be5d7bd63ccfa45a3cbfd0e78fe76d671d1b778091127384248
3
+ size 3889263
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/4c205c221643396363a8b5cb02735f9e ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a56d1ff1a0d2ec0241080533087d53ae558747b10f170656c31e084366b08c67
3
+ size 7770112
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/542858950707a854db59edc23f0ea173 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16165287133a36b9e48c79a1efb1b5bbaca998262f6c15cc55f224925de56818
3
+ size 132845568
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/549b378ec695cedab7b45cf52a6e8368 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d470ae579b0bdab9726a509212ecbf8a9333482d5fe30bafcc1ca31a7f8f1ea5
3
+ size 560226304
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/6f8ea509c991c127780f603d9a4bd074 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:39d7f5a4f0a6d66d81ed5c8b43d812ad89e9826dffc4f955269a77dfb72a76ac
3
+ size 2240233472
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/795bf0c88d72bbe505df8f18145d7f8a ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cab854dce6f3064bf2a7bc97799eb96ce9effa026b8708c8a775f53e861fe399
3
+ size 559742976
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/7ee35eb20e1d0424c594c1dac75c550f ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b5f1dbc2527ce7eeb13d6cce13f2f9198968be746b49d472114e13d19c5c01d
3
+ size 13864960
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/82133cc4974bb608681fc2f04729a44d ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4328cac2cfba9d7c2340a14dbf54f1d3aaaae33b4366c9b9856e3cf53a13b8f
3
+ size 1120104448
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/99f20e6033de2f272444a7a7bbf45eab ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad83c9ff0e089b8e40633c12acba63343926015a36601012554c09177d07feb7
3
+ size 653207
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/a197b1c7302ef78cdb81e701df35f1ef ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ec9c5f25ea4e214e4ad2480257f2c8386e0444eff02398847f0fcdad80c73e2
3
+ size 993775616
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/ab620365459ad785774e6ca1c70186bf ADDED
Binary file (213 Bytes). View file
 
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/b6fd90504a67bc0e14d2698c5ef74b2b ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:784966528e9e6669387d8b8027d0c9abdbf0bab357eea21d35fe10dc1471be62
3
+ size 2240221184
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/bc25c3b762ba42ca0ffa5a12defce179 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80e7a769ed09cf961498182a2b14d4f4d65850d1b5eaf595cc4190a54291f521
3
+ size 279232512
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/cbba0567628d092367ca59c9696424c9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da1e1ea036331b3b4857cb047ea0fd32fe14e48680eaf50bde086c126892d5b4
3
+ size 70090752
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/ee6016c7ae952cfa8558b1bcc8a9e8bc ADDED
Binary file (2.24 kB). View file
 
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/d/f1c77bc54705f740062e47aba530bc3d ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:998b8c8d6b8026dabd6ba4f942d2ab66fda475b97175503907b22d0ccdc2b6cc
3
+ size 538849280
vl__30_0__jerk_limited__ee__fixed/5000/params/ocdbt.process_0/manifest.ocdbt ADDED
Binary file (253 Bytes). View file
 
vl__30_0__jerk_limited__ee__fixed/config.yaml ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object:openpi.training.config.TrainConfig
2
+ name: vl__30_0__jerk_limited__ee
3
+ project_name: openpi
4
+ exp_name: vl__30_0__jerk_limited__ee__fixed
5
+ model: !!python/object:openpi.models.pi0.Pi0Config
6
+ action_dim: 32
7
+ action_horizon: 50
8
+ max_token_len: 48
9
+ dtype: bfloat16
10
+ paligemma_variant: gemma_2b
11
+ action_expert_variant: gemma_300m
12
+ img_variant: So400m/14
13
+ tv_loss_weight: 30.0
14
+ tv_loss_type: jerk-limited
15
+ tv_loss_include_prev_actions: true
16
+ weight_loader: !!python/object:openpi.training.weight_loaders.CheckpointWeightLoader
17
+ local_dir: null
18
+ repo_id: robotgeneralist/openpi_checkpoint_mirrors
19
+ revision: main
20
+ checkpoint_path: pi0_base
21
+ lr_schedule: !!python/object:openpi.training.optimizer.CosineDecaySchedule
22
+ warmup_steps: 1000
23
+ peak_lr: 2.5e-05
24
+ decay_steps: 30000
25
+ decay_lr: 2.5e-06
26
+ optimizer: !!python/object:openpi.training.optimizer.AdamW
27
+ b1: 0.9
28
+ b2: 0.95
29
+ eps: 1.0e-08
30
+ weight_decay: 1.0e-10
31
+ clip_gradient_norm: 1.0
32
+ ema_decay: 0.99
33
+ freeze_filter: !!python/name:flax.nnx.filterlib.Nothing ''
34
+ data: !!python/object:openpi.training.config.LeRobotNomagicURXDataConfig
35
+ repo_ids: !!python/tuple
36
+ - robotgeneralist/pp__1__train
37
+ - robotgeneralist/pp__1__adversarial
38
+ - robotgeneralist/pp__2
39
+ - robotgeneralist/pp__2__rotations
40
+ - robotgeneralist/pp__3__with_ft
41
+ - robotgeneralist/pp__3__with_ft__rotations
42
+ - robotgeneralist/pp__3__with_ft__adversarial_start
43
+ - robotgeneralist/pp__4__with_ft
44
+ - robotgeneralist/pp__5__with_ft__adversarial_target
45
+ - robotgeneralist/pp__6__with_ft
46
+ - robotgeneralist/precise-packing-2025-10-22-part-1
47
+ - robotgeneralist/precise-packing-2025-10-22-part-2
48
+ - robotgeneralist/precise-packing-2025-10-22-part-3
49
+ - robotgeneralist/precise-packing-2025-10-22-part-4
50
+ assets: !!python/object:openpi.training.config.AssetsConfig
51
+ assets_dir: null
52
+ asset_id: ur10e
53
+ norm_stats_sample_ratio: 0.05
54
+ base_config: !!python/object:openpi.training.config.DataConfig
55
+ repo_ids: null
56
+ asset_id: null
57
+ norm_stats_data: null
58
+ norm_stats_sample_ratio: 1.0
59
+ norm_stats_dir: null
60
+ repack_transforms: !!python/object:openpi.transforms.Group
61
+ inputs: !!python/tuple []
62
+ outputs: !!python/tuple []
63
+ data_transforms: !!python/object:openpi.transforms.Group
64
+ inputs: !!python/tuple []
65
+ outputs: !!python/tuple []
66
+ model_transforms: !!python/object:openpi.transforms.Group
67
+ inputs: !!python/tuple []
68
+ outputs: !!python/tuple []
69
+ use_quantile_norm: false
70
+ action_sequence_keys: !!python/tuple
71
+ - actions
72
+ fourier_config: null
73
+ prompt_from_task: true
74
+ default_prompt: null
75
+ use_poses: true
76
+ end_effector_frame: true
77
+ use_dynamic_state: true
78
+ fourier_config: null
79
+ assets_base_dir: ./assets
80
+ checkpoint_base_dir: /storage_nvme_1/pzal/checkpoints
81
+ seed: 42
82
+ batch_size: 32
83
+ num_workers: 2
84
+ num_train_steps: 10001
85
+ log_interval: 100
86
+ save_interval: 1000
87
+ keep_period: 5000
88
+ overwrite: false
89
+ resume: true
90
+ wandb_enabled: true
91
+ policy_metadata: null
92
+ fsdp_devices: 1
93
+ test_split_ratio: 0.05
94
+ test_eval_interval: 2000
95
+ test_eval_use_train_mode: true
vl__30_0__jerk_limited__ee__fixed/wandb_id.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ vvtcuzew