pzal commited on
Commit
9777d7c
·
verified ·
1 Parent(s): c4d9778

Add files using upload-large-folder tool

Browse files
Files changed (35) hide show
  1. .gitattributes +23 -0
  2. v5__tvl__0_2__no_dyn/5000/_CHECKPOINT_METADATA +1 -0
  3. v5__tvl__0_2__no_dyn/5000/assets/ur10e/norm_stats.json +353 -0
  4. v5__tvl__0_2__no_dyn/5000/params/_METADATA +1 -0
  5. v5__tvl__0_2__no_dyn/5000/params/_sharding +1 -0
  6. v5__tvl__0_2__no_dyn/5000/params/d/3c182aa46976fd2e721dceb8360c4b22 +0 -0
  7. v5__tvl__0_2__no_dyn/5000/params/manifest.ocdbt +0 -0
  8. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/1c46ea152f6ee9bc4d9126460d97814c +3 -0
  9. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/2d865394065c3a189a9cc815ac543675 +3 -0
  10. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/393951591af519b35f197089e6cce648 +3 -0
  11. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/3951b7e6ea6af4f16a89ec7700c6bf7a +3 -0
  12. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/476b28bf312620085f6a2a0a215398b2 +3 -0
  13. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/4f0c00bd6c62b09500044074cd95c733 +0 -0
  14. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/59814fab450c8db24dc8f54c7e5a5bc1 +3 -0
  15. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/5c355134126cba0188eb3359c050f340 +3 -0
  16. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/67ae3f487a818a239e757f6d22c4a2aa +3 -0
  17. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/70c219e996e6c66123717aea55e14525 +3 -0
  18. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/84c30b89170ef6a2e8af11adcabd21be +0 -0
  19. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/87e0a64717ea759c1be20bf4d9fa225e +3 -0
  20. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/918e7c377fd7a6527543997a1116a21a +3 -0
  21. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/94138de6f1ae96f4e3a3ba5a7e837665 +3 -0
  22. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/97c06f306c38269754a930b14e5a1a4b +3 -0
  23. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/aa82994c4bb23d2fcaf1dff385906a40 +3 -0
  24. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c376fb098840eafe6c0357dddd39f675 +3 -0
  25. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c4ed3ef81206ba9443b2433f7f4c7fd0 +3 -0
  26. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c8e7e58e56d51d4ee40301406c504601 +3 -0
  27. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c939b3c6a981afffa208fa8fb58b0c9d +3 -0
  28. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d0e6fb07ccb3e52867497951f89d3306 +3 -0
  29. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d92f425ddca63265c0f39f142d8b25de +3 -0
  30. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/dfedaf9c8fe7c197d541df8a0b04d209 +3 -0
  31. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/f93f9bdfcfc987a73372568d013c967c +3 -0
  32. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/ffad2779778a2a8a8d3316b3aad48f36 +3 -0
  33. v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/manifest.ocdbt +0 -0
  34. v5__tvl__0_2__no_dyn/config.yaml +95 -0
  35. v5__tvl__0_2__no_dyn/wandb_id.txt +1 -0
.gitattributes CHANGED
@@ -1887,3 +1887,26 @@ v6_standard/30000/params/ocdbt.process_0/d/1a6244b6b30e8bf1dd42bcd5344a5a3c filt
1887
  v6_standard/30000/params/ocdbt.process_0/d/c7b23bf30c07503c04eafc3d4de9fae7 filter=lfs diff=lfs merge=lfs -text
1888
  v6_standard/30000/params/ocdbt.process_0/d/287e8cbe6fb3c8b376f1fb9c4f058022 filter=lfs diff=lfs merge=lfs -text
1889
  v6_standard/30000/params/ocdbt.process_0/d/70a83f0292b7984e9f2cede82eaaad31 filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1887
  v6_standard/30000/params/ocdbt.process_0/d/c7b23bf30c07503c04eafc3d4de9fae7 filter=lfs diff=lfs merge=lfs -text
1888
  v6_standard/30000/params/ocdbt.process_0/d/287e8cbe6fb3c8b376f1fb9c4f058022 filter=lfs diff=lfs merge=lfs -text
1889
  v6_standard/30000/params/ocdbt.process_0/d/70a83f0292b7984e9f2cede82eaaad31 filter=lfs diff=lfs merge=lfs -text
