Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- pi0_nomagic_cube/cube/10000/_CHECKPOINT_METADATA +1 -1
- pi0_nomagic_cube/cube/10000/assets/ur10e/norm_stats.json +54 -54
- pi0_nomagic_cube/cube/10000/params/d/0bfb0da20db013400c7700e6599c951d +0 -0
- pi0_nomagic_cube/cube/10000/params/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/0eb91ed77e3168a3417af469a669bfd0 +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/305fbca3a1a7fe456083cfe240bae06f +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/6f7a8200b5181fe727af4d7835cca7eb +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/955622a56429f5f4c7dad64584baea47 +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/a2e1ca9f88f18c75d908f5610a77f930 +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/a7b76be2cd2cf4e0d5a9f883c278defc +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/ceeed61099c48c5f162748b5c9ff8bbd +0 -0
- pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/10000/train_state/d/e45334c557c2ae3394581eeff5f65d35 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/0beec69e157f447d217b8c438e351c70 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/1c4a27f6d4c5e05cbd24ed033a525d50 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/1e6b0970090fb26284ec628df4618369 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/20fab110a4ef5f6960d23ef494ebb9c5 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/292d80e2274c5151abb06343f35259ef +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/2f564ff4520dfdf1955fd4bae130c065 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/51f069a82cb9c6aa49aec8bb67fbb2f9 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/59e4788a628f10442caaa5c9eabe9fca +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/622a96541c5a20518bdf1034583276d6 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7035a3ca62b317d8913c3d3fdbc9f3a1 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7618a8b1eaf2350caa5f97912f9242d8 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7f812ec1cb5cd6217ad6940581c5e1be +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/8a4df1588af73f11cacb8fa6f72bd648 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/9013f95fcdfde7b733dff1e58269d2c8 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/954169160e46f3363502f38b4e4931ab +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/a784fc8d86d5d8c673dc7e904db240a6 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/e38232e3fe587c07ce3e3fe34da686c3 +0 -0
- pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/12000/_CHECKPOINT_METADATA +1 -0
- pi0_nomagic_cube/cube/12000/train_state/_sharding +1 -0
- pi0_nomagic_cube/cube/12000/train_state/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/d/0991ea07f18fd891f7f82ceb61fff75a +0 -0
- pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/d/2974a68c87787d972d177299f3e67d12 +0 -0
- pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/5000/_CHECKPOINT_METADATA +1 -1
- pi0_nomagic_cube/cube/5000/assets/ur10e/norm_stats.json +54 -54
- pi0_nomagic_cube/cube/5000/params/d/7fc0f570ec588553eca43991ce76d171 +0 -0
- pi0_nomagic_cube/cube/5000/params/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/9eefdc24d5c0a9e07073542524c846a8 +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/c9cb823e5d1592481aea332fd21a6dbd +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/cae18f9eefad68bf50794f7dee69b76d +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/d917e67ba66e59070eea2b9d4715c291 +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/e3d24940dc8662202eaf2283b1cc7440 +0 -0
- pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/manifest.ocdbt +0 -0
- pi0_nomagic_cube/cube/5000/train_state/d/856bc029188a8c667a4b3f04722f1ffa +0 -0
- pi0_nomagic_cube/cube/5000/train_state/manifest.ocdbt +0 -0
pi0_nomagic_cube/cube/10000/_CHECKPOINT_METADATA
CHANGED
@@ -1 +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":
|
|
|
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": 1751061563845013198, "commit_timestamp_nsecs": 1751061592988756982, "custom": {}}
|
pi0_nomagic_cube/cube/10000/assets/ur10e/norm_stats.json
CHANGED
@@ -2,13 +2,13 @@
|
|
2 |
"norm_stats": {
|
3 |
"state": {
|
4 |
"mean": [
|
5 |
-
|
6 |
-
-0.
|
7 |
-
-0.
|
8 |
-
0.
|
9 |
-
1.
|
10 |
-
|
11 |
-
0.
|
12 |
0.0,
|
13 |
0.0,
|
14 |
0.0,
|
@@ -36,13 +36,13 @@
|
|
36 |
0.0
|
37 |
],
|
38 |
"std": [
|
39 |
-
0.
|
40 |
-
0.
|
41 |
-
0.
|
42 |
-
1.
|
43 |
-
0.
|
44 |
-
1.
|
45 |
-
0.
|
46 |
0.0,
|
47 |
0.0,
|
48 |
0.0,
|
@@ -70,13 +70,13 @@
|
|
70 |
0.0
|
71 |
],
|
72 |
"q01": [
|
73 |
-
-0.
|
74 |
-
-0.
|
75 |
-
-0.
|
76 |
-
-3.
|
77 |
-
1.
|
78 |
-
-2.
|
79 |
-
0.
|
80 |
0.0,
|
81 |
0.0,
|
82 |
0.0,
|
@@ -104,12 +104,12 @@
|
|
104 |
0.0
|
105 |
],
|
106 |
"q99": [
|
107 |
-
0.
|
108 |
-
-0.
|
109 |
-
0.
|
110 |
-
3.
|
111 |
-
1.
|
112 |
-
2.
|
113 |
0.9631223678588867,
|
114 |
0.0,
|
115 |
0.0,
|
@@ -140,13 +140,13 @@
|
|
140 |
},
|
141 |
"actions": {
|
142 |
"mean": [
|
143 |
-
-0.
|
144 |
-
|
145 |
-
-0.
|
146 |
-
|
147 |
-
-0.
|
148 |
-
0.
|
149 |
-
0.
|
150 |
0.0,
|
151 |
0.0,
|
152 |
0.0,
|
@@ -174,13 +174,13 @@
|
|
174 |
0.0
|
175 |
],
|
176 |
"std": [
|
177 |
-
0.
|
178 |
-
0.
|
179 |
-
0.
|
180 |
-
0.
|
181 |
-
0.
|
182 |
-
0.
|
183 |
-
0.
|
184 |
0.0,
|
185 |
0.0,
|
186 |
0.0,
|
@@ -208,13 +208,13 @@
|
|
208 |
0.0
|
209 |
],
|
210 |
"q01": [
|
211 |
-
-0.
|
212 |
-
-0.
|
213 |
-
-0.
|
214 |
-
-0.
|
215 |
-
-0.
|
216 |
-
-0.
|
217 |
-
0.
|
218 |
0.0,
|
219 |
0.0,
|
220 |
0.0,
|
@@ -242,12 +242,12 @@
|
|
242 |
0.0
|
243 |
],
|
244 |
"q99": [
|
245 |
-
0.
|
246 |
-
0.
|
247 |
-
0.
|
248 |
-
0.
|
249 |
-
0.
|
250 |
-
0.
|
251 |
0.9631223678588867,
|
252 |
0.0,
|
253 |
0.0,
|
|
|
2 |
"norm_stats": {
|
3 |
"state": {
|
4 |
"mean": [
|
5 |
+
0.004325279500335455,
|
6 |
+
-0.6513074040412903,
|
7 |
+
-0.0028520487248897552,
|
8 |
+
0.38555651903152466,
|
9 |
+
1.522105097770691,
|
10 |
+
0.010858089663088322,
|
11 |
+
0.69498211145401,
|
12 |
0.0,
|
13 |
0.0,
|
14 |
0.0,
|
|
|
36 |
0.0
|
37 |
],
|
38 |
"std": [
|
39 |
+
0.19369296729564667,
|
40 |
+
0.08634746819734573,
|
41 |
+
0.10158221423625946,
|
42 |
+
1.9630029201507568,
|
43 |
+
0.028965409845113754,
|
44 |
+
1.4673080444335938,
|
45 |
+
0.3125312328338623,
|
46 |
0.0,
|
47 |
0.0,
|
48 |
0.0,
|
|
|
70 |
0.0
|
71 |
],
|
72 |
"q01": [
|
73 |
+
-0.5217792987823486,
|
74 |
+
-0.8281643986701965,
|
75 |
+
-0.11651834845542908,
|
76 |
+
-3.0988481044769287,
|
77 |
+
1.436232328414917,
|
78 |
+
-2.9956183433532715,
|
79 |
+
0.2908651530742645,
|
80 |
0.0,
|
81 |
0.0,
|
82 |
0.0,
|
|
|
104 |
0.0
|
105 |
],
|
106 |
"q99": [
|
107 |
+
0.3276591897010803,
|
108 |
+
-0.4599743187427521,
|
109 |
+
0.2406783401966095,
|
110 |
+
3.0950117111206055,
|
111 |
+
1.5655417442321777,
|
112 |
+
2.953033447265625,
|
113 |
0.9631223678588867,
|
114 |
0.0,
|
115 |
0.0,
|
|
|
140 |
},
|
141 |
"actions": {
|
142 |
"mean": [
|
143 |
+
-0.0005260662874206901,
|
144 |
+
0.00020510565082076937,
|
145 |
+
-0.00034676582436077297,
|
146 |
+
0.0004370925889816135,
|
147 |
+
-0.00009827941539697349,
|
148 |
+
0.00003694952465593815,
|
149 |
+
0.6949819922447205,
|
150 |
0.0,
|
151 |
0.0,
|
152 |
0.0,
|
|
|
174 |
0.0
|
175 |
],
|
176 |
"std": [
|
177 |
+
0.0038953707553446293,
|
178 |
+
0.004066351801156998,
|
179 |
+
0.0036852764897048473,
|
180 |
+
0.014113146811723709,
|
181 |
+
0.0026273801922798157,
|
182 |
+
0.0021055934485048056,
|
183 |
+
0.31253060698509216,
|
184 |
0.0,
|
185 |
0.0,
|
186 |
0.0,
|
|
|
208 |
0.0
|
209 |
],
|
210 |
"q01": [
|
211 |
+
-0.016714327037334442,
|
212 |
+
-0.013572817668318748,
|
213 |
+
-0.014442654326558113,
|
214 |
+
-0.04759380966424942,
|
215 |
+
-0.00866752490401268,
|
216 |
+
-0.0062255337834358215,
|
217 |
+
0.2908651530742645,
|
218 |
0.0,
|
219 |
0.0,
|
220 |
0.0,
|
|
|
242 |
0.0
|
243 |
],
|
244 |
"q99": [
|
245 |
+
0.006875673308968544,
|
246 |
+
0.015214478597044945,
|
247 |
+
0.01048312894999981,
|
248 |
+
0.05176602303981781,
|
249 |
+
0.007177960127592087,
|
250 |
+
0.006772628054022789,
|
251 |
0.9631223678588867,
|
252 |
0.0,
|
253 |
0.0,
|
pi0_nomagic_cube/cube/10000/params/d/0bfb0da20db013400c7700e6599c951d
ADDED
Binary file (2.2 kB). View file
|
|
pi0_nomagic_cube/cube/10000/params/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/10000/params/manifest.ocdbt and b/pi0_nomagic_cube/cube/10000/params/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/0eb91ed77e3168a3417af469a669bfd0
ADDED
Binary file (1.07 kB). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/305fbca3a1a7fe456083cfe240bae06f
ADDED
Binary file (213 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/6f7a8200b5181fe727af4d7835cca7eb
ADDED
Binary file (496 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/955622a56429f5f4c7dad64584baea47
ADDED
Binary file (444 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/a2e1ca9f88f18c75d908f5610a77f930
ADDED
Binary file (1.05 kB). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/a7b76be2cd2cf4e0d5a9f883c278defc
ADDED
Binary file (1.09 kB). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/d/ceeed61099c48c5f162748b5c9ff8bbd
ADDED
Binary file (537 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/manifest.ocdbt and b/pi0_nomagic_cube/cube/10000/params/ocdbt.process_0/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/10000/train_state/d/e45334c557c2ae3394581eeff5f65d35
ADDED
Binary file (3.74 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/10000/train_state/manifest.ocdbt and b/pi0_nomagic_cube/cube/10000/train_state/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/0beec69e157f447d217b8c438e351c70
ADDED
Binary file (1.33 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/1c4a27f6d4c5e05cbd24ed033a525d50
ADDED
Binary file (1.16 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/1e6b0970090fb26284ec628df4618369
ADDED
Binary file (555 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/20fab110a4ef5f6960d23ef494ebb9c5
ADDED
Binary file (1.17 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/292d80e2274c5151abb06343f35259ef
ADDED
Binary file (1.17 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/2f564ff4520dfdf1955fd4bae130c065
ADDED
Binary file (1.14 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/51f069a82cb9c6aa49aec8bb67fbb2f9
ADDED
Binary file (1.19 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/59e4788a628f10442caaa5c9eabe9fca
ADDED
Binary file (1.24 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/622a96541c5a20518bdf1034583276d6
ADDED
Binary file (537 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7035a3ca62b317d8913c3d3fdbc9f3a1
ADDED
Binary file (1.2 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7618a8b1eaf2350caa5f97912f9242d8
ADDED
Binary file (1.16 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/7f812ec1cb5cd6217ad6940581c5e1be
ADDED
Binary file (1.99 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/8a4df1588af73f11cacb8fa6f72bd648
ADDED
Binary file (1.21 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/9013f95fcdfde7b733dff1e58269d2c8
ADDED
Binary file (1.17 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/954169160e46f3363502f38b4e4931ab
ADDED
Binary file (171 Bytes). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/a784fc8d86d5d8c673dc7e904db240a6
ADDED
Binary file (1.17 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/d/e38232e3fe587c07ce3e3fe34da686c3
ADDED
Binary file (1.18 kB). View file
|
|
pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/manifest.ocdbt and b/pi0_nomagic_cube/cube/10000/train_state/ocdbt.process_0/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/12000/_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": 1751066474760143165, "commit_timestamp_nsecs": 1751066504715075741, "custom": {}}
|
pi0_nomagic_cube/cube/12000/train_state/_sharding
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"b3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJfbm9ybS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMC5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzEua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAua2V5LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAua2V5Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAub3V0LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAub3V0Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAucXVlcnkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAucXVlcnkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAudmFsdWUuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAudmFsdWUua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLmVtYmVkZGluZy5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLmVtYmVkZGluZy5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLmhlYWQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLmhlYWQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5QYWxpR2VtbWEuaW1nLnBvc19lbWJlZGRpbmcudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25faW5fcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25faW5fcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fb3V0X3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fb3V0X3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fdGltZV9tbHBfaW4uYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fdGltZV9tbHBfaW4ua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fdGltZV9tbHBfb3V0LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5hY3Rpb25fdGltZV9tbHBfb3V0Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5zdGF0ZV9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5tdS5zdGF0ZV9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJfbm9ybS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMC5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5MYXllck5vcm1fMS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzAuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzAua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzEuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NbHBCbG9ja18wLkRlbnNlXzEua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAua2V5LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAua2V5Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAub3V0LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAub3V0Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAucXVlcnkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAucXVlcnkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAudmFsdWUuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLlRyYW5zZm9ybWVyLmVuY29kZXJibG9jay5NdWx0aUhlYWREb3RQcm9kdWN0QXR0ZW50aW9uXzAudmFsdWUua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLmVtYmVkZGluZy5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLmVtYmVkZGluZy5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLmhlYWQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLmhlYWQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5QYWxpR2VtbWEuaW1nLnBvc19lbWJlZGRpbmcudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25faW5fcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25faW5fcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fb3V0X3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fb3V0X3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fdGltZV9tbHBfaW4uYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fdGltZV9tbHBfaW4ua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fdGltZV9tbHBfb3V0LmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5hY3Rpb25fdGltZV9tbHBfb3V0Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5zdGF0ZV9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMC5udS5zdGF0ZV9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","b3B0X3N0YXRlLjEuMi5jb3VudA==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","c3RlcA==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","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}]]}}"}
|
pi0_nomagic_cube/cube/12000/train_state/manifest.ocdbt
ADDED
Binary file (118 Bytes). View file
|
|
pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/d/0991ea07f18fd891f7f82ceb61fff75a
ADDED
Binary file (1.24 kB). View file
|
|
pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/d/2974a68c87787d972d177299f3e67d12
ADDED
Binary file (1.26 kB). View file
|
|
pi0_nomagic_cube/cube/12000/train_state/ocdbt.process_0/manifest.ocdbt
ADDED
Binary file (316 Bytes). View file
|
|
pi0_nomagic_cube/cube/5000/_CHECKPOINT_METADATA
CHANGED
@@ -1 +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":
|
|
|
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": 1751048820203497379, "commit_timestamp_nsecs": 1751048850301155024, "custom": {}}
|
pi0_nomagic_cube/cube/5000/assets/ur10e/norm_stats.json
CHANGED
@@ -2,13 +2,13 @@
|
|
2 |
"norm_stats": {
|
3 |
"state": {
|
4 |
"mean": [
|
5 |
-
|
6 |
-
-0.
|
7 |
-
-0.
|
8 |
-
0.
|
9 |
-
1.
|
10 |
-
|
11 |
-
0.
|
12 |
0.0,
|
13 |
0.0,
|
14 |
0.0,
|
@@ -36,13 +36,13 @@
|
|
36 |
0.0
|
37 |
],
|
38 |
"std": [
|
39 |
-
0.
|
40 |
-
0.
|
41 |
-
0.
|
42 |
-
1.
|
43 |
-
0.
|
44 |
-
1.
|
45 |
-
0.
|
46 |
0.0,
|
47 |
0.0,
|
48 |
0.0,
|
@@ -70,13 +70,13 @@
|
|
70 |
0.0
|
71 |
],
|
72 |
"q01": [
|
73 |
-
-0.
|
74 |
-
-0.
|
75 |
-
-0.
|
76 |
-
-3.
|
77 |
-
1.
|
78 |
-
-2.
|
79 |
-
0.
|
80 |
0.0,
|
81 |
0.0,
|
82 |
0.0,
|
@@ -104,12 +104,12 @@
|
|
104 |
0.0
|
105 |
],
|
106 |
"q99": [
|
107 |
-
0.
|
108 |
-
-0.
|
109 |
-
0.
|
110 |
-
3.
|
111 |
-
1.
|
112 |
-
2.
|
113 |
0.9631223678588867,
|
114 |
0.0,
|
115 |
0.0,
|
@@ -140,13 +140,13 @@
|
|
140 |
},
|
141 |
"actions": {
|
142 |
"mean": [
|
143 |
-
-0.
|
144 |
-
|
145 |
-
-0.
|
146 |
-
|
147 |
-
-0.
|
148 |
-
0.
|
149 |
-
0.
|
150 |
0.0,
|
151 |
0.0,
|
152 |
0.0,
|
@@ -174,13 +174,13 @@
|
|
174 |
0.0
|
175 |
],
|
176 |
"std": [
|
177 |
-
0.
|
178 |
-
0.
|
179 |
-
0.
|
180 |
-
0.
|
181 |
-
0.
|
182 |
-
0.
|
183 |
-
0.
|
184 |
0.0,
|
185 |
0.0,
|
186 |
0.0,
|
@@ -208,13 +208,13 @@
|
|
208 |
0.0
|
209 |
],
|
210 |
"q01": [
|
211 |
-
-0.
|
212 |
-
-0.
|
213 |
-
-0.
|
214 |
-
-0.
|
215 |
-
-0.
|
216 |
-
-0.
|
217 |
-
0.
|
218 |
0.0,
|
219 |
0.0,
|
220 |
0.0,
|
@@ -242,12 +242,12 @@
|
|
242 |
0.0
|
243 |
],
|
244 |
"q99": [
|
245 |
-
0.
|
246 |
-
0.
|
247 |
-
0.
|
248 |
-
0.
|
249 |
-
0.
|
250 |
-
0.
|
251 |
0.9631223678588867,
|
252 |
0.0,
|
253 |
0.0,
|
|
|
2 |
"norm_stats": {
|
3 |
"state": {
|
4 |
"mean": [
|
5 |
+
0.004325279500335455,
|
6 |
+
-0.6513074040412903,
|
7 |
+
-0.0028520487248897552,
|
8 |
+
0.38555651903152466,
|
9 |
+
1.522105097770691,
|
10 |
+
0.010858089663088322,
|
11 |
+
0.69498211145401,
|
12 |
0.0,
|
13 |
0.0,
|
14 |
0.0,
|
|
|
36 |
0.0
|
37 |
],
|
38 |
"std": [
|
39 |
+
0.19369296729564667,
|
40 |
+
0.08634746819734573,
|
41 |
+
0.10158221423625946,
|
42 |
+
1.9630029201507568,
|
43 |
+
0.028965409845113754,
|
44 |
+
1.4673080444335938,
|
45 |
+
0.3125312328338623,
|
46 |
0.0,
|
47 |
0.0,
|
48 |
0.0,
|
|
|
70 |
0.0
|
71 |
],
|
72 |
"q01": [
|
73 |
+
-0.5217792987823486,
|
74 |
+
-0.8281643986701965,
|
75 |
+
-0.11651834845542908,
|
76 |
+
-3.0988481044769287,
|
77 |
+
1.436232328414917,
|
78 |
+
-2.9956183433532715,
|
79 |
+
0.2908651530742645,
|
80 |
0.0,
|
81 |
0.0,
|
82 |
0.0,
|
|
|
104 |
0.0
|
105 |
],
|
106 |
"q99": [
|
107 |
+
0.3276591897010803,
|
108 |
+
-0.4599743187427521,
|
109 |
+
0.2406783401966095,
|
110 |
+
3.0950117111206055,
|
111 |
+
1.5655417442321777,
|
112 |
+
2.953033447265625,
|
113 |
0.9631223678588867,
|
114 |
0.0,
|
115 |
0.0,
|
|
|
140 |
},
|
141 |
"actions": {
|
142 |
"mean": [
|
143 |
+
-0.0005260662874206901,
|
144 |
+
0.00020510565082076937,
|
145 |
+
-0.00034676582436077297,
|
146 |
+
0.0004370925889816135,
|
147 |
+
-0.00009827941539697349,
|
148 |
+
0.00003694952465593815,
|
149 |
+
0.6949819922447205,
|
150 |
0.0,
|
151 |
0.0,
|
152 |
0.0,
|
|
|
174 |
0.0
|
175 |
],
|
176 |
"std": [
|
177 |
+
0.0038953707553446293,
|
178 |
+
0.004066351801156998,
|
179 |
+
0.0036852764897048473,
|
180 |
+
0.014113146811723709,
|
181 |
+
0.0026273801922798157,
|
182 |
+
0.0021055934485048056,
|
183 |
+
0.31253060698509216,
|
184 |
0.0,
|
185 |
0.0,
|
186 |
0.0,
|
|
|
208 |
0.0
|
209 |
],
|
210 |
"q01": [
|
211 |
+
-0.016714327037334442,
|
212 |
+
-0.013572817668318748,
|
213 |
+
-0.014442654326558113,
|
214 |
+
-0.04759380966424942,
|
215 |
+
-0.00866752490401268,
|
216 |
+
-0.0062255337834358215,
|
217 |
+
0.2908651530742645,
|
218 |
0.0,
|
219 |
0.0,
|
220 |
0.0,
|
|
|
242 |
0.0
|
243 |
],
|
244 |
"q99": [
|
245 |
+
0.006875673308968544,
|
246 |
+
0.015214478597044945,
|
247 |
+
0.01048312894999981,
|
248 |
+
0.05176602303981781,
|
249 |
+
0.007177960127592087,
|
250 |
+
0.006772628054022789,
|
251 |
0.9631223678588867,
|
252 |
0.0,
|
253 |
0.0,
|
pi0_nomagic_cube/cube/5000/params/d/7fc0f570ec588553eca43991ce76d171
ADDED
Binary file (2.24 kB). View file
|
|
pi0_nomagic_cube/cube/5000/params/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/5000/params/manifest.ocdbt and b/pi0_nomagic_cube/cube/5000/params/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/9eefdc24d5c0a9e07073542524c846a8
ADDED
Binary file (270 Bytes). View file
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/c9cb823e5d1592481aea332fd21a6dbd
ADDED
Binary file (1.66 kB). View file
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/cae18f9eefad68bf50794f7dee69b76d
ADDED
Binary file (1.04 kB). View file
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/d917e67ba66e59070eea2b9d4715c291
ADDED
Binary file (870 Bytes). View file
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/d/e3d24940dc8662202eaf2283b1cc7440
ADDED
Binary file (213 Bytes). View file
|
|
pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/manifest.ocdbt and b/pi0_nomagic_cube/cube/5000/params/ocdbt.process_0/manifest.ocdbt differ
|
|
pi0_nomagic_cube/cube/5000/train_state/d/856bc029188a8c667a4b3f04722f1ffa
ADDED
Binary file (3.66 kB). View file
|
|
pi0_nomagic_cube/cube/5000/train_state/manifest.ocdbt
CHANGED
Binary files a/pi0_nomagic_cube/cube/5000/train_state/manifest.ocdbt and b/pi0_nomagic_cube/cube/5000/train_state/manifest.ocdbt differ
|
|