1890
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c4ed3ef81206ba9443b2433f7f4c7fd0 filter=lfs diff=lfs merge=lfs -text
1891
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c8e7e58e56d51d4ee40301406c504601 filter=lfs diff=lfs merge=lfs -text
1892
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c376fb098840eafe6c0357dddd39f675 filter=lfs diff=lfs merge=lfs -text
1893
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/f93f9bdfcfc987a73372568d013c967c filter=lfs diff=lfs merge=lfs -text
1894
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/3951b7e6ea6af4f16a89ec7700c6bf7a filter=lfs diff=lfs merge=lfs -text
1895
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/476b28bf312620085f6a2a0a215398b2 filter=lfs diff=lfs merge=lfs -text
1896
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/97c06f306c38269754a930b14e5a1a4b filter=lfs diff=lfs merge=lfs -text
1897
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d92f425ddca63265c0f39f142d8b25de filter=lfs diff=lfs merge=lfs -text
1898
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/393951591af519b35f197089e6cce648 filter=lfs diff=lfs merge=lfs -text
1899
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/94138de6f1ae96f4e3a3ba5a7e837665 filter=lfs diff=lfs merge=lfs -text
1900
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d0e6fb07ccb3e52867497951f89d3306 filter=lfs diff=lfs merge=lfs -text
1901
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/1c46ea152f6ee9bc4d9126460d97814c filter=lfs diff=lfs merge=lfs -text
1902
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/87e0a64717ea759c1be20bf4d9fa225e filter=lfs diff=lfs merge=lfs -text
1903
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/59814fab450c8db24dc8f54c7e5a5bc1 filter=lfs diff=lfs merge=lfs -text
1904
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/ffad2779778a2a8a8d3316b3aad48f36 filter=lfs diff=lfs merge=lfs -text
1905
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/918e7c377fd7a6527543997a1116a21a filter=lfs diff=lfs merge=lfs -text
1906
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/dfedaf9c8fe7c197d541df8a0b04d209 filter=lfs diff=lfs merge=lfs -text
1907
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/67ae3f487a818a239e757f6d22c4a2aa filter=lfs diff=lfs merge=lfs -text
1908
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/2d865394065c3a189a9cc815ac543675 filter=lfs diff=lfs merge=lfs -text
1909
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c939b3c6a981afffa208fa8fb58b0c9d filter=lfs diff=lfs merge=lfs -text
1910
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/5c355134126cba0188eb3359c050f340 filter=lfs diff=lfs merge=lfs -text
1911
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/70c219e996e6c66123717aea55e14525 filter=lfs diff=lfs merge=lfs -text
1912
+ v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/aa82994c4bb23d2fcaf1dff385906a40 filter=lfs diff=lfs merge=lfs -text
v5__tvl__0_2__no_dyn/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": 1764592109165080399, "commit_timestamp_nsecs": 1764592155494044781, "custom": {}}
v5__tvl__0_2__no_dyn/5000/assets/ur10e/norm_stats.json ADDED
@@ -0,0 +1,353 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "norm_stats": {
3
+ "state": {
4
+ "mean": [
5
+ 0.05039028078317642,
6
+ 0.0004953373572789133,
7
+ 0.10711561888456345,
8
+ -0.0005407895660027862,
9
+ -0.006484619807451963,
10
+ 0.003617143025621772,
11
+ 0.5981546640396118,
12
+ 0.0,
13
+ 0.0,
14
+ 0.0,
15
+ 0.0,
16
+ 0.0,
17
+ 0.0,
18
+ 0.0,
19
+ 0.0,
20
+ 0.0,
21
+ 0.0,
22
+ 0.0,
23
+ 0.0,
24
+ 0.0,
25
+ 0.0,
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.11618480086326599,
40
+ 0.1681893765926361,
41
+ 0.07788223028182983,
42
+ 0.04501405730843544,
43
+ 0.0661778524518013,
44
+ 0.5958529710769653,
45
+ 0.4902710020542145,
46
+ 0.0,
47
+ 0.0,
48
+ 0.0,
49
+ 0.0,
50
+ 0.0,
51
+ 0.0,
52
+ 0.0,
53
+ 0.0,
54
+ 0.0,
55
+ 0.0,
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
+ ],
72
+ "q01": [
73
+ -0.19741441309452057,
74
+ -0.3133265972137451,
75
+ 0.008638229221105576,
76
+ -0.11799314618110657,
77
+ -0.16739001870155334,
78
+ -1.4435412883758545,
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0,
84
+ 0.0,
85
+ 0.0,
86
+ 0.0,
87
+ 0.0,
88
+ 0.0,
89
+ 0.0,
90
+ 0.0,
91
+ 0.0,
92
+ 0.0,
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.29779767990112305,
108
+ 0.4064065217971802,
109
+ 0.3194635212421417,
110
+ 0.13833022117614746,
111
+ 0.19703328609466553,
112
+ 1.6112463474273682,
113
+ 0.9997999668121338,
114
+ 0.0,
115
+ 0.0,
116
+ 0.0,
117
+ 0.0,
118
+ 0.0,
119
+ 0.0,
120
+ 0.0,
121
+ 0.0,
122
+ 0.0,
123
+ 0.0,
124
+ 0.0,
125
+ 0.0,
126
+ 0.0,
127
+ 0.0,
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.00005176717240829021,
144
+ 0.00012534917914308608,
145
+ 0.0002480015973560512,
146
+ 0.00005367679477785714,
147
+ -3.479610484191653e-7,
148
+ -0.0002569151984062046,
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.002336727222427726,
178
+ 0.0034517920576035976,
179
+ 0.003022634657099843,
180
+ 0.0012890173820778728,
181
+ 0.0020518647506833076,
182
+ 0.010340424254536629,
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.006998317316174507,
212
+ -0.011659257113933563,
213
+ -0.0055110761895775795,
214
+ -0.003918211907148361,
215
+ -0.007409645244479179,
216
+ -0.034376636147499084,
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.007169187068939209,
246
+ 0.011561842635273933,
247
+ 0.009297588840126991,
248
+ 0.004103695973753929,
249
+ 0.0058465879410505295,
250
+ 0.034828104078769684,
251
+ 0.9997999668121338,
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
+ },
280
+ "sample_ratio": 0.05,
281
+ "dataset_infos": {
282
+ "robotgeneralist/pp__1__train": {
283
+ "commit_hash": "b8158e3f7eb1eb4fcad944d3dfbcdba2796a214b",
284
+ "num_episodes": 166,
285
+ "num_frames": 69615
286
+ },
287
+ "robotgeneralist/pp__1__adversarial": {
288
+ "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
289
+ "num_episodes": 64,
290
+ "num_frames": 17906
291
+ },
292
+ "robotgeneralist/pp__2": {
293
+ "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
294
+ "num_episodes": 92,
295
+ "num_frames": 37458
296
+ },
297
+ "robotgeneralist/pp__2__rotations": {
298
+ "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
299
+ "num_episodes": 102,
300
+ "num_frames": 13935
301
+ },
302
+ "robotgeneralist/pp__3__with_ft": {
303
+ "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
304
+ "num_episodes": 99,
305
+ "num_frames": 38424
306
+ },
307
+ "robotgeneralist/pp__3__with_ft__rotations": {
308
+ "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
309
+ "num_episodes": 49,
310
+ "num_frames": 7151
311
+ },
312
+ "robotgeneralist/pp__3__with_ft__adversarial_start": {
313
+ "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
314
+ "num_episodes": 147,
315
+ "num_frames": 21313
316
+ },
317
+ "robotgeneralist/pp__4__with_ft": {
318
+ "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
319
+ "num_episodes": 94,
320
+ "num_frames": 33046
321
+ },
322
+ "robotgeneralist/pp__5__with_ft__adversarial_target": {
323
+ "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
324
+ "num_episodes": 48,
325
+ "num_frames": 10615
326
+ },
327
+ "robotgeneralist/pp__6__with_ft": {
328
+ "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
329
+ "num_episodes": 379,
330
+ "num_frames": 123611
331
+ },
332
+ "robotgeneralist/precise-packing-2025-10-22-part-1": {
333
+ "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
334
+ "num_episodes": 98,
335
+ "num_frames": 41410
336
+ },
337
+ "robotgeneralist/precise-packing-2025-10-22-part-2": {
338
+ "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
339
+ "num_episodes": 94,
340
+ "num_frames": 37859
341
+ },
342
+ "robotgeneralist/precise-packing-2025-10-22-part-3": {
343
+ "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
344
+ "num_episodes": 127,
345
+ "num_frames": 53341
346
+ },
347
+ "robotgeneralist/precise-packing-2025-10-22-part-4": {
348
+ "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
349
+ "num_episodes": 102,
350
+ "num_frames": 38263
351
+ }
352
+ }
353
+ }
v5__tvl__0_2__no_dyn/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}
v5__tvl__0_2__no_dyn/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}]]}}"}
v5__tvl__0_2__no_dyn/5000/params/d/3c182aa46976fd2e721dceb8360c4b22 ADDED
Binary file (2.37 kB). View file
 
v5__tvl__0_2__no_dyn/5000/params/manifest.ocdbt ADDED
Binary file (117 Bytes). View file
 
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/1c46ea152f6ee9bc4d9126460d97814c ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f45dee6da6ca86829dcda21f4154dbb81cf1f83a7675a5ed0ca21e902c3f6cb
3
+ size 132968448
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/2d865394065c3a189a9cc815ac543675 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:114dbf2908cd557ae828170e7a30971b36a80658a6b285550271ed4bf08a78e7
3
+ size 1120112640
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/393951591af519b35f197089e6cce648 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c92926bba8dd279bd2e903bdc448661f4c42a671832ce664d2f13043d3244b8
3
+ size 132972544
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/3951b7e6ea6af4f16a89ec7700c6bf7a ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff1643270c101d7be32113318ed39c572e2b48cdd315d8d90d8ec241eb85152c
3
+ size 70086656
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/476b28bf312620085f6a2a0a215398b2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e83e231ab19a9ac0ea530bea6c45e59092a53cfc58d79f2815626218b9ba00e4
3
+ size 34975744
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/4f0c00bd6c62b09500044074cd95c733 ADDED
Binary file (213 Bytes). View file
 
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/59814fab450c8db24dc8f54c7e5a5bc1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43b1075a27a767c419d1248a27a460f878c145c11497e0b4c7d5553b9b94cf94
3
+ size 280039424
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/5c355134126cba0188eb3359c050f340 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:914012752ded365b0030c12170763dccaa86fa09a5c3c3d139ead539be975744
3
+ size 1120129024
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/67ae3f487a818a239e757f6d22c4a2aa ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3e70ece0a03dc6f36db4b362e364b47f3b771ea22343075278969fc33b811f3
3
+ size 1120108544
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/70c219e996e6c66123717aea55e14525 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a5013c92cb82ca90b4a24160ad81f2e5a7e37cbd7fd8a98b91de57a099effb5
3
+ size 1120137216
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/84c30b89170ef6a2e8af11adcabd21be ADDED
Binary file (5.37 kB). View file
 
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/87e0a64717ea759c1be20bf4d9fa225e ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1784cb81a9c4ecff219147d35741f40f2a2c3db2ec43dcc34a024c648be2afd9
3
+ size 139915264
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/918e7c377fd7a6527543997a1116a21a ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb32c8637078da3b66b5a52d88848cb1132fa7333675aa42ab2f6ce3ab1816f3
3
+ size 559730688
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/94138de6f1ae96f4e3a3ba5a7e837665 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be17f7bd65ae8903fa2802e9dfbad6742f12319a24406a0a6b9e5b3a21d4378c
3
+ size 132939776
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/97c06f306c38269754a930b14e5a1a4b ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37a1ab93c0fbd657df6aa7ddf4a67563cc20e90543def006b690abadb10dc3db
3
+ size 139976704
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/aa82994c4bb23d2fcaf1dff385906a40 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:829b77976222ae07187b0e5e9f75294b609bc2646aa244e992c0c328fc88162b
3
+ size 1120120832
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c376fb098840eafe6c0357dddd39f675 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e1d6f916dd20a3f959feee0a7c8a0716252e2c6a3b91be0af16d4ad14cb3723
3
+ size 17408000
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c4ed3ef81206ba9443b2433f7f4c7fd0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba1e5d086d32792aebc414b601cc6cf1d5652ddd8af143ed580662111780e096
3
+ size 1957433344
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c8e7e58e56d51d4ee40301406c504601 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b5605792bdf64aa75b7ca851689e822e8ac065dda11f485185978f8975ff428
3
+ size 129714
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/c939b3c6a981afffa208fa8fb58b0c9d ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd52a714dfefda237c7379e3c08176ab0e0cec95cb5dc13003ece81c913b183a
3
+ size 1120112640
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d0e6fb07ccb3e52867497951f89d3306 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ee125e2a683cfd219b0b7b71b5a816227e057b54040b2087004c0475d554f6f
3
+ size 132845568
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/d92f425ddca63265c0f39f142d8b25de ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fb8e5a7bdbc96ff85dbfe86fc3707bbd0b1bfdfe83dce17d78fd024ee9e834a
3
+ size 279236608
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/dfedaf9c8fe7c197d541df8a0b04d209 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4aff9af79c941be4270cf119eb5b1227d5211d3ab7ba33a222fb802365905d4a
3
+ size 993787904
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/f93f9bdfcfc987a73372568d013c967c ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6bde0216979e02ee020206a8573d8bf0fdb3933f80991a008174b25851356283
3
+ size 8765440
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/d/ffad2779778a2a8a8d3316b3aad48f36 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9dbb4b4cde6175e97aa3af2f62b6bf4d300ec23536fca29cd157d7efb1d92ad5
3
+ size 280190976
v5__tvl__0_2__no_dyn/5000/params/ocdbt.process_0/manifest.ocdbt ADDED
Binary file (546 Bytes). View file
 
v5__tvl__0_2__no_dyn/config.yaml ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ !!python/object:openpi.training.config.TrainConfig
2
+ name: v5__tvl__0_2__no_dyn
3
+ project_name: openpi
4
+ exp_name: v5__tvl__0_2__no_dyn
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: 0.2
14
+ tv_loss_type: l2
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: false
77
+ use_dynamic_state: false
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: 15001
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
v5__tvl__0_2__no_dyn/wandb_id.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ ggc562h3