diff --git a/pp__ee_baseline/10000/_CHECKPOINT_METADATA b/pp__ee_baseline/10000/_CHECKPOINT_METADATA
deleted file mode 100644
index a95f677b7aa19aa1c51c184a8a70be3ddd913ab3..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/_CHECKPOINT_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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": 1768748562436907773, "commit_timestamp_nsecs": 1768748881118661685, "custom_metadata": {}}
\ No newline at end of file
diff --git a/pp__ee_baseline/10000/assets/ur10e/norm_stats.json b/pp__ee_baseline/10000/assets/ur10e/norm_stats.json
deleted file mode 100644
index 8a1b886d6b668da778297b91b3d82381c2eab28a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/assets/ur10e/norm_stats.json
+++ /dev/null
@@ -1,1525 +0,0 @@
-{
-  "norm_stats": {
-    "state": {
-      "mean": [
-        0.05313729867339134,
-        0.0037223484832793474,
-        0.106409952044487,
-        0.0008666226058267057,
-        -0.00854077935218811,
-        0.01460699550807476,
-        0.5940247178077698
-      ],
-      "std": [
-        0.1176244392991066,
-        0.16962096095085144,
-        0.07773701101541519,
-        0.04537155479192734,
-        0.06616315245628357,
-        0.6038344502449036,
-        0.49107977747917175
-      ],
-      "q01": [
-        -0.2025573879778385,
-        -0.31106759839057924,
-        0.00890737680345774,
-        -0.11727181684970855,
-        -0.16711108471155167,
-        -1.423702568435669,
-        0.0
-      ],
-      "q99": [
-        0.29901714037060734,
-        0.4115018085956573,
-        0.3217185490995646,
-        0.13774808425903318,
-        0.19241279754638663,
-        1.598234023094177,
-        0.9998
-      ]
-    },
-    "actions": {
-      "mean": [
-        0.0001451610296498984,
-        0.00017207347264047712,
-        0.000048651279939804226,
-        0.00003654604734038003,
-        -0.000019767590856645256,
-        -0.00019861181499436498,
-        0.5944076776504517,
-        0.0002841090608853847,
-        0.00034722191048786044,
-        0.00011135452223243192,
-        0.00006827477045590058,
-        -0.000043589698179857805,
-        -0.0004028017574455589,
-        0.594025731086731,
-        0.00041486899135634303,
-        0.0005213047261349857,
-        0.00018569435633253306,
-        0.00009245114779332653,
-        -0.00007049143459880725,
-        -0.000616009347140789,
-        0.5935360789299011,
-        0.000542119552846998,
-        0.0006975074065849185,
-        0.00027102616149932146,
-        0.00011072693450842053,
-        -0.00010138262587133795,
-        -0.0008296052110381424,
-        0.593919575214386,
-        0.0006618681945838034,
-        0.0008754940354265273,
-        0.0003684645926114172,
-        0.00012383276771288365,
-        -0.000135598995257169,
-        -0.0010419095633551478,
-        0.5929783582687378,
-        0.0007787057547830045,
-        0.00105847988743335,
-        0.0004721089790109545,
-        0.00013335929543245584,
-        -0.00017038211808539927,
-        -0.0012720258673653007,
-        0.5933270454406738,
-        0.0008861633250489831,
-        0.001237610587850213,
-        0.0005911999032832682,
-        0.00013674379442818463,
-        -0.00020881379896309227,
-        -0.0014943619025871158,
-        0.5934669971466064,
-        0.0009874902898445725,
-        0.0014197468990460038,
-        0.0007170690805651248,
-        0.00013581194798462093,
-        -0.0002465852303430438,
-        -0.0017284756759181619,
-        0.5931888818740845,
-        0.0010826619109138846,
-        0.0016015341971069574,
-        0.0008584632305428386,
-        0.000128391373436898,
-        -0.00028913054848089814,
-        -0.0019626382272690535,
-        0.5930843949317932,
-        0.001174500328488648,
-        0.0017860672669485211,
-        0.0010054412996396422,
-        0.00012193901784485206,
-        -0.000334887852659449,
-        -0.0022029897663742304,
-        0.5935000777244568,
-        0.0012568456586450338,
-        0.0019742222502827644,
-        0.0011647609062492847,
-        0.00011021356476703659,
-        -0.00038211053470149636,
-        -0.0024467180483043194,
-        0.5931875109672546,
-        0.0013310370268300176,
-        0.0021642642095685005,
-        0.0013335321564227343,
-        0.00009949743980541825,
-        -0.0004296436673030257,
-        -0.0027013549115508795,
-        0.5931177139282227,
-        0.0014006041456013918,
-        0.002352928975597024,
-        0.0015152853447943926,
-        0.00008505478035658598,
-        -0.00048136088298633695,
-        -0.0029538467060774565,
-        0.5930479764938354,
-        0.0014608069323003292,
-        0.0025415997952222824,
-        0.001700855791568756,
-        0.00006727704749209806,
-        -0.0005341243813745677,
-        -0.003210614435374737,
-        0.5928036570549011,
-        0.0015100123127922416,
-        0.002731549320742488,
-        0.0018991812830790877,
-        0.0000493652478326112,
-        -0.0005809388239867985,
-        -0.003473253920674324,
-        0.5928736925125122,
-        0.0015514988917857409,
-        0.002922238316386938,
-        0.0021079820580780506,
-        0.000029167196771595627,
-        -0.0006329127354547381,
-        -0.0037316547241061926,
-        0.5928043127059937,
-        0.0015838317340239882,
-        0.0031066250521689653,
-        0.002319576218724251,
-        8.898059604689479e-6,
-        -0.0006800378323532641,
-        -0.004003639332950115,
-        0.5929091572761536,
-        0.0016050132689997554,
-        0.003295252099633217,
-        0.0025419394951313734,
-        -0.000013710295206692535,
-        -0.0007257868419401348,
-        -0.004269901663064957,
-        0.5930485725402832,
-        0.0016164487460628152,
-        0.0034825268667191267,
-        0.002774182241410017,
-        -0.000037491106922971085,
-        -0.0007724225870333612,
-        -0.004538912326097488,
-        0.5931875705718994,
-        0.001615947112441063,
-        0.003664180636405945,
-        0.0030124683398753405,
-        -0.00006199560448294505,
-        -0.0008142396691255271,
-        -0.00480045098811388,
-        0.5933974385261536,
-        0.0016095866449177265,
-        0.0038478909991681576,
-        0.003257976844906807,
-        -0.00008698971942067146,
-        -0.0008547022007405758,
-        -0.005078013986349106,
-        0.5936415791511536,
-        0.0015872354852035642,
-        0.004021089058369398,
-        0.0035109000746160746,
-        -0.00011568223999347538,
-        -0.0008942089625634253,
-        -0.005354971159249544,
-        0.5934317708015442,
-        0.0015591973206028342,
-        0.004198537208139896,
-        0.003773036180064082,
-        -0.00014495827781502157,
-        -0.0009303062106482685,
-        -0.005638875067234039,
-        0.5936412215232849,
-        0.0015177299501374364,
-        0.004366280045360327,
-        0.004040749743580818,
-        -0.00017588783521205187,
-        -0.0009663543896749616,
-        -0.005925025325268507,
-        0.593501627445221,
-        0.0014705799985677004,
-        0.004531881771981716,
-        0.00431715976446867,
-        -0.00021041395666543394,
-        -0.0010010612895712256,
-        -0.006207368802279234,
-        0.5935365557670593,
-        0.001413213089108467,
-        0.004692579619586468,
-        0.004599601496011019,
-        -0.0002457766968291253,
-        -0.0010327493073418736,
-        -0.006505406461656094,
-        0.5933269262313843,
-        0.0013434475986286998,
-        0.004848736338317394,
-        0.004890408832579851,
-        -0.00028711999766528606,
-        -0.00106858741492033,
-        -0.006807356141507626,
-        0.5931181311607361,
-        0.0012650334974750876,
-        0.004994421266019344,
-        0.0051898653618991375,
-        -0.00033172621624544263,
-        -0.0010990891605615616,
-        -0.0071100820787250996,
-        0.5929442048072815,
-        0.0011793962912634015,
-        0.005135250277817249,
-        0.005499166436493397,
-        -0.000378662021830678,
-        -0.0011317729949951172,
-        -0.007409243378788233,
-        0.5926647782325745,
-        0.0010838431771844625,
-        0.005268934182822704,
-        0.005817332770675421,
-        -0.0004281000583432615,
-        -0.001166283618658781,
-        -0.007717216853052378,
-        0.5926998853683472,
-        0.0009804241126403213,
-        0.0053992304019629955,
-        0.0061420174315571785,
-        -0.00048094423254951835,
-        -0.0011993126245215535,
-        -0.008030007593333721,
-        0.5924902558326721,
-        0.0008692154078744352,
-        0.005519451107829809,
-        0.006479023955762386,
-        -0.000537763989996165,
-        -0.001236212207004428,
-        -0.008356506004929543,
-        0.5929440855979919,
-        0.0007513284799642861,
-        0.00563538633286953,
-        0.006823732051998377,
-        -0.000594516284763813,
-        -0.0012742418330162764,
-        -0.008685515262186527,
-        0.5926643013954163,
-        0.0006232408923096955,
-        0.005743546411395073,
-        0.007178108207881451,
-        -0.0006553126149810851,
-        -0.0013130278093740344,
-        -0.00901535339653492,
-        0.5930123329162598,
-        0.00048613501712679863,
-        0.005843072663992643,
-        0.007545872125774622,
-        -0.0007216006051748991,
-        -0.001357274129986763,
-        -0.009348330087959766,
-        0.5926995277404785,
-        0.000340388563927263,
-        0.005928103346377611,
-        0.007921233773231506,
-        -0.0007906133541837335,
-        -0.0013983722310513258,
-        -0.00968977902084589,
-        0.5930134654045105,
-        0.00018408334290143102,
-        0.006012112833559513,
-        0.008309606462717056,
-        -0.0008611961966380477,
-        -0.00144277885556221,
-        -0.010034735314548016,
-        0.5933630466461182,
-        0.000022206833818927407,
-        0.0060895816422998905,
-        0.008708948269486427,
-        -0.0009344238787889481,
-        -0.0014912645565345883,
-        -0.01038195751607418,
-        0.5936756134033203,
-        -0.00014933485363144428,
-        0.006156634073704481,
-        0.009113290347158909,
-        -0.001009406871162355,
-        -0.0015364158898591995,
-        -0.01074444130063057,
-        0.5944079756736755,
-        -0.0003295359492767602,
-        0.006216940935701132,
-        0.009534457698464394,
-        -0.0010867133969441056,
-        -0.0015858380356803536,
-        -0.011105277575552464,
-        0.5946523547172546,
-        -0.0005184217588976026,
-        0.006267391610890627,
-        0.009968035854399204,
-        -0.00116726232226938,
-        -0.001635841210372746,
-        -0.011476362124085426,
-        0.5947223901748657,
-        -0.0007183448178693652,
-        0.006307759787887335,
-        0.010411876253783703,
-        -0.0012491614324972034,
-        -0.0016832650871947408,
-        -0.011858409270644188,
-        0.5945469737052917,
-        -0.000922650913707912,
-        0.006340802181512117,
-        0.01086830347776413,
-        -0.0013331895461305976,
-        -0.0017334959702566266,
-        -0.012237261049449444,
-        0.5951407551765442,
-        -0.001138067222200334,
-        0.006367202382534742,
-        0.011336571536958218,
-        -0.0014204677427187562,
-        -0.001783861662261188,
-        -0.012612935155630112,
-        0.5946868658065796,
-        -0.0013612991897389293,
-        0.006382528692483902,
-        0.011817813850939274,
-        -0.0015092657413333654,
-        -0.0018364684656262398,
-        -0.013003477826714516,
-        0.595176637172699,
-        -0.0015915771946310997,
-        0.0063857706263661385,
-        0.01231638714671135,
-        -0.0015994333662092686,
-        -0.0018854403169825673,
-        -0.013393816538155079,
-        0.5955926179885864,
-        -0.001827688654884696,
-        0.006383033003658056,
-        0.012827028520405293,
-        -0.0016901197377592325,
-        -0.0019397168653085828,
-        -0.01378911454230547,
-        0.5955935120582581,
-        -0.002072748262435198,
-        0.006370946299284697,
-        0.013353490270674229,
-        -0.0017832767916843295,
-        -0.001991107827052474,
-        -0.014198213815689087,
-        0.595174252986908,
-        -0.0023247504141181707,
-        0.006349846720695496,
-        0.013896841555833817,
-        -0.0018769301241263747,
-        -0.0020415729377418756,
-        -0.014593549072742462,
-        0.5950352549552917,
-        -0.0025832781102508307,
-        0.0063181049190461636,
-        0.01445536408573389,
-        -0.0019729691557586193,
-        -0.002093689749017358,
-        -0.015006774105131626,
-        0.5953840613365173
-      ],
-      "std": [
-        0.002411161782220006,
-        0.0035220407880842686,
-        0.0029013415332883596,
-        0.0013258246472105384,
-        0.0020179853308945894,
-        0.010011205449700356,
-        0.4910063147544861,
-        0.0048064119182527065,
-        0.007010283879935741,
-        0.005793850403279066,
-        0.0026270404923707247,
-        0.004016784951090813,
-        0.019959986209869385,
-        0.4910795986652374,
-        0.0072022695094347,
-        0.01047250535339117,
-        0.008696537464857101,
-        0.003923765849322081,
-        0.006003526505082846,
-        0.029928741976618767,
-        0.4911730885505676,
-        0.009601284749805927,
-        0.013944732025265694,
-        0.011596043594181538,
-        0.0051947347819805145,
-        0.007982160896062851,
-        0.039855923503637314,
-        0.49109989404678345,
-        0.011995380744338036,
-        0.017401276156306267,
-        0.01450654398649931,
-        0.006452825851738453,
-        0.009959585964679718,
-        0.049764279276132584,
-        0.4912789762020111,
-        0.014385105110704899,
-        0.020854491740465164,
-        0.017407990992069244,
-        0.007683087605983019,
-        0.011908459477126598,
-        0.059651929885149,
-        0.4912128448486328,
-        0.016765346750617027,
-        0.024284135550260544,
-        0.02030768245458603,
-        0.008882720954716206,
-        0.013833398930728436,
-        0.06945177167654037,
-        0.4911862313747406,
-        0.01913418062031269,
-        0.02769860439002514,
-        0.023200584575533867,
-        0.010069170035421848,
-        0.01573152281343937,
-        0.07920067012310028,
-        0.49123910069465637,
-        0.021495427936315536,
-        0.03108437918126583,
-        0.02608773671090603,
-        0.011218286119401455,
-        0.017600294202566147,
-        0.08888415992259979,
-        0.4912588894367218,
-        0.023848513141274452,
-        0.03446579724550247,
-        0.028954684734344482,
-        0.012335631065070629,
-        0.0194363035261631,
-        0.0984557494521141,
-        0.4911799430847168,
-        0.02618129923939705,
-        0.0377996489405632,
-        0.031812794506549835,
-        0.013426545076072216,
-        0.021240200847387314,
-        0.10793018341064453,
-        0.4912393391132355,
-        0.02850136160850525,
-        0.04113452509045601,
-        0.03465236350893974,
-        0.01448102854192257,
-        0.023004556074738503,
-        0.11730524897575378,
-        0.491252601146698,
-        0.030810443684458733,
-        0.0444553941488266,
-        0.03749171271920204,
-        0.01552011538296938,
-        0.02474648877978325,
-        0.12660986185073853,
-        0.49126577377319336,
-        0.033104438334703445,
-        0.047724947333335876,
-        0.04029623046517372,
-        0.01652422547340393,
-        0.026438293978571892,
-        0.1357518434524536,
-        0.4913119971752167,
-        0.03538680821657181,
-        0.05097673833370209,
-        0.04309060424566269,
-        0.017495229840278625,
-        0.02809188887476921,
-        0.14484061300754547,
-        0.49129876494407654,
-        0.037644777446985245,
-        0.054201558232307434,
-        0.045869018882513046,
-        0.018448922783136368,
-        0.029708286747336388,
-        0.1537874937057495,
-        0.4913118779659271,
-        0.03989279642701149,
-        0.05739012360572815,
-        0.04861682280898094,
-        0.019360067322850227,
-        0.031284283846616745,
-        0.16262821853160858,
-        0.4912920594215393,
-        0.04212509095668793,
-        0.060563746839761734,
-        0.05134649947285652,
-        0.020253626629710197,
-        0.03281664475798607,
-        0.17136116325855255,
-        0.4912656545639038,
-        0.04433578997850418,
-        0.06369910389184952,
-        0.05404900759458542,
-        0.021123351529240608,
-        0.03430699184536934,
-        0.1799684315919876,
-        0.4912393391132355,
-        0.0465271882712841,
-        0.06679350882768631,
-        0.05673054978251457,
-        0.02196338027715683,
-        0.035762034356594086,
-        0.1884494125843048,
-        0.49119946360588074,
-        0.04870201274752617,
-        0.0698690414428711,
-        0.0593833364546299,
-        0.02277866192162037,
-        0.03716905042529106,
-        0.19684050977230072,
-        0.4911530017852783,
-        0.05086364969611168,
-        0.0729081928730011,
-        0.06201274320483208,
-        0.023575762286782265,
-        0.03854267671704292,
-        0.20511800050735474,
-        0.49119293689727783,
-        0.052999626845121384,
-        0.07590420544147491,
-        0.06461741775274277,
-        0.024347545579075813,
-        0.03986750543117523,
-        0.21329139173030853,
-        0.4911530613899231,
-        0.05511622130870819,
-        0.07886126637458801,
-        0.06719071418046951,
-        0.02509898692369461,
-        0.04115770384669304,
-        0.22134073078632355,
-        0.4911796748638153,
-        0.05720451846718788,
-        0.0817820280790329,
-        0.06973543763160706,
-        0.02582353726029396,
-        0.04240873083472252,
-        0.22928737103939056,
-        0.49117299914360046,
-        0.05928165465593338,
-        0.08468781411647797,
-        0.07225947827100754,
-        0.026536913588643074,
-        0.0436241552233696,
-        0.23713935911655426,
-        0.4912128746509552,
-        0.06133345514535904,
-        0.08753105252981186,
-        0.07475195825099945,
-        0.02723076194524765,
-        0.04480902478098869,
-        0.2449100911617279,
-        0.49125248193740845,
-        0.06336378306150436,
-        0.09035338461399078,
-        0.07722371071577072,
-        0.027915935963392258,
-        0.04596531763672829,
-        0.25256893038749695,
-        0.49128544330596924,
-        0.06536917388439178,
-        0.0931331068277359,
-        0.07966616004705429,
-        0.028593653813004494,
-        0.04708719253540039,
-        0.26014769077301025,
-        0.49133822321891785,
-        0.06735322624444962,
-        0.09587426483631134,
-        0.08208146691322327,
-        0.029253413900732994,
-        0.04818820580840111,
-        0.2676284909248352,
-        0.4913316071033478,
-        0.06931550055742264,
-        0.09857368469238281,
-        0.084467314183712,
-        0.029913172125816345,
-        0.049258600920438766,
-        0.27503275871276855,
-        0.4913710951805115,
-        0.07124854624271393,
-        0.10124263167381287,
-        0.08683151751756668,
-        0.03056473471224308,
-        0.05031215026974678,
-        0.2823602259159088,
-        0.4912854731082916,
-        0.07316309213638306,
-        0.10387503355741501,
-        0.08917013555765152,
-        0.031205736100673676,
-        0.051339127123355865,
-        0.2896381616592407,
-        0.491338312625885,
-        0.0750511884689331,
-        0.10646932572126389,
-        0.09148410707712173,
-        0.03184282034635544,
-        0.05234614759683609,
-        0.2968275845050812,
-        0.49127253890037537,
-        0.07691479474306107,
-        0.10901817679405212,
-        0.0937764048576355,
-        0.032469242811203,
-        0.053335245698690414,
-        0.30398809909820557,
-        0.49133166670799255,
-        0.07874985039234161,
-        0.11152230948209763,
-        0.0960441529750824,
-        0.03308483585715294,
-        0.054303981363773346,
-        0.311068594455719,
-        0.49127233028411865,
-        0.08056764304637909,
-        0.11400332301855087,
-        0.09828827530145645,
-        0.03369200974702835,
-        0.055251531302928925,
-        0.31809186935424805,
-        0.49120599031448364,
-        0.08236116915941238,
-        0.11642979085445404,
-        0.10051009804010391,
-        0.034294869750738144,
-        0.05618305504322052,
-        0.32508960366249084,
-        0.4911465048789978,
-        0.0841296911239624,
-        0.11883242428302765,
-        0.10270429402589798,
-        0.034887079149484634,
-        0.0570959635078907,
-        0.3320097327232361,
-        0.4910062551498413,
-        0.08587083965539932,
-        0.12118621915578842,
-        0.10488012433052063,
-        0.0354686863720417,
-        0.05799192935228348,
-        0.3389015197753906,
-        0.49095919728279114,
-        0.08759026974439621,
-        0.12350363284349442,
-        0.10703437030315399,
-        0.03604672849178314,
-        0.05887560918927193,
-        0.3457629084587097,
-        0.4909456968307495,
-        0.08928582817316055,
-        0.12578748166561127,
-        0.10916976630687714,
-        0.03662176802754402,
-        0.059738170355558395,
-        0.35257115960121155,
-        0.49097952246665955,
-        0.09095519781112671,
-        0.1280231475830078,
-        0.11128115653991699,
-        0.03717926889657974,
-        0.06058768928050995,
-        0.35931187868118286,
-        0.4908647835254669,
-        0.09259875863790512,
-        0.13021688163280487,
-        0.11337121576070786,
-        0.03773583471775055,
-        0.061423756182193756,
-        0.3660118579864502,
-        0.4909525215625763,
-        0.09422165900468826,
-        0.13238278031349182,
-        0.11544124782085419,
-        0.03828861564397812,
-        0.06224775314331055,
-        0.3726641535758972,
-        0.4908578395843506,
-        0.09581982344388962,
-        0.13450181484222412,
-        0.11749458312988281,
-        0.03884110972285271,
-        0.06304997950792313,
-        0.3792511522769928,
-        0.49077698588371277,
-        0.09739767760038376,
-        0.13658517599105835,
-        0.11952484399080276,
-        0.03938090428709984,
-        0.06384975463151932,
-        0.3857972025871277,
-        0.49077683687210083,
-        0.09894638508558273,
-        0.1386304795742035,
-        0.12153798341751099,
-        0.03992167487740517,
-        0.06463739275932312,
-        0.39230042695999146,
-        0.4908582866191864,
-        0.10048258304595947,
-        0.14064013957977295,
-        0.12353785336017609,
-        0.040460873395204544,
-        0.06541262567043304,
-        0.398761510848999,
-        0.4908851981163025,
-        0.10199275612831116,
-        0.14262202382087708,
-        0.1255190521478653,
-        0.04099829122424126,
-        0.0661839172244072,
-        0.40516018867492676,
-        0.4908175766468048
-      ],
-      "q01": [
-        -0.007347712090983987,
-        -0.011486765748262403,
-        -0.005602361589856446,
-        -0.004156268547289073,
-        -0.007298904010653494,
-        -0.03456402559429407,
-        0.0,
-        -0.014532031904906033,
-        -0.023063078013062475,
-        -0.011146851527690887,
-        -0.008034631153941155,
-        -0.014402847960591318,
-        -0.0688410488665104,
-        0.0,
-        -0.02174613598883152,
-        -0.03454083656966686,
-        -0.016602809804677963,
-        -0.012004367487877608,
-        -0.02169839859008789,
-        -0.10323694312572479,
-        0.0,
-        -0.02924629563242197,
-        -0.046255945521593095,
-        -0.02208325160816312,
-        -0.01599733569249511,
-        -0.029179477728903294,
-        -0.13703150334358216,
-        0.0,
-        -0.0364012400791049,
-        -0.057891583660244945,
-        -0.027614226043969393,
-        -0.020049774289131163,
-        -0.03620416237413883,
-        -0.17225709611177442,
-        0.0,
-        -0.04366054475307465,
-        -0.06915460828691722,
-        -0.03313590524941683,
-        -0.023996917428076275,
-        -0.0432967299669981,
-        -0.20717854452133178,
-        0.0,
-        -0.0507191886767745,
-        -0.08091499265134335,
-        -0.038471172228455544,
-        -0.02769389592409134,
-        -0.05027697439491748,
-        -0.240767266279459,
-        0.0,
-        -0.05764442007690668,
-        -0.0928664004802704,
-        -0.04389051187932491,
-        -0.031378988534212116,
-        -0.05700045920312405,
-        -0.27346896691322325,
-        0.0,
-        -0.06481191511154175,
-        -0.10385700676739215,
-        -0.04937767976522446,
-        -0.034899483323097225,
-        -0.0640901755988598,
-        -0.3072322327375412,
-        0.0,
-        -0.07216909334659577,
-        -0.11465946695804596,
-        -0.05446739228963852,
-        -0.0384699240922928,
-        -0.07105188292860984,
-        -0.34030354830026627,
-        0.0,
-        -0.0791815688431263,
-        -0.12541235695779324,
-        -0.05973851401358843,
-        -0.04202334811687469,
-        -0.07781512894928456,
-        -0.37346914051771163,
-        0.0,
-        -0.08563443173170089,
-        -0.13601192309856414,
-        -0.06484265891909599,
-        -0.045433390045166,
-        -0.0843420141339302,
-        -0.40387091467380526,
-        0.0,
-        -0.09227466106414796,
-        -0.14718177989125253,
-        -0.06990853486061097,
-        -0.04866062937080862,
-        -0.09135572135448455,
-        -0.43357558735609053,
-        0.0,
-        -0.09871668579578399,
-        -0.15755651119351388,
-        -0.07494964042305946,
-        -0.05228844651579856,
-        -0.0973639042377472,
-        -0.4662614274024963,
-        0.0,
-        -0.10474277555942535,
-        -0.16660683050751687,
-        -0.07973185749202967,
-        -0.055503451734781245,
-        -0.1043376196026802,
-        -0.49795184282064436,
-        0.0,
-        -0.11146926714479923,
-        -0.17675033633708956,
-        -0.08481720062792301,
-        -0.05867235680818558,
-        -0.11146564957499505,
-        -0.5234594867229462,
-        0.0,
-        -0.1183029289394617,
-        -0.18591568176746368,
-        -0.08976946114003659,
-        -0.06189141765832901,
-        -0.11618306014537812,
-        -0.5559864712238312,
-        0.0,
-        -0.1254465562880039,
-        -0.1967654939889908,
-        -0.09452381261885166,
-        -0.06458974866867065,
-        -0.12229588437080384,
-        -0.5838854856491089,
-        0.0,
-        -0.13160774660110475,
-        -0.20561206495761872,
-        -0.09930612608790398,
-        -0.06763142464160918,
-        -0.12729683189988136,
-        -0.6118898140192032,
-        0.0,
-        -0.13866937249898909,
-        -0.21400038709640504,
-        -0.10383093198537827,
-        -0.07044867560863496,
-        -0.1330213810920715,
-        -0.6420615591049195,
-        0.0,
-        -0.14518664595484734,
-        -0.22208737790584565,
-        -0.10839852929115296,
-        -0.07363164942264558,
-        -0.13909859127998353,
-        -0.6659443435192108,
-        0.0,
-        -0.15202519185543062,
-        -0.23166744391918181,
-        -0.1132360664010048,
-        -0.07554293347597121,
-        -0.1431354057431221,
-        -0.6935547601938247,
-        0.0,
-        -0.1575157383620739,
-        -0.24115877079963685,
-        -0.11769197665154933,
-        -0.07880985802412033,
-        -0.14886759107112885,
-        -0.717457844042778,
-        0.0,
-        -0.1637449370443821,
-        -0.25002506982684136,
-        -0.12177611397504806,
-        -0.08111752626895904,
-        -0.15336844004392625,
-        -0.7450204710006714,
-        0.0,
-        -0.1702753342092037,
-        -0.25778238211274146,
-        -0.12590762196183203,
-        -0.08350524621009828,
-        -0.15739910807609558,
-        -0.7672739201784134,
-        0.0,
-        -0.17570297884345054,
-        -0.26699419630765914,
-        -0.12999144119024275,
-        -0.08586254574060437,
-        -0.15982728340625763,
-        -0.7919603824377061,
-        0.0,
-        -0.18209079703688621,
-        -0.2762185108423233,
-        -0.1341805691242218,
-        -0.08846430575847625,
-        -0.1640431535243988,
-        -0.8073872842788696,
-        0.0,
-        -0.18759027983546256,
-        -0.28415180140137675,
-        -0.13808169324100017,
-        -0.09098501863479613,
-        -0.1681320061683655,
-        -0.8263913946390152,
-        0.0,
-        -0.193171888589859,
-        -0.29068409420847896,
-        -0.14178944163620472,
-        -0.09304863989353179,
-        -0.1707449821949005,
-        -0.8507119262695311,
-        0.0,
-        -0.19844869979619978,
-        -0.2987963397145271,
-        -0.14549240586161613,
-        -0.09557797043323518,
-        -0.1739663097858429,
-        -0.8648041958332062,
-        0.0,
-        -0.20466184456944464,
-        -0.3049391875267029,
-        -0.14948453903496267,
-        -0.09817119854688644,
-        -0.17706590107083323,
-        -0.8849965750694275,
-        0.0,
-        -0.20983124628067015,
-        -0.3116496807098389,
-        -0.15297152189016341,
-        -0.10035031213760376,
-        -0.18140975254774094,
-        -0.9072247592926024,
-        0.0,
-        -0.215193039560318,
-        -0.31998531384468076,
-        -0.1565369277715683,
-        -0.102377852499485,
-        -0.1837332302033901,
-        -0.9293165444850922,
-        0.0,
-        -0.22114226788282393,
-        -0.32775930384397506,
-        -0.16037838854789732,
-        -0.10488562031984328,
-        -0.18766805601119993,
-        -0.9526331702470778,
-        0.0,
-        -0.22637617546319963,
-        -0.33561479420661927,
-        -0.16403944179415703,
-        -0.10686019870638847,
-        -0.19061433601379393,
-        -0.9705972829818725,
-        0.0,
-        -0.23192334132194517,
-        -0.3404881935119629,
-        -0.16749516949653626,
-        -0.10840021744370465,
-        -0.19327300127148628,
-        -0.9914580883026123,
-        0.0,
-        -0.23655534625053407,
-        -0.3483747916221619,
-        -0.1713618674516678,
-        -0.11093877178430556,
-        -0.19658433060646058,
-        -1.0074259456634522,
-        0.0,
-        -0.24189636796712874,
-        -0.35468626945018766,
-        -0.174676346886158,
-        -0.1137608123421669,
-        -0.1998320359289646,
-        -1.02401462726593,
-        0.0,
-        -0.24619000346064568,
-        -0.3617886746406555,
-        -0.17737202043533326,
-        -0.11608646607995032,
-        -0.20212158007621764,
-        -1.0396305271863937,
-        0.0,
-        -0.25069174761176105,
-        -0.3690223491430283,
-        -0.18019738556742668,
-        -0.11791741763353347,
-        -0.20542705215215684,
-        -1.056211252117157,
-        0.0,
-        -0.25575273798704146,
-        -0.37492762922048567,
-        -0.18413128078579902,
-        -0.11963063821196557,
-        -0.2084871412396431,
-        -1.0777596963644027,
-        0.0,
-        -0.26051444165706633,
-        -0.38124460592269893,
-        -0.18717371021509172,
-        -0.12171284379959108,
-        -0.21154849707484244,
-        -1.0918209570646285,
-        0.0,
-        -0.2645109035611153,
-        -0.38571969106197357,
-        -0.19055300951004028,
-        -0.12364264481067655,
-        -0.21497644165158272,
-        -1.1041330133199692,
-        0.0,
-        -0.26856355381011965,
-        -0.3920482913136482,
-        -0.19341499418020247,
-        -0.12596707943677904,
-        -0.21652717173099517,
-        -1.1284716510772705,
-        0.0,
-        -0.273536373257637,
-        -0.3985861983895302,
-        -0.19648525762557983,
-        -0.12863396686315537,
-        -0.2178426690995693,
-        -1.1498000781059265,
-        0.0,
-        -0.2773340993881226,
-        -0.4043089840650559,
-        -0.19916273007392882,
-        -0.1301111794948578,
-        -0.21953763974308968,
-        -1.1649874792098998,
-        0.0,
-        -0.28152843511104586,
-        -0.4105489842891693,
-        -0.20206411379575728,
-        -0.13238328326940535,
-        -0.22030045819282532,
-        -1.186898805141449,
-        0.0,
-        -0.28521604499816894,
-        -0.41719991471767426,
-        -0.20488486919403076,
-        -0.13367428585290908,
-        -0.2215240351676941,
-        -1.2064387265205383,
-        0.0,
-        -0.28908531114459035,
-        -0.4219293355822563,
-        -0.20779289478063584,
-        -0.1346039095699787,
-        -0.22363287283182146,
-        -1.225643195438385,
-        0.0,
-        -0.294126475507021,
-        -0.42881255321502687,
-        -0.21011174390912055,
-        -0.13763879417181013,
-        -0.22653749623298647,
-        -1.2400950877666475,
-        0.0
-      ],
-      "q99": [
-        0.00760505946278572,
-        0.01227881843298674,
-        0.009394833338446915,
-        0.0041071553006768215,
-        0.005846777586638929,
-        0.03394978983253241,
-        0.9998,
-        0.015092343334481124,
-        0.024545711845159537,
-        0.018547782172262668,
-        0.008136489582061764,
-        0.011534944232553239,
-        0.0680388870239258,
-        0.9998,
-        0.022567954233288764,
-        0.03644986734837294,
-        0.0278723166257143,
-        0.012022978414595133,
-        0.017261751674115658,
-        0.1018139317035675,
-        0.9998,
-        0.030176736226677894,
-        0.048319679215550415,
-        0.037043299572169786,
-        0.015823813125491137,
-        0.022935147598385802,
-        0.13534477652311327,
-        0.9998,
-        0.03749606138020754,
-        0.05974762381315232,
-        0.046320767011493444,
-        0.01970605190098286,
-        0.028365900403261188,
-        0.16783747158050544,
-        0.9998,
-        0.04486110335588456,
-        0.07147922609001398,
-        0.055334820427000525,
-        0.023239636448025694,
-        0.034125556316971783,
-        0.2017985020875931,
-        0.9998,
-        0.052173979285359384,
-        0.08346393676698208,
-        0.06446339850723744,
-        0.026847184759378445,
-        0.03955972051620485,
-        0.23437213500142096,
-        0.9998,
-        0.059150983133912094,
-        0.09537746536731717,
-        0.0733001764550805,
-        0.029705484813451755,
-        0.04516481494158506,
-        0.26646220774650575,
-        0.9998,
-        0.06645039129257202,
-        0.1076395375341177,
-        0.08239804291725158,
-        0.0328832214474678,
-        0.04995994846820831,
-        0.29610841860771175,
-        0.9998,
-        0.07374503636360169,
-        0.11985380845069887,
-        0.09145532561540604,
-        0.035987181842327104,
-        0.05490816103816032,
-        0.3266934569597244,
-        0.9998,
-        0.08122003978192807,
-        0.1319003387391567,
-        0.10024555467069149,
-        0.038626870799064655,
-        0.05953236908912657,
-        0.35580547877550117,
-        0.9998,
-        0.08822669934630395,
-        0.14304091303348543,
-        0.1088144795835018,
-        0.041924731212854394,
-        0.06486271733045579,
-        0.3876368781208992,
-        0.9998,
-        0.09569223463535309,
-        0.1534084576368332,
-        0.11751553359031677,
-        0.04469002693891522,
-        0.06991170048713685,
-        0.41592443135976787,
-        0.9998,
-        0.10324303380250932,
-        0.1640547949969769,
-        0.1258540060549974,
-        0.04742067799866201,
-        0.0741334307938814,
-        0.4491768553853035,
-        0.9998,
-        0.1099683414697647,
-        0.17510256106853483,
-        0.13403582906574008,
-        0.050331409102678304,
-        0.07791369587183,
-        0.475221610414982,
-        0.9998,
-        0.1170920331031084,
-        0.1857688842058181,
-        0.14203765797317025,
-        0.05343856880664827,
-        0.0820521941632032,
-        0.5052882493734359,
-        0.9998,
-        0.12414569870531561,
-        0.19753187570571906,
-        0.14974090793728828,
-        0.0557895161241293,
-        0.0865580612182617,
-        0.5326116788625717,
-        0.9998,
-        0.13112460071146492,
-        0.20844777574539186,
-        0.15755365291237833,
-        0.05866813796758652,
-        0.09100452079772947,
-        0.5617906215190887,
-        0.9998,
-        0.1379249924659729,
-        0.21917983024120336,
-        0.16431246396899224,
-        0.06069063851833345,
-        0.09500974670350554,
-        0.5912433532238006,
-        0.9998,
-        0.14485807346105578,
-        0.2286938194990158,
-        0.1716713245987892,
-        0.0637721403002739,
-        0.09919737915396692,
-        0.6155789003372192,
-        0.9998,
-        0.15117210847139356,
-        0.23827072948217387,
-        0.1779224501848221,
-        0.06597466281652448,
-        0.10370546174049378,
-        0.644374561071396,
-        0.9998,
-        0.15787530031800268,
-        0.24751081448793422,
-        0.18454510100185867,
-        0.06831172816753389,
-        0.10792329218983648,
-        0.6659029854536058,
-        0.9998,
-        0.1641180765330792,
-        0.2572556215524673,
-        0.19053607739508155,
-        0.07030175954103468,
-        0.11277426195144652,
-        0.686683092904091,
-        0.9998,
-        0.16966928005814552,
-        0.26658033823966976,
-        0.19653976464271544,
-        0.07203448786735533,
-        0.11643967783451081,
-        0.7125654235839844,
-        0.9998,
-        0.17621982062458996,
-        0.27701201248764995,
-        0.20197029424905777,
-        0.07416592988967896,
-        0.12035323338508608,
-        0.7360334300994871,
-        0.9998,
-        0.1819627068042755,
-        0.2855635068178177,
-        0.2084233057498932,
-        0.07603480890989306,
-        0.1241462073802948,
-        0.7568318559646605,
-        0.9998,
-        0.18748426148295405,
-        0.29386607773303985,
-        0.21488302874565124,
-        0.0777944794893265,
-        0.1279024946689606,
-        0.7800548309087754,
-        0.9998,
-        0.19402539111375805,
-        0.304579441523552,
-        0.22076346032619476,
-        0.07946800847053526,
-        0.13082660132646562,
-        0.8035489074707032,
-        0.9998,
-        0.1999460756778717,
-        0.31222576675415037,
-        0.2264517953634262,
-        0.08081280782818795,
-        0.13460262061357497,
-        0.828994060134888,
-        0.9998,
-        0.20509765752553943,
-        0.3203624894678593,
-        0.23198571143150332,
-        0.08211237077713013,
-        0.13750549141168594,
-        0.8529169931411746,
-        0.9998,
-        0.209518383550644,
-        0.3281992249488831,
-        0.23816398755311968,
-        0.08322777889966959,
-        0.14037072820067403,
-        0.8746512705802916,
-        0.9998,
-        0.21442690343856818,
-        0.33666496825218195,
-        0.24309153485894203,
-        0.08529254732131952,
-        0.14304055777192115,
-        0.8924013189554216,
-        0.9998,
-        0.21941403126716608,
-        0.3448349338173866,
-        0.2489651171147823,
-        0.08667572829127312,
-        0.14580873468518263,
-        0.9107246013164523,
-        0.9998,
-        0.22309734332561493,
-        0.3517628137350083,
-        0.2547237434387207,
-        0.08896696496009826,
-        0.1489701636433602,
-        0.9272562242746352,
-        0.9998,
-        0.22793451219797134,
-        0.3591047751426697,
-        0.2597683270871639,
-        0.09042025351524352,
-        0.1528471918106079,
-        0.9483436929702762,
-        0.9998,
-        0.23258997063636777,
-        0.36902411174774175,
-        0.26485966441631315,
-        0.09243017539381981,
-        0.15506044900417326,
-        0.9667170533418656,
-        0.9998,
-        0.23665347018241878,
-        0.3762736029624939,
-        0.27067578415870663,
-        0.09449174585938458,
-        0.15888449568748475,
-        0.9834513599395751,
-        0.9998,
-        0.24093078261613854,
-        0.3843366762638093,
-        0.2768812180876732,
-        0.09594143676757816,
-        0.16144601993560792,
-        1.0022019856452942,
-        0.9998,
-        0.2459272434949874,
-        0.3907656828403474,
-        0.2815030888676643,
-        0.09760697330236434,
-        0.16464969681501385,
-        1.019599228835106,
-        0.9998,
-        0.2511993800699711,
-        0.3962667070865631,
-        0.28746061475872997,
-        0.09892506269812584,
-        0.1685606077313423,
-        1.0345355377674106,
-        0.9998,
-        0.25586065447330475,
-        0.4064154948949814,
-        0.29352941393256193,
-        0.10010581925511364,
-        0.17137197246551505,
-        1.0455808718204502,
-        0.9998,
-        0.26066332874298104,
-        0.41295039305686965,
-        0.29917095521688464,
-        0.10042237902879714,
-        0.17334867343306548,
-        1.0644763714551928,
-        0.9998,
-        0.26424761767387395,
-        0.4171489597439766,
-        0.3047941813468933,
-        0.10030385837554934,
-        0.1771283437609672,
-        1.0775494430065153,
-        0.9998,
-        0.266823394203186,
-        0.42120621981620787,
-        0.31085526600480073,
-        0.10133219141960148,
-        0.18098006631135943,
-        1.0927770781517028,
-        0.9998,
-        0.27192984950542454,
-        0.4258830359578132,
-        0.31648782253265384,
-        0.10210902032852165,
-        0.18391375703215596,
-        1.1100240562438963,
-        0.9998,
-        0.2774573613643646,
-        0.4328473631143568,
-        0.3229170823454858,
-        0.10328457579612738,
-        0.18559744659066202,
-        1.127002547836304,
-        0.9998,
-        0.280809062719345,
-        0.43832696628570567,
-        0.32813225684762004,
-        0.10458858071565635,
-        0.18929516908526423,
-        1.1399588096141815,
-        0.9998,
-        0.28525815182924275,
-        0.44289318814277645,
-        0.3333625353813172,
-        0.1055313608527183,
-        0.1918839191913605,
-        1.1545847342491151,
-        0.9998,
-        0.2899247071444988,
-        0.4499819297909735,
-        0.33962785017490393,
-        0.10614367849826811,
-        0.19452174307107928,
-        1.1747124679088592,
-        0.9998,
-        0.29274274289608004,
-        0.4549682779312134,
-        0.3463232246458531,
-        0.10680567473173141,
-        0.19644608254432672,
-        1.185605804681778,
-        0.9998
-      ]
-    }
-  },
-  "sample_ratio": 0.05,
-  "dataset_infos": {
-    "robotgeneralist/pp__1": {
-      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
-      "num_episodes": 237,
-      "num_frames": 99363
-    },
-    "robotgeneralist/pp__1__adversarial": {
-      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
-      "num_episodes": 64,
-      "num_frames": 17906
-    },
-    "robotgeneralist/pp__2": {
-      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
-      "num_episodes": 92,
-      "num_frames": 37458
-    },
-    "robotgeneralist/pp__2__rotations": {
-      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
-      "num_episodes": 102,
-      "num_frames": 13935
-    },
-    "robotgeneralist/pp__3__with_ft": {
-      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
-      "num_episodes": 99,
-      "num_frames": 38424
-    },
-    "robotgeneralist/pp__3__with_ft__rotations": {
-      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
-      "num_episodes": 49,
-      "num_frames": 7151
-    },
-    "robotgeneralist/pp__3__with_ft__adversarial_start": {
-      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
-      "num_episodes": 147,
-      "num_frames": 21313
-    },
-    "robotgeneralist/pp__4__with_ft": {
-      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
-      "num_episodes": 94,
-      "num_frames": 33046
-    },
-    "robotgeneralist/pp__5__with_ft__adversarial_target": {
-      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
-      "num_episodes": 48,
-      "num_frames": 10615
-    },
-    "robotgeneralist/pp__6__with_ft": {
-      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
-      "num_episodes": 379,
-      "num_frames": 123611
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-1": {
-      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
-      "num_episodes": 98,
-      "num_frames": 41410
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-2": {
-      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
-      "num_episodes": 94,
-      "num_frames": 37859
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-3": {
-      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
-      "num_episodes": 127,
-      "num_frames": 53341
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-4": {
-      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
-      "num_episodes": 102,
-      "num_frames": 38263
-    }
-  }
-}
\ No newline at end of file
diff --git a/pp__ee_baseline/10000/params/_METADATA b/pp__ee_baseline/10000/params/_METADATA
deleted file mode 100644
index 9eb73624158f4514d49041e0806d2d702c62eee1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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, "write_shape": [1152]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304]}}, "('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, "write_shape": [27, 1152, 4304]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 16, 72, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [14, 14, 3, 1152]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1152, 2048]}}, "('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, "write_shape": [1, 256, 1152]}}, "('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, "write_shape": [257152, 2048]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [18, 8, 256, 2048]}}, "('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, "write_shape": [18, 8, 256, 1024]}}, "('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, "write_shape": [18, 2, 1, 2048, 256]}}, "('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, "write_shape": [18, 2, 1, 1024, 256]}}, "('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, "write_shape": [18, 8, 2048, 256]}}, "('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, "write_shape": [18, 8, 1024, 256]}}, "('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, "write_shape": [18, 2, 2048, 16384]}}, "('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, "write_shape": [18, 16384, 2048]}}, "('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, "write_shape": [18, 2, 1024, 4096]}}, "('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, "write_shape": [18, 4096, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}, "('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, "write_shape": [32]}}, "('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, "write_shape": [1024, 32]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [2048, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [1024, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/pp__ee_baseline/10000/params/_sharding b/pp__ee_baseline/10000/params/_sharding
deleted file mode 100644
index 924608abdcb99abb3c864bb9d1bfca6eb60f780e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/_sharding
+++ /dev/null
@@ -1 +0,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}]]}}"}
\ No newline at end of file
diff --git a/pp__ee_baseline/10000/params/array_metadatas/process_0 b/pp__ee_baseline/10000/params/array_metadatas/process_0
deleted file mode 100644
index 6aa9fd1decf72607133eb54664d98802bf563b5c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/array_metadatas/process_0
+++ /dev/null
@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/pp__ee_baseline/10000/params/d/64caffe671e3861abcc2f6f63de7e3bf b/pp__ee_baseline/10000/params/d/64caffe671e3861abcc2f6f63de7e3bf
deleted file mode 100644
index 00e845b1f48161c036bd3b99c2ea7d888958f986..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/d/64caffe671e3861abcc2f6f63de7e3bf and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/d/6fa24ff7c1524142ef3a98293288c86d b/pp__ee_baseline/10000/params/d/6fa24ff7c1524142ef3a98293288c86d
deleted file mode 100644
index 836fdab085f758eca1a06d057703413270ae27d5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/d/6fa24ff7c1524142ef3a98293288c86d and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/manifest.ocdbt b/pp__ee_baseline/10000/params/manifest.ocdbt
deleted file mode 100644
index c876ca2dfec04d467a844406d16b34eb07b28fa4..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/019a25f7c816d93d1532cffe87ca4fba b/pp__ee_baseline/10000/params/ocdbt.process_0/d/019a25f7c816d93d1532cffe87ca4fba
deleted file mode 100644
index 4880c3ed30c54f93e4bca07f8c725e6402bb681d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/019a25f7c816d93d1532cffe87ca4fba
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:14314f9310db685f4d3fc618d4a2c6b93ca3f4a695088f85ac759032c76e7f89
-size 496756117
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/0481fb299f75ed688ffd20ac1469caf2 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/0481fb299f75ed688ffd20ac1469caf2
deleted file mode 100644
index 9bc5bc7f9eec2745e2ebf26da8acc3cadd7b4694..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/0481fb299f75ed688ffd20ac1469caf2
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a7abdec44c731ed70672074a0d0625d6734a8a615d411c2ac1358d097a8eb106
-size 126325
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/087551093b64c3dfcfafa3e4ed4da6e4 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/087551093b64c3dfcfafa3e4ed4da6e4
deleted file mode 100644
index e20c4043f1c101220d608558b803eac98724dc72..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/087551093b64c3dfcfafa3e4ed4da6e4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f223b4b43eada56595cd73dfee8ad74bf3110871ca6b5aa61d742a5b0381b8c8
-size 132966908
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/09a9331e13384e739fcbeb1a48f8f5fa b/pp__ee_baseline/10000/params/ocdbt.process_0/d/09a9331e13384e739fcbeb1a48f8f5fa
deleted file mode 100644
index 45ab8f99cf0eb7d6881ab1d4064a4bb6c9b32323..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/09a9331e13384e739fcbeb1a48f8f5fa
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b8115cdc272d324b37d65d459a1f40821bfc93798b2d39739a6965a82a2bf67a
-size 265933406
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/0e152be72d456410d931725925c4e05a b/pp__ee_baseline/10000/params/ocdbt.process_0/d/0e152be72d456410d931725925c4e05a
deleted file mode 100644
index 1703352692394f0786d0223898d4d27da26e4069..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/0e152be72d456410d931725925c4e05a and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/12b81d51f17757c52b8603ce12bae39c b/pp__ee_baseline/10000/params/ocdbt.process_0/d/12b81d51f17757c52b8603ce12bae39c
deleted file mode 100644
index 991325b83998a53a9d44053f47b02d42afac0531..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/12b81d51f17757c52b8603ce12bae39c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7562bd122479ee341c6057bf3a824df1b04953e870a58a3e3a507b67c4cca033
-size 2240220466
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/19aa52193d5a2eab86bd9aeccc15d7f6 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/19aa52193d5a2eab86bd9aeccc15d7f6
deleted file mode 100644
index 3836d6811b94fdded5af1f52e300435b2e71d1b8..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/19aa52193d5a2eab86bd9aeccc15d7f6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5213104da9a2949ae259e291f88ba6719d898bba1531e4064d5dd5072f6c67a6
-size 132846192
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1b5628f075cbe1dc1493c4b980515dbd b/pp__ee_baseline/10000/params/ocdbt.process_0/d/1b5628f075cbe1dc1493c4b980515dbd
deleted file mode 100644
index 705f57fce9330ef416e3e0f1582f1db387998ed6..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1b5628f075cbe1dc1493c4b980515dbd
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:55cded5c66245c84035aa24c66ad4ef494af51c1ccb01ae88d3ecb773ff3b0bb
-size 420161312
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1cd349cbead5749a4627d7691d161319 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/1cd349cbead5749a4627d7691d161319
deleted file mode 100644
index f334754de4254bde76f828af471b6338496000f2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1cd349cbead5749a4627d7691d161319
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:855eafa051ea84cce1210b0ac7465c11050edff00cc0aace38245f05d1fb50d5
-size 497034683
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1f977a19f1905b5548dcde25ed450d6b b/pp__ee_baseline/10000/params/ocdbt.process_0/d/1f977a19f1905b5548dcde25ed450d6b
deleted file mode 100644
index 35ba4e010fc84ea35e42f2996eef4fbd65937ba5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/1f977a19f1905b5548dcde25ed450d6b and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/24698c6b3ec1ba38b1228e81eb1db105 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/24698c6b3ec1ba38b1228e81eb1db105
deleted file mode 100644
index dbacacc164b39a8c70cc66e59db69591aae15951..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/24698c6b3ec1ba38b1228e81eb1db105
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c506e39fb6f91cb39d2060d2b42060a2449b2507b4d35d026169bcd2442d38c2
-size 21823301
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/331d51f3288166627f64764d6803b0bd b/pp__ee_baseline/10000/params/ocdbt.process_0/d/331d51f3288166627f64764d6803b0bd
deleted file mode 100644
index 25964399cb02464164faed2b4246c47f97b9ffcc..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/331d51f3288166627f64764d6803b0bd
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4004af616b0b79a98e5e1821c789f88876d74a98d13dc3b7a76d733f94a9d052
-size 2240246012
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/39c6cfbe2087f2d95106a08dd4156e25 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/39c6cfbe2087f2d95106a08dd4156e25
deleted file mode 100644
index efcc2a5917ba789a8e0f2ddc920e4f13186c93fe..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/39c6cfbe2087f2d95106a08dd4156e25
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:80211421a54a866851a2a79e211f6a55d563878d37278710ec3d73f8a65f46fe
-size 210053134
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/3fdfa849a83f6bf866207b002311048d b/pp__ee_baseline/10000/params/ocdbt.process_0/d/3fdfa849a83f6bf866207b002311048d
deleted file mode 100644
index bc52959dee7a34b30544f2daa6513d04f69accb2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/3fdfa849a83f6bf866207b002311048d
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:edb18f05db613d4224aed78e7f22718d46e0a9f423779179a5364a36392d2163
-size 258879
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/40f16d0adbb2baab0da6e8bb0d87e2f0 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/40f16d0adbb2baab0da6e8bb0d87e2f0
deleted file mode 100644
index 00256fdc2bd4d167ddd6322d2a5b88628b57dc42..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/40f16d0adbb2baab0da6e8bb0d87e2f0
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7f4eff48c68dbbdf59b89d1eb5cae2b7662488c30235aed88f555fd69dce2afb
-size 1120134600
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/45dd7b8a685a334d5f60d1d8d00295a1 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/45dd7b8a685a334d5f60d1d8d00295a1
deleted file mode 100644
index 258d4a26552becd517257ec151a8192f6cf6a9e6..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/45dd7b8a685a334d5f60d1d8d00295a1
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:24d64d14967d06d580b66033cd948d4ac0e4c8b47524eeea3dc915b0ffaf5a74
-size 139914364
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/4f0b92e628e23d7c371c772ef5ca1246 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/4f0b92e628e23d7c371c772ef5ca1246
deleted file mode 100644
index 00d30b658a59a62696d59c106d3ae8eb776307a2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/4f0b92e628e23d7c371c772ef5ca1246
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8c4e376d59e1fea14f0a8bdf41b1237834f0203da4c055758079a76ae9d94db1
-size 559275070
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/51a121cf733d9bf4384fe5150c563646 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/51a121cf733d9bf4384fe5150c563646
deleted file mode 100644
index ca9984d871524d314a07c194929239af7eaaee57..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/51a121cf733d9bf4384fe5150c563646
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:74818d180db5debb2ffbc51dabd371971f12f36efae0b1a35d91842f1547fd70
-size 559726481
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/5f290fa84d42df457b9d8258fd5e4484 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/5f290fa84d42df457b9d8258fd5e4484
deleted file mode 100644
index 776ab3fd190b00ceb0527035258de8e8327060fa..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/5f290fa84d42df457b9d8258fd5e4484
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ccc9fa94b7a28a79042578896191c821ffdb15414836089547aefdc018a1b19c
-size 1120110723
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/64f354c787df6b5c565108c5db5e5eb0 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/64f354c787df6b5c565108c5db5e5eb0
deleted file mode 100644
index 6ab467212281002cb1ebcf3d6bf95e0ab80f0c4f..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/64f354c787df6b5c565108c5db5e5eb0
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1273eb30699f3f825ab9fb7d705361225bba8f91a269725b8583f6f1687c573b
-size 34975093
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6684d4aa0a30c667c448a2ce8539eb32 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/6684d4aa0a30c667c448a2ce8539eb32
deleted file mode 100644
index 4f03c1c1f9fea1a0d9c2efb5133ffad752b66284..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6684d4aa0a30c667c448a2ce8539eb32
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8da663270a5117fc74b0e5f68652dea464b9c828d625f51f8363d89896b3f277
-size 1957442747
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6a1440548cd7212326726317df5be8db b/pp__ee_baseline/10000/params/ocdbt.process_0/d/6a1440548cd7212326726317df5be8db
deleted file mode 100644
index ffc57c5cefe3160b4f5ec2c8030ab168d9324180..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6a1440548cd7212326726317df5be8db and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6b1caa67c5e2e1dac4a31947b3df3f2e b/pp__ee_baseline/10000/params/ocdbt.process_0/d/6b1caa67c5e2e1dac4a31947b3df3f2e
deleted file mode 100644
index 053fe96aa7dfe4caa19f45901d0f45a36cb78c79..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/6b1caa67c5e2e1dac4a31947b3df3f2e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:439e1269164348713e99b9b4fd1aa77b3f51f66551b56066e18dafc7324a7ec4
-size 1120132587
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/7006c014b0daa53c04f1493aa21c17a5 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/7006c014b0daa53c04f1493aa21c17a5
deleted file mode 100644
index 82ac134b9a0ca09d3a5d223e97db2672307d1d2d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/7006c014b0daa53c04f1493aa21c17a5
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:632a29a5a6d13bc3f1f8e9b547d6820c3e3c915e346bb7ba5b1124500a60cab2
-size 2240219627
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/707dcc33103413a3305af91394b0ba44 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/707dcc33103413a3305af91394b0ba44
deleted file mode 100644
index 8b518638f99607b51e5949d4427b2f8e52d68453..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/707dcc33103413a3305af91394b0ba44
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f947983d364378798c9db81d51bfd0f79b8e8473b04f9f73d83a319890882dc2
-size 1120113348
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/71edca3654852fa173676d75b09e18d4 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/71edca3654852fa173676d75b09e18d4
deleted file mode 100644
index 2b33647267431c8e1ce1cd55d1d28380fcd5d650..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/71edca3654852fa173676d75b09e18d4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a6dd324f4db35fe3c054dadba8826356ed25e0e2e250a0fc069332693cd6fe7b
-size 497031537
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/77521ce735d761eea532dc7d664286c6 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/77521ce735d761eea532dc7d664286c6
deleted file mode 100644
index ca4cdc30f92ecec548e871054c1d94aa90fb8d96..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/77521ce735d761eea532dc7d664286c6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b5f1973c8c43cfde6ec89315359186cc275435c5fe5c8493f54eeaa0f1560377
-size 1957442621
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/791ae89fe6c033f359ada115935cfc35 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/791ae89fe6c033f359ada115935cfc35
deleted file mode 100644
index ffc57c5cefe3160b4f5ec2c8030ab168d9324180..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/791ae89fe6c033f359ada115935cfc35 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/7c97f4dc8dcabd69aba7f65bfe810ed2 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/7c97f4dc8dcabd69aba7f65bfe810ed2
deleted file mode 100644
index efdabd3ef06ac4016c51b237efd608eaf0af5532..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/7c97f4dc8dcabd69aba7f65bfe810ed2
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2b414884d3ebc3b4642045cdceb437c17ae30439fd61b6523b779fa3a7cd0920
-size 132936139
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/80bd6202a3fce6885076f5e442529f58 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/80bd6202a3fce6885076f5e442529f58
deleted file mode 100644
index a9fabd482b5745cdaf27edefc87b10d6192c9899..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/80bd6202a3fce6885076f5e442529f58
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:997031a6b24f3a57c97802d9799e7211f3788c530c1e6828332291c810fe8cce
-size 34975312
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/84f6a79ed1ad4a2765fb9c4c2d0523be b/pp__ee_baseline/10000/params/ocdbt.process_0/d/84f6a79ed1ad4a2765fb9c4c2d0523be
deleted file mode 100644
index 5a93e45becfe334b87a96377afe0df2833cc8f39..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/84f6a79ed1ad4a2765fb9c4c2d0523be
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e140372b71d30d3aa306cd20eb76892ddfa763cb465de00449f49860ce30f8a2
-size 559731600
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/89ace7fe8fc3849793247be9d2798c7d b/pp__ee_baseline/10000/params/ocdbt.process_0/d/89ace7fe8fc3849793247be9d2798c7d
deleted file mode 100644
index 0659551e77865f6f8603cecf7706067274c6a5bf..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/89ace7fe8fc3849793247be9d2798c7d
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:64e24e7d7e5aca72c302bddf8cc01d0b6e54e6896d394e9a7a3bda5fe8bb18b5
-size 139914428
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8a39179c6eec3744a106bd0db9e64554 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/8a39179c6eec3744a106bd0db9e64554
deleted file mode 100644
index 0d22451f0fcde6e73380d698607a16fe17612c5a..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8a39179c6eec3744a106bd0db9e64554 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8abe8a9d9cbaa63161d184bd45d3684d b/pp__ee_baseline/10000/params/ocdbt.process_0/d/8abe8a9d9cbaa63161d184bd45d3684d
deleted file mode 100644
index 44c207101bde9c40edbdab1b17226be732226686..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8abe8a9d9cbaa63161d184bd45d3684d
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:03f8d519f1cc5a2f7f2ed4537926a245f4a4ab1503f53b9d8cf61fa20fb52c0b
-size 132937141
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8ac47f752ed1e41142b2cc41bdd3824e b/pp__ee_baseline/10000/params/ocdbt.process_0/d/8ac47f752ed1e41142b2cc41bdd3824e
deleted file mode 100644
index 8acf42ee2c1d90c6c00d4e3d4063860182ed154b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/8ac47f752ed1e41142b2cc41bdd3824e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6dfed161aacec3ee366d239317328e881c7ff590b1bbc84bf2920d67d2bbc444
-size 279242282
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/93f7ce1f72c301b5a7e6efdf51768941 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/93f7ce1f72c301b5a7e6efdf51768941
deleted file mode 100644
index 5d074fba94a076974636822808d9b5db175794ba..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/93f7ce1f72c301b5a7e6efdf51768941
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b4259584767c31bd8d243e005b79d246e20d763b07b0816b875aa8f29d551840
-size 132967967
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/a032e292a41fa056512a40d6b7065f5f b/pp__ee_baseline/10000/params/ocdbt.process_0/d/a032e292a41fa056512a40d6b7065f5f
deleted file mode 100644
index 76f75ab08568cdc40ab085503d19333f51279598..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/a032e292a41fa056512a40d6b7065f5f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:428707c0ca3a010ca25b608b0a7d2f10ef45d681ed69bebaf0cbbe3a294cdfa9
-size 8878986
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/a97f0ad5c0bf179d7bd6e36a6efbc5aa b/pp__ee_baseline/10000/params/ocdbt.process_0/d/a97f0ad5c0bf179d7bd6e36a6efbc5aa
deleted file mode 100644
index 1fad40a5d98a0e6ee1c35861e158d9e747984bdf..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/a97f0ad5c0bf179d7bd6e36a6efbc5aa and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/ab75d254589876883e492d81b296d044 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/ab75d254589876883e492d81b296d044
deleted file mode 100644
index 9439d8f2aefda0b72e6fe5bcd95f14e68a7449bd..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/ab75d254589876883e492d81b296d044
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7785cfc27e6206a77a16d5184cf5098a1cb86a731aaf28dad0269b176e6f3f61
-size 16809448
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/abb20658929263a54d4472dbc6d5e946 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/abb20658929263a54d4472dbc6d5e946
deleted file mode 100644
index 965d02715274162e81cae48ce5dc971c2273b8b5..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/abb20658929263a54d4472dbc6d5e946
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3eca8f6259e99df25d6d9ed9693d7095a076cea11d8875901946d11085c3506c
-size 132846482
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/b2c5f61ac68396d6033f2765196e7307 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/b2c5f61ac68396d6033f2765196e7307
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/b2c5f61ac68396d6033f2765196e7307 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/c097b4f81f2052b9c41827a0d2615524 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/c097b4f81f2052b9c41827a0d2615524
deleted file mode 100644
index 68057754bd32f99b007c31c734b7cb88b6279675..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/c097b4f81f2052b9c41827a0d2615524
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3e2e09db0f7af143d14e013c0ce7193e90f2b134d7f07e5780c3761662f514ce
-size 4353995
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/c9039cea8c9b7dabbd48e65579acc033 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/c9039cea8c9b7dabbd48e65579acc033
deleted file mode 100644
index a64dd69aa017269d5a4e060e9be20f7925ebf9a1..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/c9039cea8c9b7dabbd48e65579acc033 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/cf484530167125ac412d47e695aa4700 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/cf484530167125ac412d47e695aa4700
deleted file mode 100644
index 2ee73c2e2d17733080446dde0781adca96f7bb24..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/cf484530167125ac412d47e695aa4700
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d107adff593ffd8f23cbcbe3bfd1cae962a55ac754e881e58ca55db6b9448292
-size 234234
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d0168e56d7a1134467280d1407b44412 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/d0168e56d7a1134467280d1407b44412
deleted file mode 100644
index fafb4b2ef3b739d20010b130ede890ed0a81b03e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d0168e56d7a1134467280d1407b44412
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f98c6744ea4fdfd4dae174f9f6bc3fe1a51ad8f542df6f8f4eb5ce9f185b2a26
-size 280037472
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d071bb56d024a330a6e679eb3e047a92 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/d071bb56d024a330a6e679eb3e047a92
deleted file mode 100644
index 0c09986f2002e024f9df454075329ec6c0de5417..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d071bb56d024a330a6e679eb3e047a92
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b0b7110ea73b4ce256629ea4a22c69f5b76f15db606842db47449fe1615e8c2a
-size 280190006
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d1bb43e79bed2686d547a2cc13a639b1 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/d1bb43e79bed2686d547a2cc13a639b1
deleted file mode 100644
index 069c99698b09c27bc16fbd47ecb5d0fd3b38c8df..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/d1bb43e79bed2686d547a2cc13a639b1
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:146806716712eddf830bd86061cf5f3ec1fe8ff20fe8895a220a9b1d9e61cfe8
-size 496757025
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/dd001062f1d37dd5faae4ba77e852b2e b/pp__ee_baseline/10000/params/ocdbt.process_0/d/dd001062f1d37dd5faae4ba77e852b2e
deleted file mode 100644
index 67ad76ea5d90477a1d6f2112d098ab0d124e0c5a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/dd001062f1d37dd5faae4ba77e852b2e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3eff4c3cb1ae1636624cefd9d88e8c110093db0a92d9d4e4165e794d3a260f05
-size 1120116677
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/e692f7d578b7b439f32968477d9724b8 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/e692f7d578b7b439f32968477d9724b8
deleted file mode 100644
index ba5484d9604650d4a7155c2a66a4bab789e1b4e7..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/e692f7d578b7b439f32968477d9724b8
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c79d8fb3a09ef491d966fc0411e597475dc98688096288550ad8a597f95e8331
-size 122376
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/e73375e88a1069dbac0221d60c340dbb b/pp__ee_baseline/10000/params/ocdbt.process_0/d/e73375e88a1069dbac0221d60c340dbb
deleted file mode 100644
index 44f10fa287eef012efaf96dbf35c1d7179cb01e4..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/e73375e88a1069dbac0221d60c340dbb
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bd501dffd15595a4929651826bbcae722f24acb476a3d5229b0ff8b2c393bab8
-size 1120129648
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/eba48fdaa226d59f5333d8e453705cc8 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/eba48fdaa226d59f5333d8e453705cc8
deleted file mode 100644
index 765412ed54dd9be45632a2fd31a92f617af4c730..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/eba48fdaa226d59f5333d8e453705cc8 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/ec410ec7580fb821cadd057409c068f3 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/ec410ec7580fb821cadd057409c068f3
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/d/ec410ec7580fb821cadd057409c068f3 and /dev/null differ
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/d/f45d504ab6cbe0b74ff663e94839b5a3 b/pp__ee_baseline/10000/params/ocdbt.process_0/d/f45d504ab6cbe0b74ff663e94839b5a3
deleted file mode 100644
index 2af2c5e62ebd9dc252a41ca0a11bea7a31566424..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/10000/params/ocdbt.process_0/d/f45d504ab6cbe0b74ff663e94839b5a3
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ff33cdf5ee71db0c0e514df29db884ed59aaa839577834dce7765f2ed8202499
-size 70083558
diff --git a/pp__ee_baseline/10000/params/ocdbt.process_0/manifest.ocdbt b/pp__ee_baseline/10000/params/ocdbt.process_0/manifest.ocdbt
deleted file mode 100644
index d9788f09c28bef03c34a207b056bb4f5466a42b7..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/10000/params/ocdbt.process_0/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/15000/_CHECKPOINT_METADATA b/pp__ee_baseline/15000/_CHECKPOINT_METADATA
deleted file mode 100644
index 519fceb3cc504373a7987cc70f33e5791a6328c5..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/_CHECKPOINT_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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": 1768757363979461930, "commit_timestamp_nsecs": 1768757456251770373, "custom_metadata": {}}
\ No newline at end of file
diff --git a/pp__ee_baseline/15000/assets/ur10e/norm_stats.json b/pp__ee_baseline/15000/assets/ur10e/norm_stats.json
deleted file mode 100644
index 8a1b886d6b668da778297b91b3d82381c2eab28a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/assets/ur10e/norm_stats.json
+++ /dev/null
@@ -1,1525 +0,0 @@
-{
-  "norm_stats": {
-    "state": {
-      "mean": [
-        0.05313729867339134,
-        0.0037223484832793474,
-        0.106409952044487,
-        0.0008666226058267057,
-        -0.00854077935218811,
-        0.01460699550807476,
-        0.5940247178077698
-      ],
-      "std": [
-        0.1176244392991066,
-        0.16962096095085144,
-        0.07773701101541519,
-        0.04537155479192734,
-        0.06616315245628357,
-        0.6038344502449036,
-        0.49107977747917175
-      ],
-      "q01": [
-        -0.2025573879778385,
-        -0.31106759839057924,
-        0.00890737680345774,
-        -0.11727181684970855,
-        -0.16711108471155167,
-        -1.423702568435669,
-        0.0
-      ],
-      "q99": [
-        0.29901714037060734,
-        0.4115018085956573,
-        0.3217185490995646,
-        0.13774808425903318,
-        0.19241279754638663,
-        1.598234023094177,
-        0.9998
-      ]
-    },
-    "actions": {
-      "mean": [
-        0.0001451610296498984,
-        0.00017207347264047712,
-        0.000048651279939804226,
-        0.00003654604734038003,
-        -0.000019767590856645256,
-        -0.00019861181499436498,
-        0.5944076776504517,
-        0.0002841090608853847,
-        0.00034722191048786044,
-        0.00011135452223243192,
-        0.00006827477045590058,
-        -0.000043589698179857805,
-        -0.0004028017574455589,
-        0.594025731086731,
-        0.00041486899135634303,
-        0.0005213047261349857,
-        0.00018569435633253306,
-        0.00009245114779332653,
-        -0.00007049143459880725,
-        -0.000616009347140789,
-        0.5935360789299011,
-        0.000542119552846998,
-        0.0006975074065849185,
-        0.00027102616149932146,
-        0.00011072693450842053,
-        -0.00010138262587133795,
-        -0.0008296052110381424,
-        0.593919575214386,
-        0.0006618681945838034,
-        0.0008754940354265273,
-        0.0003684645926114172,
-        0.00012383276771288365,
-        -0.000135598995257169,
-        -0.0010419095633551478,
-        0.5929783582687378,
-        0.0007787057547830045,
-        0.00105847988743335,
-        0.0004721089790109545,
-        0.00013335929543245584,
-        -0.00017038211808539927,
-        -0.0012720258673653007,
-        0.5933270454406738,
-        0.0008861633250489831,
-        0.001237610587850213,
-        0.0005911999032832682,
-        0.00013674379442818463,
-        -0.00020881379896309227,
-        -0.0014943619025871158,
-        0.5934669971466064,
-        0.0009874902898445725,
-        0.0014197468990460038,
-        0.0007170690805651248,
-        0.00013581194798462093,
-        -0.0002465852303430438,
-        -0.0017284756759181619,
-        0.5931888818740845,
-        0.0010826619109138846,
-        0.0016015341971069574,
-        0.0008584632305428386,
-        0.000128391373436898,
-        -0.00028913054848089814,
-        -0.0019626382272690535,
-        0.5930843949317932,
-        0.001174500328488648,
-        0.0017860672669485211,
-        0.0010054412996396422,
-        0.00012193901784485206,
-        -0.000334887852659449,
-        -0.0022029897663742304,
-        0.5935000777244568,
-        0.0012568456586450338,
-        0.0019742222502827644,
-        0.0011647609062492847,
-        0.00011021356476703659,
-        -0.00038211053470149636,
-        -0.0024467180483043194,
-        0.5931875109672546,
-        0.0013310370268300176,
-        0.0021642642095685005,
-        0.0013335321564227343,
-        0.00009949743980541825,
-        -0.0004296436673030257,
-        -0.0027013549115508795,
-        0.5931177139282227,
-        0.0014006041456013918,
-        0.002352928975597024,
-        0.0015152853447943926,
-        0.00008505478035658598,
-        -0.00048136088298633695,
-        -0.0029538467060774565,
-        0.5930479764938354,
-        0.0014608069323003292,
-        0.0025415997952222824,
-        0.001700855791568756,
-        0.00006727704749209806,
-        -0.0005341243813745677,
-        -0.003210614435374737,
-        0.5928036570549011,
-        0.0015100123127922416,
-        0.002731549320742488,
-        0.0018991812830790877,
-        0.0000493652478326112,
-        -0.0005809388239867985,
-        -0.003473253920674324,
-        0.5928736925125122,
-        0.0015514988917857409,
-        0.002922238316386938,
-        0.0021079820580780506,
-        0.000029167196771595627,
-        -0.0006329127354547381,
-        -0.0037316547241061926,
-        0.5928043127059937,
-        0.0015838317340239882,
-        0.0031066250521689653,
-        0.002319576218724251,
-        8.898059604689479e-6,
-        -0.0006800378323532641,
-        -0.004003639332950115,
-        0.5929091572761536,
-        0.0016050132689997554,
-        0.003295252099633217,
-        0.0025419394951313734,
-        -0.000013710295206692535,
-        -0.0007257868419401348,
-        -0.004269901663064957,
-        0.5930485725402832,
-        0.0016164487460628152,
-        0.0034825268667191267,
-        0.002774182241410017,
-        -0.000037491106922971085,
-        -0.0007724225870333612,
-        -0.004538912326097488,
-        0.5931875705718994,
-        0.001615947112441063,
-        0.003664180636405945,
-        0.0030124683398753405,
-        -0.00006199560448294505,
-        -0.0008142396691255271,
-        -0.00480045098811388,
-        0.5933974385261536,
-        0.0016095866449177265,
-        0.0038478909991681576,
-        0.003257976844906807,
-        -0.00008698971942067146,
-        -0.0008547022007405758,
-        -0.005078013986349106,
-        0.5936415791511536,
-        0.0015872354852035642,
-        0.004021089058369398,
-        0.0035109000746160746,
-        -0.00011568223999347538,
-        -0.0008942089625634253,
-        -0.005354971159249544,
-        0.5934317708015442,
-        0.0015591973206028342,
-        0.004198537208139896,
-        0.003773036180064082,
-        -0.00014495827781502157,
-        -0.0009303062106482685,
-        -0.005638875067234039,
-        0.5936412215232849,
-        0.0015177299501374364,
-        0.004366280045360327,
-        0.004040749743580818,
-        -0.00017588783521205187,
-        -0.0009663543896749616,
-        -0.005925025325268507,
-        0.593501627445221,
-        0.0014705799985677004,
-        0.004531881771981716,
-        0.00431715976446867,
-        -0.00021041395666543394,
-        -0.0010010612895712256,
-        -0.006207368802279234,
-        0.5935365557670593,
-        0.001413213089108467,
-        0.004692579619586468,
-        0.004599601496011019,
-        -0.0002457766968291253,
-        -0.0010327493073418736,
-        -0.006505406461656094,
-        0.5933269262313843,
-        0.0013434475986286998,
-        0.004848736338317394,
-        0.004890408832579851,
-        -0.00028711999766528606,
-        -0.00106858741492033,
-        -0.006807356141507626,
-        0.5931181311607361,
-        0.0012650334974750876,
-        0.004994421266019344,
-        0.0051898653618991375,
-        -0.00033172621624544263,
-        -0.0010990891605615616,
-        -0.0071100820787250996,
-        0.5929442048072815,
-        0.0011793962912634015,
-        0.005135250277817249,
-        0.005499166436493397,
-        -0.000378662021830678,
-        -0.0011317729949951172,
-        -0.007409243378788233,
-        0.5926647782325745,
-        0.0010838431771844625,
-        0.005268934182822704,
-        0.005817332770675421,
-        -0.0004281000583432615,
-        -0.001166283618658781,
-        -0.007717216853052378,
-        0.5926998853683472,
-        0.0009804241126403213,
-        0.0053992304019629955,
-        0.0061420174315571785,
-        -0.00048094423254951835,
-        -0.0011993126245215535,
-        -0.008030007593333721,
-        0.5924902558326721,
-        0.0008692154078744352,
-        0.005519451107829809,
-        0.006479023955762386,
-        -0.000537763989996165,
-        -0.001236212207004428,
-        -0.008356506004929543,
-        0.5929440855979919,
-        0.0007513284799642861,
-        0.00563538633286953,
-        0.006823732051998377,
-        -0.000594516284763813,
-        -0.0012742418330162764,
-        -0.008685515262186527,
-        0.5926643013954163,
-        0.0006232408923096955,
-        0.005743546411395073,
-        0.007178108207881451,
-        -0.0006553126149810851,
-        -0.0013130278093740344,
-        -0.00901535339653492,
-        0.5930123329162598,
-        0.00048613501712679863,
-        0.005843072663992643,
-        0.007545872125774622,
-        -0.0007216006051748991,
-        -0.001357274129986763,
-        -0.009348330087959766,
-        0.5926995277404785,
-        0.000340388563927263,
-        0.005928103346377611,
-        0.007921233773231506,
-        -0.0007906133541837335,
-        -0.0013983722310513258,
-        -0.00968977902084589,
-        0.5930134654045105,
-        0.00018408334290143102,
-        0.006012112833559513,
-        0.008309606462717056,
-        -0.0008611961966380477,
-        -0.00144277885556221,
-        -0.010034735314548016,
-        0.5933630466461182,
-        0.000022206833818927407,
-        0.0060895816422998905,
-        0.008708948269486427,
-        -0.0009344238787889481,
-        -0.0014912645565345883,
-        -0.01038195751607418,
-        0.5936756134033203,
-        -0.00014933485363144428,
-        0.006156634073704481,
-        0.009113290347158909,
-        -0.001009406871162355,
-        -0.0015364158898591995,
-        -0.01074444130063057,
-        0.5944079756736755,
-        -0.0003295359492767602,
-        0.006216940935701132,
-        0.009534457698464394,
-        -0.0010867133969441056,
-        -0.0015858380356803536,
-        -0.011105277575552464,
-        0.5946523547172546,
-        -0.0005184217588976026,
-        0.006267391610890627,
-        0.009968035854399204,
-        -0.00116726232226938,
-        -0.001635841210372746,
-        -0.011476362124085426,
-        0.5947223901748657,
-        -0.0007183448178693652,
-        0.006307759787887335,
-        0.010411876253783703,
-        -0.0012491614324972034,
-        -0.0016832650871947408,
-        -0.011858409270644188,
-        0.5945469737052917,
-        -0.000922650913707912,
-        0.006340802181512117,
-        0.01086830347776413,
-        -0.0013331895461305976,
-        -0.0017334959702566266,
-        -0.012237261049449444,
-        0.5951407551765442,
-        -0.001138067222200334,
-        0.006367202382534742,
-        0.011336571536958218,
-        -0.0014204677427187562,
-        -0.001783861662261188,
-        -0.012612935155630112,
-        0.5946868658065796,
-        -0.0013612991897389293,
-        0.006382528692483902,
-        0.011817813850939274,
-        -0.0015092657413333654,
-        -0.0018364684656262398,
-        -0.013003477826714516,
-        0.595176637172699,
-        -0.0015915771946310997,
-        0.0063857706263661385,
-        0.01231638714671135,
-        -0.0015994333662092686,
-        -0.0018854403169825673,
-        -0.013393816538155079,
-        0.5955926179885864,
-        -0.001827688654884696,
-        0.006383033003658056,
-        0.012827028520405293,
-        -0.0016901197377592325,
-        -0.0019397168653085828,
-        -0.01378911454230547,
-        0.5955935120582581,
-        -0.002072748262435198,
-        0.006370946299284697,
-        0.013353490270674229,
-        -0.0017832767916843295,
-        -0.001991107827052474,
-        -0.014198213815689087,
-        0.595174252986908,
-        -0.0023247504141181707,
-        0.006349846720695496,
-        0.013896841555833817,
-        -0.0018769301241263747,
-        -0.0020415729377418756,
-        -0.014593549072742462,
-        0.5950352549552917,
-        -0.0025832781102508307,
-        0.0063181049190461636,
-        0.01445536408573389,
-        -0.0019729691557586193,
-        -0.002093689749017358,
-        -0.015006774105131626,
-        0.5953840613365173
-      ],
-      "std": [
-        0.002411161782220006,
-        0.0035220407880842686,
-        0.0029013415332883596,
-        0.0013258246472105384,
-        0.0020179853308945894,
-        0.010011205449700356,
-        0.4910063147544861,
-        0.0048064119182527065,
-        0.007010283879935741,
-        0.005793850403279066,
-        0.0026270404923707247,
-        0.004016784951090813,
-        0.019959986209869385,
-        0.4910795986652374,
-        0.0072022695094347,
-        0.01047250535339117,
-        0.008696537464857101,
-        0.003923765849322081,
-        0.006003526505082846,
-        0.029928741976618767,
-        0.4911730885505676,
-        0.009601284749805927,
-        0.013944732025265694,
-        0.011596043594181538,
-        0.0051947347819805145,
-        0.007982160896062851,
-        0.039855923503637314,
-        0.49109989404678345,
-        0.011995380744338036,
-        0.017401276156306267,
-        0.01450654398649931,
-        0.006452825851738453,
-        0.009959585964679718,
-        0.049764279276132584,
-        0.4912789762020111,
-        0.014385105110704899,
-        0.020854491740465164,
-        0.017407990992069244,
-        0.007683087605983019,
-        0.011908459477126598,
-        0.059651929885149,
-        0.4912128448486328,
-        0.016765346750617027,
-        0.024284135550260544,
-        0.02030768245458603,
-        0.008882720954716206,
-        0.013833398930728436,
-        0.06945177167654037,
-        0.4911862313747406,
-        0.01913418062031269,
-        0.02769860439002514,
-        0.023200584575533867,
-        0.010069170035421848,
-        0.01573152281343937,
-        0.07920067012310028,
-        0.49123910069465637,
-        0.021495427936315536,
-        0.03108437918126583,
-        0.02608773671090603,
-        0.011218286119401455,
-        0.017600294202566147,
-        0.08888415992259979,
-        0.4912588894367218,
-        0.023848513141274452,
-        0.03446579724550247,
-        0.028954684734344482,
-        0.012335631065070629,
-        0.0194363035261631,
-        0.0984557494521141,
-        0.4911799430847168,
-        0.02618129923939705,
-        0.0377996489405632,
-        0.031812794506549835,
-        0.013426545076072216,
-        0.021240200847387314,
-        0.10793018341064453,
-        0.4912393391132355,
-        0.02850136160850525,
-        0.04113452509045601,
-        0.03465236350893974,
-        0.01448102854192257,
-        0.023004556074738503,
-        0.11730524897575378,
-        0.491252601146698,
-        0.030810443684458733,
-        0.0444553941488266,
-        0.03749171271920204,
-        0.01552011538296938,
-        0.02474648877978325,
-        0.12660986185073853,
-        0.49126577377319336,
-        0.033104438334703445,
-        0.047724947333335876,
-        0.04029623046517372,
-        0.01652422547340393,
-        0.026438293978571892,
-        0.1357518434524536,
-        0.4913119971752167,
-        0.03538680821657181,
-        0.05097673833370209,
-        0.04309060424566269,
-        0.017495229840278625,
-        0.02809188887476921,
-        0.14484061300754547,
-        0.49129876494407654,
-        0.037644777446985245,
-        0.054201558232307434,
-        0.045869018882513046,
-        0.018448922783136368,
-        0.029708286747336388,
-        0.1537874937057495,
-        0.4913118779659271,
-        0.03989279642701149,
-        0.05739012360572815,
-        0.04861682280898094,
-        0.019360067322850227,
-        0.031284283846616745,
-        0.16262821853160858,
-        0.4912920594215393,
-        0.04212509095668793,
-        0.060563746839761734,
-        0.05134649947285652,
-        0.020253626629710197,
-        0.03281664475798607,
-        0.17136116325855255,
-        0.4912656545639038,
-        0.04433578997850418,
-        0.06369910389184952,
-        0.05404900759458542,
-        0.021123351529240608,
-        0.03430699184536934,
-        0.1799684315919876,
-        0.4912393391132355,
-        0.0465271882712841,
-        0.06679350882768631,
-        0.05673054978251457,
-        0.02196338027715683,
-        0.035762034356594086,
-        0.1884494125843048,
-        0.49119946360588074,
-        0.04870201274752617,
-        0.0698690414428711,
-        0.0593833364546299,
-        0.02277866192162037,
-        0.03716905042529106,
-        0.19684050977230072,
-        0.4911530017852783,
-        0.05086364969611168,
-        0.0729081928730011,
-        0.06201274320483208,
-        0.023575762286782265,
-        0.03854267671704292,
-        0.20511800050735474,
-        0.49119293689727783,
-        0.052999626845121384,
-        0.07590420544147491,
-        0.06461741775274277,
-        0.024347545579075813,
-        0.03986750543117523,
-        0.21329139173030853,
-        0.4911530613899231,
-        0.05511622130870819,
-        0.07886126637458801,
-        0.06719071418046951,
-        0.02509898692369461,
-        0.04115770384669304,
-        0.22134073078632355,
-        0.4911796748638153,
-        0.05720451846718788,
-        0.0817820280790329,
-        0.06973543763160706,
-        0.02582353726029396,
-        0.04240873083472252,
-        0.22928737103939056,
-        0.49117299914360046,
-        0.05928165465593338,
-        0.08468781411647797,
-        0.07225947827100754,
-        0.026536913588643074,
-        0.0436241552233696,
-        0.23713935911655426,
-        0.4912128746509552,
-        0.06133345514535904,
-        0.08753105252981186,
-        0.07475195825099945,
-        0.02723076194524765,
-        0.04480902478098869,
-        0.2449100911617279,
-        0.49125248193740845,
-        0.06336378306150436,
-        0.09035338461399078,
-        0.07722371071577072,
-        0.027915935963392258,
-        0.04596531763672829,
-        0.25256893038749695,
-        0.49128544330596924,
-        0.06536917388439178,
-        0.0931331068277359,
-        0.07966616004705429,
-        0.028593653813004494,
-        0.04708719253540039,
-        0.26014769077301025,
-        0.49133822321891785,
-        0.06735322624444962,
-        0.09587426483631134,
-        0.08208146691322327,
-        0.029253413900732994,
-        0.04818820580840111,
-        0.2676284909248352,
-        0.4913316071033478,
-        0.06931550055742264,
-        0.09857368469238281,
-        0.084467314183712,
-        0.029913172125816345,
-        0.049258600920438766,
-        0.27503275871276855,
-        0.4913710951805115,
-        0.07124854624271393,
-        0.10124263167381287,
-        0.08683151751756668,
-        0.03056473471224308,
-        0.05031215026974678,
-        0.2823602259159088,
-        0.4912854731082916,
-        0.07316309213638306,
-        0.10387503355741501,
-        0.08917013555765152,
-        0.031205736100673676,
-        0.051339127123355865,
-        0.2896381616592407,
-        0.491338312625885,
-        0.0750511884689331,
-        0.10646932572126389,
-        0.09148410707712173,
-        0.03184282034635544,
-        0.05234614759683609,
-        0.2968275845050812,
-        0.49127253890037537,
-        0.07691479474306107,
-        0.10901817679405212,
-        0.0937764048576355,
-        0.032469242811203,
-        0.053335245698690414,
-        0.30398809909820557,
-        0.49133166670799255,
-        0.07874985039234161,
-        0.11152230948209763,
-        0.0960441529750824,
-        0.03308483585715294,
-        0.054303981363773346,
-        0.311068594455719,
-        0.49127233028411865,
-        0.08056764304637909,
-        0.11400332301855087,
-        0.09828827530145645,
-        0.03369200974702835,
-        0.055251531302928925,
-        0.31809186935424805,
-        0.49120599031448364,
-        0.08236116915941238,
-        0.11642979085445404,
-        0.10051009804010391,
-        0.034294869750738144,
-        0.05618305504322052,
-        0.32508960366249084,
-        0.4911465048789978,
-        0.0841296911239624,
-        0.11883242428302765,
-        0.10270429402589798,
-        0.034887079149484634,
-        0.0570959635078907,
-        0.3320097327232361,
-        0.4910062551498413,
-        0.08587083965539932,
-        0.12118621915578842,
-        0.10488012433052063,
-        0.0354686863720417,
-        0.05799192935228348,
-        0.3389015197753906,
-        0.49095919728279114,
-        0.08759026974439621,
-        0.12350363284349442,
-        0.10703437030315399,
-        0.03604672849178314,
-        0.05887560918927193,
-        0.3457629084587097,
-        0.4909456968307495,
-        0.08928582817316055,
-        0.12578748166561127,
-        0.10916976630687714,
-        0.03662176802754402,
-        0.059738170355558395,
-        0.35257115960121155,
-        0.49097952246665955,
-        0.09095519781112671,
-        0.1280231475830078,
-        0.11128115653991699,
-        0.03717926889657974,
-        0.06058768928050995,
-        0.35931187868118286,
-        0.4908647835254669,
-        0.09259875863790512,
-        0.13021688163280487,
-        0.11337121576070786,
-        0.03773583471775055,
-        0.061423756182193756,
-        0.3660118579864502,
-        0.4909525215625763,
-        0.09422165900468826,
-        0.13238278031349182,
-        0.11544124782085419,
-        0.03828861564397812,
-        0.06224775314331055,
-        0.3726641535758972,
-        0.4908578395843506,
-        0.09581982344388962,
-        0.13450181484222412,
-        0.11749458312988281,
-        0.03884110972285271,
-        0.06304997950792313,
-        0.3792511522769928,
-        0.49077698588371277,
-        0.09739767760038376,
-        0.13658517599105835,
-        0.11952484399080276,
-        0.03938090428709984,
-        0.06384975463151932,
-        0.3857972025871277,
-        0.49077683687210083,
-        0.09894638508558273,
-        0.1386304795742035,
-        0.12153798341751099,
-        0.03992167487740517,
-        0.06463739275932312,
-        0.39230042695999146,
-        0.4908582866191864,
-        0.10048258304595947,
-        0.14064013957977295,
-        0.12353785336017609,
-        0.040460873395204544,
-        0.06541262567043304,
-        0.398761510848999,
-        0.4908851981163025,
-        0.10199275612831116,
-        0.14262202382087708,
-        0.1255190521478653,
-        0.04099829122424126,
-        0.0661839172244072,
-        0.40516018867492676,
-        0.4908175766468048
-      ],
-      "q01": [
-        -0.007347712090983987,
-        -0.011486765748262403,
-        -0.005602361589856446,
-        -0.004156268547289073,
-        -0.007298904010653494,
-        -0.03456402559429407,
-        0.0,
-        -0.014532031904906033,
-        -0.023063078013062475,
-        -0.011146851527690887,
-        -0.008034631153941155,
-        -0.014402847960591318,
-        -0.0688410488665104,
-        0.0,
-        -0.02174613598883152,
-        -0.03454083656966686,
-        -0.016602809804677963,
-        -0.012004367487877608,
-        -0.02169839859008789,
-        -0.10323694312572479,
-        0.0,
-        -0.02924629563242197,
-        -0.046255945521593095,
-        -0.02208325160816312,
-        -0.01599733569249511,
-        -0.029179477728903294,
-        -0.13703150334358216,
-        0.0,
-        -0.0364012400791049,
-        -0.057891583660244945,
-        -0.027614226043969393,
-        -0.020049774289131163,
-        -0.03620416237413883,
-        -0.17225709611177442,
-        0.0,
-        -0.04366054475307465,
-        -0.06915460828691722,
-        -0.03313590524941683,
-        -0.023996917428076275,
-        -0.0432967299669981,
-        -0.20717854452133178,
-        0.0,
-        -0.0507191886767745,
-        -0.08091499265134335,
-        -0.038471172228455544,
-        -0.02769389592409134,
-        -0.05027697439491748,
-        -0.240767266279459,
-        0.0,
-        -0.05764442007690668,
-        -0.0928664004802704,
-        -0.04389051187932491,
-        -0.031378988534212116,
-        -0.05700045920312405,
-        -0.27346896691322325,
-        0.0,
-        -0.06481191511154175,
-        -0.10385700676739215,
-        -0.04937767976522446,
-        -0.034899483323097225,
-        -0.0640901755988598,
-        -0.3072322327375412,
-        0.0,
-        -0.07216909334659577,
-        -0.11465946695804596,
-        -0.05446739228963852,
-        -0.0384699240922928,
-        -0.07105188292860984,
-        -0.34030354830026627,
-        0.0,
-        -0.0791815688431263,
-        -0.12541235695779324,
-        -0.05973851401358843,
-        -0.04202334811687469,
-        -0.07781512894928456,
-        -0.37346914051771163,
-        0.0,
-        -0.08563443173170089,
-        -0.13601192309856414,
-        -0.06484265891909599,
-        -0.045433390045166,
-        -0.0843420141339302,
-        -0.40387091467380526,
-        0.0,
-        -0.09227466106414796,
-        -0.14718177989125253,
-        -0.06990853486061097,
-        -0.04866062937080862,
-        -0.09135572135448455,
-        -0.43357558735609053,
-        0.0,
-        -0.09871668579578399,
-        -0.15755651119351388,
-        -0.07494964042305946,
-        -0.05228844651579856,
-        -0.0973639042377472,
-        -0.4662614274024963,
-        0.0,
-        -0.10474277555942535,
-        -0.16660683050751687,
-        -0.07973185749202967,
-        -0.055503451734781245,
-        -0.1043376196026802,
-        -0.49795184282064436,
-        0.0,
-        -0.11146926714479923,
-        -0.17675033633708956,
-        -0.08481720062792301,
-        -0.05867235680818558,
-        -0.11146564957499505,
-        -0.5234594867229462,
-        0.0,
-        -0.1183029289394617,
-        -0.18591568176746368,
-        -0.08976946114003659,
-        -0.06189141765832901,
-        -0.11618306014537812,
-        -0.5559864712238312,
-        0.0,
-        -0.1254465562880039,
-        -0.1967654939889908,
-        -0.09452381261885166,
-        -0.06458974866867065,
-        -0.12229588437080384,
-        -0.5838854856491089,
-        0.0,
-        -0.13160774660110475,
-        -0.20561206495761872,
-        -0.09930612608790398,
-        -0.06763142464160918,
-        -0.12729683189988136,
-        -0.6118898140192032,
-        0.0,
-        -0.13866937249898909,
-        -0.21400038709640504,
-        -0.10383093198537827,
-        -0.07044867560863496,
-        -0.1330213810920715,
-        -0.6420615591049195,
-        0.0,
-        -0.14518664595484734,
-        -0.22208737790584565,
-        -0.10839852929115296,
-        -0.07363164942264558,
-        -0.13909859127998353,
-        -0.6659443435192108,
-        0.0,
-        -0.15202519185543062,
-        -0.23166744391918181,
-        -0.1132360664010048,
-        -0.07554293347597121,
-        -0.1431354057431221,
-        -0.6935547601938247,
-        0.0,
-        -0.1575157383620739,
-        -0.24115877079963685,
-        -0.11769197665154933,
-        -0.07880985802412033,
-        -0.14886759107112885,
-        -0.717457844042778,
-        0.0,
-        -0.1637449370443821,
-        -0.25002506982684136,
-        -0.12177611397504806,
-        -0.08111752626895904,
-        -0.15336844004392625,
-        -0.7450204710006714,
-        0.0,
-        -0.1702753342092037,
-        -0.25778238211274146,
-        -0.12590762196183203,
-        -0.08350524621009828,
-        -0.15739910807609558,
-        -0.7672739201784134,
-        0.0,
-        -0.17570297884345054,
-        -0.26699419630765914,
-        -0.12999144119024275,
-        -0.08586254574060437,
-        -0.15982728340625763,
-        -0.7919603824377061,
-        0.0,
-        -0.18209079703688621,
-        -0.2762185108423233,
-        -0.1341805691242218,
-        -0.08846430575847625,
-        -0.1640431535243988,
-        -0.8073872842788696,
-        0.0,
-        -0.18759027983546256,
-        -0.28415180140137675,
-        -0.13808169324100017,
-        -0.09098501863479613,
-        -0.1681320061683655,
-        -0.8263913946390152,
-        0.0,
-        -0.193171888589859,
-        -0.29068409420847896,
-        -0.14178944163620472,
-        -0.09304863989353179,
-        -0.1707449821949005,
-        -0.8507119262695311,
-        0.0,
-        -0.19844869979619978,
-        -0.2987963397145271,
-        -0.14549240586161613,
-        -0.09557797043323518,
-        -0.1739663097858429,
-        -0.8648041958332062,
-        0.0,
-        -0.20466184456944464,
-        -0.3049391875267029,
-        -0.14948453903496267,
-        -0.09817119854688644,
-        -0.17706590107083323,
-        -0.8849965750694275,
-        0.0,
-        -0.20983124628067015,
-        -0.3116496807098389,
-        -0.15297152189016341,
-        -0.10035031213760376,
-        -0.18140975254774094,
-        -0.9072247592926024,
-        0.0,
-        -0.215193039560318,
-        -0.31998531384468076,
-        -0.1565369277715683,
-        -0.102377852499485,
-        -0.1837332302033901,
-        -0.9293165444850922,
-        0.0,
-        -0.22114226788282393,
-        -0.32775930384397506,
-        -0.16037838854789732,
-        -0.10488562031984328,
-        -0.18766805601119993,
-        -0.9526331702470778,
-        0.0,
-        -0.22637617546319963,
-        -0.33561479420661927,
-        -0.16403944179415703,
-        -0.10686019870638847,
-        -0.19061433601379393,
-        -0.9705972829818725,
-        0.0,
-        -0.23192334132194517,
-        -0.3404881935119629,
-        -0.16749516949653626,
-        -0.10840021744370465,
-        -0.19327300127148628,
-        -0.9914580883026123,
-        0.0,
-        -0.23655534625053407,
-        -0.3483747916221619,
-        -0.1713618674516678,
-        -0.11093877178430556,
-        -0.19658433060646058,
-        -1.0074259456634522,
-        0.0,
-        -0.24189636796712874,
-        -0.35468626945018766,
-        -0.174676346886158,
-        -0.1137608123421669,
-        -0.1998320359289646,
-        -1.02401462726593,
-        0.0,
-        -0.24619000346064568,
-        -0.3617886746406555,
-        -0.17737202043533326,
-        -0.11608646607995032,
-        -0.20212158007621764,
-        -1.0396305271863937,
-        0.0,
-        -0.25069174761176105,
-        -0.3690223491430283,
-        -0.18019738556742668,
-        -0.11791741763353347,
-        -0.20542705215215684,
-        -1.056211252117157,
-        0.0,
-        -0.25575273798704146,
-        -0.37492762922048567,
-        -0.18413128078579902,
-        -0.11963063821196557,
-        -0.2084871412396431,
-        -1.0777596963644027,
-        0.0,
-        -0.26051444165706633,
-        -0.38124460592269893,
-        -0.18717371021509172,
-        -0.12171284379959108,
-        -0.21154849707484244,
-        -1.0918209570646285,
-        0.0,
-        -0.2645109035611153,
-        -0.38571969106197357,
-        -0.19055300951004028,
-        -0.12364264481067655,
-        -0.21497644165158272,
-        -1.1041330133199692,
-        0.0,
-        -0.26856355381011965,
-        -0.3920482913136482,
-        -0.19341499418020247,
-        -0.12596707943677904,
-        -0.21652717173099517,
-        -1.1284716510772705,
-        0.0,
-        -0.273536373257637,
-        -0.3985861983895302,
-        -0.19648525762557983,
-        -0.12863396686315537,
-        -0.2178426690995693,
-        -1.1498000781059265,
-        0.0,
-        -0.2773340993881226,
-        -0.4043089840650559,
-        -0.19916273007392882,
-        -0.1301111794948578,
-        -0.21953763974308968,
-        -1.1649874792098998,
-        0.0,
-        -0.28152843511104586,
-        -0.4105489842891693,
-        -0.20206411379575728,
-        -0.13238328326940535,
-        -0.22030045819282532,
-        -1.186898805141449,
-        0.0,
-        -0.28521604499816894,
-        -0.41719991471767426,
-        -0.20488486919403076,
-        -0.13367428585290908,
-        -0.2215240351676941,
-        -1.2064387265205383,
-        0.0,
-        -0.28908531114459035,
-        -0.4219293355822563,
-        -0.20779289478063584,
-        -0.1346039095699787,
-        -0.22363287283182146,
-        -1.225643195438385,
-        0.0,
-        -0.294126475507021,
-        -0.42881255321502687,
-        -0.21011174390912055,
-        -0.13763879417181013,
-        -0.22653749623298647,
-        -1.2400950877666475,
-        0.0
-      ],
-      "q99": [
-        0.00760505946278572,
-        0.01227881843298674,
-        0.009394833338446915,
-        0.0041071553006768215,
-        0.005846777586638929,
-        0.03394978983253241,
-        0.9998,
-        0.015092343334481124,
-        0.024545711845159537,
-        0.018547782172262668,
-        0.008136489582061764,
-        0.011534944232553239,
-        0.0680388870239258,
-        0.9998,
-        0.022567954233288764,
-        0.03644986734837294,
-        0.0278723166257143,
-        0.012022978414595133,
-        0.017261751674115658,
-        0.1018139317035675,
-        0.9998,
-        0.030176736226677894,
-        0.048319679215550415,
-        0.037043299572169786,
-        0.015823813125491137,
-        0.022935147598385802,
-        0.13534477652311327,
-        0.9998,
-        0.03749606138020754,
-        0.05974762381315232,
-        0.046320767011493444,
-        0.01970605190098286,
-        0.028365900403261188,
-        0.16783747158050544,
-        0.9998,
-        0.04486110335588456,
-        0.07147922609001398,
-        0.055334820427000525,
-        0.023239636448025694,
-        0.034125556316971783,
-        0.2017985020875931,
-        0.9998,
-        0.052173979285359384,
-        0.08346393676698208,
-        0.06446339850723744,
-        0.026847184759378445,
-        0.03955972051620485,
-        0.23437213500142096,
-        0.9998,
-        0.059150983133912094,
-        0.09537746536731717,
-        0.0733001764550805,
-        0.029705484813451755,
-        0.04516481494158506,
-        0.26646220774650575,
-        0.9998,
-        0.06645039129257202,
-        0.1076395375341177,
-        0.08239804291725158,
-        0.0328832214474678,
-        0.04995994846820831,
-        0.29610841860771175,
-        0.9998,
-        0.07374503636360169,
-        0.11985380845069887,
-        0.09145532561540604,
-        0.035987181842327104,
-        0.05490816103816032,
-        0.3266934569597244,
-        0.9998,
-        0.08122003978192807,
-        0.1319003387391567,
-        0.10024555467069149,
-        0.038626870799064655,
-        0.05953236908912657,
-        0.35580547877550117,
-        0.9998,
-        0.08822669934630395,
-        0.14304091303348543,
-        0.1088144795835018,
-        0.041924731212854394,
-        0.06486271733045579,
-        0.3876368781208992,
-        0.9998,
-        0.09569223463535309,
-        0.1534084576368332,
-        0.11751553359031677,
-        0.04469002693891522,
-        0.06991170048713685,
-        0.41592443135976787,
-        0.9998,
-        0.10324303380250932,
-        0.1640547949969769,
-        0.1258540060549974,
-        0.04742067799866201,
-        0.0741334307938814,
-        0.4491768553853035,
-        0.9998,
-        0.1099683414697647,
-        0.17510256106853483,
-        0.13403582906574008,
-        0.050331409102678304,
-        0.07791369587183,
-        0.475221610414982,
-        0.9998,
-        0.1170920331031084,
-        0.1857688842058181,
-        0.14203765797317025,
-        0.05343856880664827,
-        0.0820521941632032,
-        0.5052882493734359,
-        0.9998,
-        0.12414569870531561,
-        0.19753187570571906,
-        0.14974090793728828,
-        0.0557895161241293,
-        0.0865580612182617,
-        0.5326116788625717,
-        0.9998,
-        0.13112460071146492,
-        0.20844777574539186,
-        0.15755365291237833,
-        0.05866813796758652,
-        0.09100452079772947,
-        0.5617906215190887,
-        0.9998,
-        0.1379249924659729,
-        0.21917983024120336,
-        0.16431246396899224,
-        0.06069063851833345,
-        0.09500974670350554,
-        0.5912433532238006,
-        0.9998,
-        0.14485807346105578,
-        0.2286938194990158,
-        0.1716713245987892,
-        0.0637721403002739,
-        0.09919737915396692,
-        0.6155789003372192,
-        0.9998,
-        0.15117210847139356,
-        0.23827072948217387,
-        0.1779224501848221,
-        0.06597466281652448,
-        0.10370546174049378,
-        0.644374561071396,
-        0.9998,
-        0.15787530031800268,
-        0.24751081448793422,
-        0.18454510100185867,
-        0.06831172816753389,
-        0.10792329218983648,
-        0.6659029854536058,
-        0.9998,
-        0.1641180765330792,
-        0.2572556215524673,
-        0.19053607739508155,
-        0.07030175954103468,
-        0.11277426195144652,
-        0.686683092904091,
-        0.9998,
-        0.16966928005814552,
-        0.26658033823966976,
-        0.19653976464271544,
-        0.07203448786735533,
-        0.11643967783451081,
-        0.7125654235839844,
-        0.9998,
-        0.17621982062458996,
-        0.27701201248764995,
-        0.20197029424905777,
-        0.07416592988967896,
-        0.12035323338508608,
-        0.7360334300994871,
-        0.9998,
-        0.1819627068042755,
-        0.2855635068178177,
-        0.2084233057498932,
-        0.07603480890989306,
-        0.1241462073802948,
-        0.7568318559646605,
-        0.9998,
-        0.18748426148295405,
-        0.29386607773303985,
-        0.21488302874565124,
-        0.0777944794893265,
-        0.1279024946689606,
-        0.7800548309087754,
-        0.9998,
-        0.19402539111375805,
-        0.304579441523552,
-        0.22076346032619476,
-        0.07946800847053526,
-        0.13082660132646562,
-        0.8035489074707032,
-        0.9998,
-        0.1999460756778717,
-        0.31222576675415037,
-        0.2264517953634262,
-        0.08081280782818795,
-        0.13460262061357497,
-        0.828994060134888,
-        0.9998,
-        0.20509765752553943,
-        0.3203624894678593,
-        0.23198571143150332,
-        0.08211237077713013,
-        0.13750549141168594,
-        0.8529169931411746,
-        0.9998,
-        0.209518383550644,
-        0.3281992249488831,
-        0.23816398755311968,
-        0.08322777889966959,
-        0.14037072820067403,
-        0.8746512705802916,
-        0.9998,
-        0.21442690343856818,
-        0.33666496825218195,
-        0.24309153485894203,
-        0.08529254732131952,
-        0.14304055777192115,
-        0.8924013189554216,
-        0.9998,
-        0.21941403126716608,
-        0.3448349338173866,
-        0.2489651171147823,
-        0.08667572829127312,
-        0.14580873468518263,
-        0.9107246013164523,
-        0.9998,
-        0.22309734332561493,
-        0.3517628137350083,
-        0.2547237434387207,
-        0.08896696496009826,
-        0.1489701636433602,
-        0.9272562242746352,
-        0.9998,
-        0.22793451219797134,
-        0.3591047751426697,
-        0.2597683270871639,
-        0.09042025351524352,
-        0.1528471918106079,
-        0.9483436929702762,
-        0.9998,
-        0.23258997063636777,
-        0.36902411174774175,
-        0.26485966441631315,
-        0.09243017539381981,
-        0.15506044900417326,
-        0.9667170533418656,
-        0.9998,
-        0.23665347018241878,
-        0.3762736029624939,
-        0.27067578415870663,
-        0.09449174585938458,
-        0.15888449568748475,
-        0.9834513599395751,
-        0.9998,
-        0.24093078261613854,
-        0.3843366762638093,
-        0.2768812180876732,
-        0.09594143676757816,
-        0.16144601993560792,
-        1.0022019856452942,
-        0.9998,
-        0.2459272434949874,
-        0.3907656828403474,
-        0.2815030888676643,
-        0.09760697330236434,
-        0.16464969681501385,
-        1.019599228835106,
-        0.9998,
-        0.2511993800699711,
-        0.3962667070865631,
-        0.28746061475872997,
-        0.09892506269812584,
-        0.1685606077313423,
-        1.0345355377674106,
-        0.9998,
-        0.25586065447330475,
-        0.4064154948949814,
-        0.29352941393256193,
-        0.10010581925511364,
-        0.17137197246551505,
-        1.0455808718204502,
-        0.9998,
-        0.26066332874298104,
-        0.41295039305686965,
-        0.29917095521688464,
-        0.10042237902879714,
-        0.17334867343306548,
-        1.0644763714551928,
-        0.9998,
-        0.26424761767387395,
-        0.4171489597439766,
-        0.3047941813468933,
-        0.10030385837554934,
-        0.1771283437609672,
-        1.0775494430065153,
-        0.9998,
-        0.266823394203186,
-        0.42120621981620787,
-        0.31085526600480073,
-        0.10133219141960148,
-        0.18098006631135943,
-        1.0927770781517028,
-        0.9998,
-        0.27192984950542454,
-        0.4258830359578132,
-        0.31648782253265384,
-        0.10210902032852165,
-        0.18391375703215596,
-        1.1100240562438963,
-        0.9998,
-        0.2774573613643646,
-        0.4328473631143568,
-        0.3229170823454858,
-        0.10328457579612738,
-        0.18559744659066202,
-        1.127002547836304,
-        0.9998,
-        0.280809062719345,
-        0.43832696628570567,
-        0.32813225684762004,
-        0.10458858071565635,
-        0.18929516908526423,
-        1.1399588096141815,
-        0.9998,
-        0.28525815182924275,
-        0.44289318814277645,
-        0.3333625353813172,
-        0.1055313608527183,
-        0.1918839191913605,
-        1.1545847342491151,
-        0.9998,
-        0.2899247071444988,
-        0.4499819297909735,
-        0.33962785017490393,
-        0.10614367849826811,
-        0.19452174307107928,
-        1.1747124679088592,
-        0.9998,
-        0.29274274289608004,
-        0.4549682779312134,
-        0.3463232246458531,
-        0.10680567473173141,
-        0.19644608254432672,
-        1.185605804681778,
-        0.9998
-      ]
-    }
-  },
-  "sample_ratio": 0.05,
-  "dataset_infos": {
-    "robotgeneralist/pp__1": {
-      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
-      "num_episodes": 237,
-      "num_frames": 99363
-    },
-    "robotgeneralist/pp__1__adversarial": {
-      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
-      "num_episodes": 64,
-      "num_frames": 17906
-    },
-    "robotgeneralist/pp__2": {
-      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
-      "num_episodes": 92,
-      "num_frames": 37458
-    },
-    "robotgeneralist/pp__2__rotations": {
-      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
-      "num_episodes": 102,
-      "num_frames": 13935
-    },
-    "robotgeneralist/pp__3__with_ft": {
-      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
-      "num_episodes": 99,
-      "num_frames": 38424
-    },
-    "robotgeneralist/pp__3__with_ft__rotations": {
-      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
-      "num_episodes": 49,
-      "num_frames": 7151
-    },
-    "robotgeneralist/pp__3__with_ft__adversarial_start": {
-      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
-      "num_episodes": 147,
-      "num_frames": 21313
-    },
-    "robotgeneralist/pp__4__with_ft": {
-      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
-      "num_episodes": 94,
-      "num_frames": 33046
-    },
-    "robotgeneralist/pp__5__with_ft__adversarial_target": {
-      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
-      "num_episodes": 48,
-      "num_frames": 10615
-    },
-    "robotgeneralist/pp__6__with_ft": {
-      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
-      "num_episodes": 379,
-      "num_frames": 123611
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-1": {
-      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
-      "num_episodes": 98,
-      "num_frames": 41410
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-2": {
-      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
-      "num_episodes": 94,
-      "num_frames": 37859
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-3": {
-      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
-      "num_episodes": 127,
-      "num_frames": 53341
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-4": {
-      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
-      "num_episodes": 102,
-      "num_frames": 38263
-    }
-  }
-}
\ No newline at end of file
diff --git a/pp__ee_baseline/15000/params/_METADATA b/pp__ee_baseline/15000/params/_METADATA
deleted file mode 100644
index 9eb73624158f4514d49041e0806d2d702c62eee1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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, "write_shape": [1152]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304]}}, "('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, "write_shape": [27, 1152, 4304]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 16, 72, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [14, 14, 3, 1152]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1152, 2048]}}, "('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, "write_shape": [1, 256, 1152]}}, "('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, "write_shape": [257152, 2048]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [18, 8, 256, 2048]}}, "('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, "write_shape": [18, 8, 256, 1024]}}, "('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, "write_shape": [18, 2, 1, 2048, 256]}}, "('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, "write_shape": [18, 2, 1, 1024, 256]}}, "('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, "write_shape": [18, 8, 2048, 256]}}, "('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, "write_shape": [18, 8, 1024, 256]}}, "('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, "write_shape": [18, 2, 2048, 16384]}}, "('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, "write_shape": [18, 16384, 2048]}}, "('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, "write_shape": [18, 2, 1024, 4096]}}, "('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, "write_shape": [18, 4096, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}, "('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, "write_shape": [32]}}, "('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, "write_shape": [1024, 32]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [2048, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [1024, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/pp__ee_baseline/15000/params/_sharding b/pp__ee_baseline/15000/params/_sharding
deleted file mode 100644
index 924608abdcb99abb3c864bb9d1bfca6eb60f780e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/_sharding
+++ /dev/null
@@ -1 +0,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}]]}}"}
\ No newline at end of file
diff --git a/pp__ee_baseline/15000/params/array_metadatas/process_0 b/pp__ee_baseline/15000/params/array_metadatas/process_0
deleted file mode 100644
index 6aa9fd1decf72607133eb54664d98802bf563b5c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/array_metadatas/process_0
+++ /dev/null
@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/pp__ee_baseline/15000/params/d/0642a9b41480d61c48c3f6c74642c184 b/pp__ee_baseline/15000/params/d/0642a9b41480d61c48c3f6c74642c184
deleted file mode 100644
index 39479db806a36dd9aae25cc46c02647aa19ee403..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/d/0642a9b41480d61c48c3f6c74642c184 and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/d/f5f67893d624430fa4151e9280ad81c7 b/pp__ee_baseline/15000/params/d/f5f67893d624430fa4151e9280ad81c7
deleted file mode 100644
index ccf44df50fbec8d18a10ae6b45171e8ec4ebb8b5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/d/f5f67893d624430fa4151e9280ad81c7 and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/manifest.ocdbt b/pp__ee_baseline/15000/params/manifest.ocdbt
deleted file mode 100644
index 7cef839cedb9e0493c6a150555e4ecaaedfa19a0..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/04773cadd05501eeaab9559550b1c9b1 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/04773cadd05501eeaab9559550b1c9b1
deleted file mode 100644
index 626bed3bf8947e8f30eab3916f3d9c4121a4b00f..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/04773cadd05501eeaab9559550b1c9b1
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b966f98efacb76c93929390881bd06b9a4708ac841b79dae081ec811dc6bcda7
-size 132845101
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/0a68df7f3576a5ea9c125296c1b4c73a b/pp__ee_baseline/15000/params/ocdbt.process_0/d/0a68df7f3576a5ea9c125296c1b4c73a
deleted file mode 100644
index aead0a1ee5c3949f858bfea4033ce717dbcedafc..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/0a68df7f3576a5ea9c125296c1b4c73a
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c042816e7a2a1386c7297f8ccdbace4b564488b188415e0e69cbf90d57707214
-size 496758369
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/24e0a5ae1e27cd5fb5a6378a740f94c7 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/24e0a5ae1e27cd5fb5a6378a740f94c7
deleted file mode 100644
index f37f33ff392253147baf13e2b6034982bcc66521..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/24e0a5ae1e27cd5fb5a6378a740f94c7
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2410440555bf7e17d14abf6de0d1eb0a243fbc42398b515466aed949c64a97a9
-size 70083544
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/27931def2a5d68c7857a8d76775feda8 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/27931def2a5d68c7857a8d76775feda8
deleted file mode 100644
index f6263573c2387ea5741c2f9e6f94608724691288..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/27931def2a5d68c7857a8d76775feda8
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:196c2d82434aa24bf91a1f0932515ce4ef924cb6eb45d849a32c77816fd3c35f
-size 1120115517
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2e53cd2cdf91583439eb5b2a28431a34 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/2e53cd2cdf91583439eb5b2a28431a34
deleted file mode 100644
index 0da66614f5ee456deedbc4b51627c503fbb354ae..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2e53cd2cdf91583439eb5b2a28431a34
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f5b4ea17bf83ababf95c676d1fd712f274ebf05e693742a67f1652c5a1b5aebf
-size 497035932
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2ef166e815348f2158dbadf1c02bf90c b/pp__ee_baseline/15000/params/ocdbt.process_0/d/2ef166e815348f2158dbadf1c02bf90c
deleted file mode 100644
index 69263159fa6e150e38b9deadfab2dd7cd0ab73fa..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2ef166e815348f2158dbadf1c02bf90c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5092bbc52e01a0411e1f7194b8f4a62bf502e9a3c901bbca5d725aba110be5ba
-size 34974890
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2f15d33e81f1a8e9f94f1125019396b1 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/2f15d33e81f1a8e9f94f1125019396b1
deleted file mode 100644
index 38f23ba39e339a20c75e97a05f926a26458f6b2d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/2f15d33e81f1a8e9f94f1125019396b1
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:570dc1f0e4e2b62f91a50d1fa05360e945e129231e63e3c2155f0e68bb2f3d4f
-size 280036770
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/30f3deb70a6296d206343859b18f314f b/pp__ee_baseline/15000/params/ocdbt.process_0/d/30f3deb70a6296d206343859b18f314f
deleted file mode 100644
index 6a0cff993bb46d04284de082b2b234a99252ef29..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/30f3deb70a6296d206343859b18f314f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:135f377db15e6a2deadae70ec6aa846d9131a185f8d14e2fc7a594fb829070f7
-size 280187828
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/343e079d3c50f6f90f38dc8ade8c837b b/pp__ee_baseline/15000/params/ocdbt.process_0/d/343e079d3c50f6f90f38dc8ade8c837b
deleted file mode 100644
index 3b7297f08520e8d4dff00a2b0665def12ea6712d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/343e079d3c50f6f90f38dc8ade8c837b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:68302b4934ca73517556ccb3512620c5ac548eacd75e2480a7a8eb18df1acbc4
-size 496760927
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cb8a6db22f5405ddef80c15cc505f7f b/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cb8a6db22f5405ddef80c15cc505f7f
deleted file mode 100644
index 00547839887daafd9897a8c824214b31afce04fb..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cb8a6db22f5405ddef80c15cc505f7f and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cbe8bb06879cde422584b40b6442cbf b/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cbe8bb06879cde422584b40b6442cbf
deleted file mode 100644
index fe639b9d0a7dece1414c4368a7d9d5c110ab003d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/3cbe8bb06879cde422584b40b6442cbf
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3a74fbbc0f8a91c32d73c31980902adae19c4c5ac367e9640164d8ee60d8a50a
-size 451783
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/41b2961d8a5f20c237cc3d518f37e31a b/pp__ee_baseline/15000/params/ocdbt.process_0/d/41b2961d8a5f20c237cc3d518f37e31a
deleted file mode 100644
index f0586099df58bda214c71f0d8f1f2cf916e75edf..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/41b2961d8a5f20c237cc3d518f37e31a
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bad39b4d8d99e19c849f575483b4434e5ba45789d201e0eac7b2dbd44a11df2c
-size 280187073
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/429eb1f1469de162cf6306fbe8d38319 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/429eb1f1469de162cf6306fbe8d38319
deleted file mode 100644
index c690c18dcaa7481151afd136767259f3b3465bfc..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/429eb1f1469de162cf6306fbe8d38319
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ffd4ae26b3645613410ecc6bd95d591fa04100deb0ae0ba70306d8a0f2f4acb4
-size 279248306
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/437ecd1ebea63f8cd731e5de3050ca3e b/pp__ee_baseline/15000/params/ocdbt.process_0/d/437ecd1ebea63f8cd731e5de3050ca3e
deleted file mode 100644
index df47189208361344ca444c17400700eb397f7bc5..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/437ecd1ebea63f8cd731e5de3050ca3e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5948b03cb55e42e661fb5d05e989d7d7337d5f37116cacb578a6973ba48812b0
-size 1120114597
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/43f9a4ecdfbc734fff761ab192fbb91b b/pp__ee_baseline/15000/params/ocdbt.process_0/d/43f9a4ecdfbc734fff761ab192fbb91b
deleted file mode 100644
index 02b9aa7290198a3784e980ed9f9663b6160c73e8..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/43f9a4ecdfbc734fff761ab192fbb91b and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/46aeba433b084be6f6f12ebebedab4d6 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/46aeba433b084be6f6f12ebebedab4d6
deleted file mode 100644
index b714762f58bac21abeacca89fcb7441013287441..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/46aeba433b084be6f6f12ebebedab4d6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9a0ec1fc756b1f8356a5cc6f1b42edcaed9f68910c658f577b03a4e3ccc6d60f
-size 132938139
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/48094155443b23664065ecd73e045f54 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/48094155443b23664065ecd73e045f54
deleted file mode 100644
index 78fcbd2e63867a7c85d8c685047b08d300d76240..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/48094155443b23664065ecd73e045f54
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c0d8998070aa1566db9d0ae30c3adf0446e3ca2cbdda3014d7f527fc5fc1a197
-size 17536470
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/48fc0e6dac56774c8629bba70afc871f b/pp__ee_baseline/15000/params/ocdbt.process_0/d/48fc0e6dac56774c8629bba70afc871f
deleted file mode 100644
index 103703c73975de34f4977daa4f6f6622420205b3..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/48fc0e6dac56774c8629bba70afc871f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:00b89612c8236a61b177877eb3c68e8629fba6a4b7af7d0d34fa0b8357103954
-size 70081999
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5122c441dfcc25f0d1a9b10657d6d596 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/5122c441dfcc25f0d1a9b10657d6d596
deleted file mode 100644
index bbb5bb58fc0288f1da122998521494afabdb7365..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5122c441dfcc25f0d1a9b10657d6d596
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:50cea98887f5e21f51c970fc02b93f7bde4c691a015fd59ef326fa6add577758
-size 8755251
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/56c303f504b619a3be26f85fd67e37bf b/pp__ee_baseline/15000/params/ocdbt.process_0/d/56c303f504b619a3be26f85fd67e37bf
deleted file mode 100644
index 2fc03a1e9e6628cfae277d0adb66f36b7f36c9c2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/56c303f504b619a3be26f85fd67e37bf
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a2d6ba1c9485b2450f16296c6c02546eb5e96a698db25564c6ecd34d27f93be6
-size 1120113028
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5e82cf89837e25ac6e13c46e679dedba b/pp__ee_baseline/15000/params/ocdbt.process_0/d/5e82cf89837e25ac6e13c46e679dedba
deleted file mode 100644
index bcaedc3f38f742a64c2f7ac0793f4e68921fa1a8..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5e82cf89837e25ac6e13c46e679dedba
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:98cbae2fda2533dd47b6d39b34a8d09bdf77dd7252ded5dc59d0e45e626a5037
-size 280038573
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5f1e803fc9bce566b6f69757c0717eed b/pp__ee_baseline/15000/params/ocdbt.process_0/d/5f1e803fc9bce566b6f69757c0717eed
deleted file mode 100644
index fa0b5a67c60d2ee6e298b76d9f278d07ba95b5c2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/5f1e803fc9bce566b6f69757c0717eed
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a3bd33e396cd28e104c49d50d94234e407e59806917bad0ef99a545bd1d2f704
-size 1102601
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6189b28fa9437f7ccebf81fdd5a55c4b b/pp__ee_baseline/15000/params/ocdbt.process_0/d/6189b28fa9437f7ccebf81fdd5a55c4b
deleted file mode 100644
index 608e27e8ae81135e81e3f3ce6337d6da242fd004..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6189b28fa9437f7ccebf81fdd5a55c4b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:cfadf56273fd87e7178179101d2a2069ef88b00682329c4f5a647db2e10f9ce8
-size 11101670
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/670b27ef2d490fc0b55c235e66e05c5a b/pp__ee_baseline/15000/params/ocdbt.process_0/d/670b27ef2d490fc0b55c235e66e05c5a
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/670b27ef2d490fc0b55c235e66e05c5a and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6b8e6b3df5e601bb18e61b90eb43a2ff b/pp__ee_baseline/15000/params/ocdbt.process_0/d/6b8e6b3df5e601bb18e61b90eb43a2ff
deleted file mode 100644
index 238e1e04dcfa3457023d9a19c03f60cd14e64da4..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6b8e6b3df5e601bb18e61b90eb43a2ff
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1350c3bf6a12e913d6da6e3dd1321ca3cd42b26b230af957715ab415c669b567
-size 139970064
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6e4b5462124ac855300c5eab10fe265a b/pp__ee_baseline/15000/params/ocdbt.process_0/d/6e4b5462124ac855300c5eab10fe265a
deleted file mode 100644
index 43df9f45423b5ec9df531b743e11a481eaac7ee3..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/6e4b5462124ac855300c5eab10fe265a
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5d05cf6bfa09bb7473fba2014390d50d6397b5e1584c972099a690b4ccae1482
-size 497033788
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7be977f460d5b160757f51e8b82b67cb b/pp__ee_baseline/15000/params/ocdbt.process_0/d/7be977f460d5b160757f51e8b82b67cb
deleted file mode 100644
index 8f26e37da1a56cd4a7b759bc085c319393bc9fde..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7be977f460d5b160757f51e8b82b67cb
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1f44efff05e22abf3751c8edb76fc3f5beaeb4888291bb4fff37cc1f6b7f74cd
-size 559725723
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7cc3fc2f56d0ec2f321f61bd67c18d0e b/pp__ee_baseline/15000/params/ocdbt.process_0/d/7cc3fc2f56d0ec2f321f61bd67c18d0e
deleted file mode 100644
index dc674b78654c727334e61cfd53656286e0efe788..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7cc3fc2f56d0ec2f321f61bd67c18d0e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5b594f17938ce8da46e54733f86b9cf0a8fe0a0d55b443c90761ddf65c1303d5
-size 3077552484
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7dea218bec1c692836b8cd4a53f4b72b b/pp__ee_baseline/15000/params/ocdbt.process_0/d/7dea218bec1c692836b8cd4a53f4b72b
deleted file mode 100644
index 7440a7215d805fa292cc58551ef19ed508838d24..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/7dea218bec1c692836b8cd4a53f4b72b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d149f0d37bf09c06da8e0eb2eff16dfe0cf46a732a5f4b2dbc99e426b5e82425
-size 132968925
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/870122bafb763385d041ad335460dc8c b/pp__ee_baseline/15000/params/ocdbt.process_0/d/870122bafb763385d041ad335460dc8c
deleted file mode 100644
index 3cba8cf0c2133bd4558b875bd519551cd107c88a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/870122bafb763385d041ad335460dc8c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:78b5c8444050b92dd8e93a6d37d65887c13978b26615d98b696cb1ce32092524
-size 2240232091
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/872b44dd9e5093124d371b4338a93e84 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/872b44dd9e5093124d371b4338a93e84
deleted file mode 100644
index ccb217debbf67ac7a2480e8bc5d12fdfab2ddf97..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/872b44dd9e5093124d371b4338a93e84
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:82cde93fd3a38c1175b97d9fc899fa65d67ab4c843dca427e0e778efcf9d62ff
-size 132937578
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/8884eb2cbebe036a89843410b50b45c7 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/8884eb2cbebe036a89843410b50b45c7
deleted file mode 100644
index 0685c2569e255ff90d94e66a6cec4e1691707a31..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/8884eb2cbebe036a89843410b50b45c7
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8efd7f921275e7be12c2d87598c4138ef99eec62d6e9e8afd4ba25883c2cc2af
-size 1120129728
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/93f441307b0ede9c536bd217335ff976 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/93f441307b0ede9c536bd217335ff976
deleted file mode 100644
index 4d60a6468e3993cbe8c6fcb99870a1c218a62db9..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/93f441307b0ede9c536bd217335ff976
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:70d64ce90831087b5420b40951aa44ea0338a882316650f633db9471b0a09c64
-size 139971908
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a11b38946319da0ff4e1d98c86102cee b/pp__ee_baseline/15000/params/ocdbt.process_0/d/a11b38946319da0ff4e1d98c86102cee
deleted file mode 100644
index e3a59ee5b881295a450d88eefdcc3a147b05e62e..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a11b38946319da0ff4e1d98c86102cee and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a426c47110e97a527f8d280d7ec458b6 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/a426c47110e97a527f8d280d7ec458b6
deleted file mode 100644
index 7fabe0357b3f0d682419aed59d860fa51fffc468..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a426c47110e97a527f8d280d7ec458b6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:237f2d17df9c47a79a8eb624080456b4016a57556ab5be86c22c2149e6105c81
-size 279248800
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a5c8b09e019658924253eb954dc310a4 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/a5c8b09e019658924253eb954dc310a4
deleted file mode 100644
index 76675bc1baf7117b8edede108d283f57f4ad8df5..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/a5c8b09e019658924253eb954dc310a4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:00d699723497af339ae88f4ed2c36a5ddec349358b14a10e3380054db31f6303
-size 1120116122
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/aeca74801bd0bd532aa6c4251eef8508 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/aeca74801bd0bd532aa6c4251eef8508
deleted file mode 100644
index 0ab0523e2feae77e9cb0e266ca4e25724839fda6..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/aeca74801bd0bd532aa6c4251eef8508
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f2cc643fd030ecf5338fe1291f511560cf3e0a9bc01bd12189244ba1c04038c6
-size 538685454
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/af27cc8c05afe7bd5b260f5e8be36ab4 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/af27cc8c05afe7bd5b260f5e8be36ab4
deleted file mode 100644
index 9d8425b5e36e760c5e9dd63315aa0e90090eb8dc..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/af27cc8c05afe7bd5b260f5e8be36ab4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:931a4159eb658f784127b6166a169f60a8da3a482c02548a032f3a8853d02eb8
-size 8755271
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/b2ce084fc714405ab407dfcc635b8dc1 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/b2ce084fc714405ab407dfcc635b8dc1
deleted file mode 100644
index fa8d45fd6ed631f6ec084106cf1992def3561eb5..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/b2ce084fc714405ab407dfcc635b8dc1
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:dcebd131048d71dd5f0042b677b535b8b5458385f51a67a31adeaa48e2cace99
-size 559721799
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/c345916fb082ba0f2f6d2e7c879ad900 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/c345916fb082ba0f2f6d2e7c879ad900
deleted file mode 100644
index 7e9bdb67569414c3ea3b1467e6007a0235db5383..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/c345916fb082ba0f2f6d2e7c879ad900
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5d75face0c7dd675c59dd6dcecdb3d3998ebf15375c55f1e8a120d918d9447a0
-size 34975563
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/c5467d90eb6560ac25b0317550dafe14 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/c5467d90eb6560ac25b0317550dafe14
deleted file mode 100644
index 25c3d1bef7ff626a92692a316af4b30ce5d4ddea..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/c5467d90eb6560ac25b0317550dafe14
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d2c728b1429c9b167606a18b61ef33bef212989593b0283a183ac22652927f58
-size 1120115324
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/ca1f441dfe2c8eff72c5f7b142f7ce7e b/pp__ee_baseline/15000/params/ocdbt.process_0/d/ca1f441dfe2c8eff72c5f7b142f7ce7e
deleted file mode 100644
index af133a19df1c960dfc9a7bf25cbd872f452af016..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/ca1f441dfe2c8eff72c5f7b142f7ce7e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:351a5e2a054ea8d5a7d018918ef5317d480625bfc649e1914de42894712b88b1
-size 272880371
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/cd12ebebef9d93224c3bb9ce60fb15b3 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/cd12ebebef9d93224c3bb9ce60fb15b3
deleted file mode 100644
index 9099a06a9bbab8245827951207f3ed3cd2ddfa83..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/cd12ebebef9d93224c3bb9ce60fb15b3
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d999327a0dcf711ca63719ff1fcd7d3fa8c6f6891e9b56e4a531021cbe035452
-size 3077549108
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/d186679f6ddf4bdbabcdb7a845da21ec b/pp__ee_baseline/15000/params/ocdbt.process_0/d/d186679f6ddf4bdbabcdb7a845da21ec
deleted file mode 100644
index 153572562e8ed10dda3ae4910b416ecbc37d5786..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/d186679f6ddf4bdbabcdb7a845da21ec
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:69648558faa0fb5a06831f8830ff63ac8840d5d098d91dc92e57b3ef8e23cbc4
-size 1120137236
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/d3b9f7b8a34af4df54d540d397cdc23b b/pp__ee_baseline/15000/params/ocdbt.process_0/d/d3b9f7b8a34af4df54d540d397cdc23b
deleted file mode 100644
index 8f04a2770fb3e0caf5734e7f01c6215720c4b1d0..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/d3b9f7b8a34af4df54d540d397cdc23b and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/e13744720ad30fffc9105b236eb2cc69 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/e13744720ad30fffc9105b236eb2cc69
deleted file mode 100644
index ffc57c5cefe3160b4f5ec2c8030ab168d9324180..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/e13744720ad30fffc9105b236eb2cc69 and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/ea79914e9064fb9fa37e42c3a053e2ee b/pp__ee_baseline/15000/params/ocdbt.process_0/d/ea79914e9064fb9fa37e42c3a053e2ee
deleted file mode 100644
index 61035d8a118b3b9a5304f96158231a05769f3886..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/ea79914e9064fb9fa37e42c3a053e2ee
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:caa7288e3b182cd0a5a8a71762cea429719a0a5cf897192dee49a82cbc360936
-size 1120133516
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/eebd1b796db669abef0e69636c01eea5 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/eebd1b796db669abef0e69636c01eea5
deleted file mode 100644
index dca8534ae12098b79808f0b467da2b013bfc5de2..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/eebd1b796db669abef0e69636c01eea5 and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f25f6baaea8c5d81a5eaaeec5f758299 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/f25f6baaea8c5d81a5eaaeec5f758299
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f25f6baaea8c5d81a5eaaeec5f758299 and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f2b481ae7e786ab3948c6ca23c183893 b/pp__ee_baseline/15000/params/ocdbt.process_0/d/f2b481ae7e786ab3948c6ca23c183893
deleted file mode 100644
index 655964fac836bec770cfe2b0715567bcb869ed98..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f2b481ae7e786ab3948c6ca23c183893
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:331a23d32cf6bcd574fdb9534bc5e25c1aa88fb2311bf02714742d5e3a048939
-size 4888443
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f6668a44b53d5859c20203146b8214aa b/pp__ee_baseline/15000/params/ocdbt.process_0/d/f6668a44b53d5859c20203146b8214aa
deleted file mode 100644
index 5940bf061e445d32dcb9e5c3dbf1164bb16f90eb..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/d/f6668a44b53d5859c20203146b8214aa and /dev/null differ
diff --git a/pp__ee_baseline/15000/params/ocdbt.process_0/manifest.ocdbt b/pp__ee_baseline/15000/params/ocdbt.process_0/manifest.ocdbt
deleted file mode 100644
index 6e0eb8f37592f7db6eac5b3ef45ce81d76a757b7..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/15000/params/ocdbt.process_0/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/20000/_CHECKPOINT_METADATA b/pp__ee_baseline/20000/_CHECKPOINT_METADATA
deleted file mode 100644
index de2fdf1c1536e3f5b8a61d72480447f317d142df..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/_CHECKPOINT_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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": 1767233236882684101, "commit_timestamp_nsecs": 1767233282270859499, "custom_metadata": {}}
\ No newline at end of file
diff --git a/pp__ee_baseline/20000/assets/ur10e/norm_stats.json b/pp__ee_baseline/20000/assets/ur10e/norm_stats.json
deleted file mode 100644
index 8a1b886d6b668da778297b91b3d82381c2eab28a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/assets/ur10e/norm_stats.json
+++ /dev/null
@@ -1,1525 +0,0 @@
-{
-  "norm_stats": {
-    "state": {
-      "mean": [
-        0.05313729867339134,
-        0.0037223484832793474,
-        0.106409952044487,
-        0.0008666226058267057,
-        -0.00854077935218811,
-        0.01460699550807476,
-        0.5940247178077698
-      ],
-      "std": [
-        0.1176244392991066,
-        0.16962096095085144,
-        0.07773701101541519,
-        0.04537155479192734,
-        0.06616315245628357,
-        0.6038344502449036,
-        0.49107977747917175
-      ],
-      "q01": [
-        -0.2025573879778385,
-        -0.31106759839057924,
-        0.00890737680345774,
-        -0.11727181684970855,
-        -0.16711108471155167,
-        -1.423702568435669,
-        0.0
-      ],
-      "q99": [
-        0.29901714037060734,
-        0.4115018085956573,
-        0.3217185490995646,
-        0.13774808425903318,
-        0.19241279754638663,
-        1.598234023094177,
-        0.9998
-      ]
-    },
-    "actions": {
-      "mean": [
-        0.0001451610296498984,
-        0.00017207347264047712,
-        0.000048651279939804226,
-        0.00003654604734038003,
-        -0.000019767590856645256,
-        -0.00019861181499436498,
-        0.5944076776504517,
-        0.0002841090608853847,
-        0.00034722191048786044,
-        0.00011135452223243192,
-        0.00006827477045590058,
-        -0.000043589698179857805,
-        -0.0004028017574455589,
-        0.594025731086731,
-        0.00041486899135634303,
-        0.0005213047261349857,
-        0.00018569435633253306,
-        0.00009245114779332653,
-        -0.00007049143459880725,
-        -0.000616009347140789,
-        0.5935360789299011,
-        0.000542119552846998,
-        0.0006975074065849185,
-        0.00027102616149932146,
-        0.00011072693450842053,
-        -0.00010138262587133795,
-        -0.0008296052110381424,
-        0.593919575214386,
-        0.0006618681945838034,
-        0.0008754940354265273,
-        0.0003684645926114172,
-        0.00012383276771288365,
-        -0.000135598995257169,
-        -0.0010419095633551478,
-        0.5929783582687378,
-        0.0007787057547830045,
-        0.00105847988743335,
-        0.0004721089790109545,
-        0.00013335929543245584,
-        -0.00017038211808539927,
-        -0.0012720258673653007,
-        0.5933270454406738,
-        0.0008861633250489831,
-        0.001237610587850213,
-        0.0005911999032832682,
-        0.00013674379442818463,
-        -0.00020881379896309227,
-        -0.0014943619025871158,
-        0.5934669971466064,
-        0.0009874902898445725,
-        0.0014197468990460038,
-        0.0007170690805651248,
-        0.00013581194798462093,
-        -0.0002465852303430438,
-        -0.0017284756759181619,
-        0.5931888818740845,
-        0.0010826619109138846,
-        0.0016015341971069574,
-        0.0008584632305428386,
-        0.000128391373436898,
-        -0.00028913054848089814,
-        -0.0019626382272690535,
-        0.5930843949317932,
-        0.001174500328488648,
-        0.0017860672669485211,
-        0.0010054412996396422,
-        0.00012193901784485206,
-        -0.000334887852659449,
-        -0.0022029897663742304,
-        0.5935000777244568,
-        0.0012568456586450338,
-        0.0019742222502827644,
-        0.0011647609062492847,
-        0.00011021356476703659,
-        -0.00038211053470149636,
-        -0.0024467180483043194,
-        0.5931875109672546,
-        0.0013310370268300176,
-        0.0021642642095685005,
-        0.0013335321564227343,
-        0.00009949743980541825,
-        -0.0004296436673030257,
-        -0.0027013549115508795,
-        0.5931177139282227,
-        0.0014006041456013918,
-        0.002352928975597024,
-        0.0015152853447943926,
-        0.00008505478035658598,
-        -0.00048136088298633695,
-        -0.0029538467060774565,
-        0.5930479764938354,
-        0.0014608069323003292,
-        0.0025415997952222824,
-        0.001700855791568756,
-        0.00006727704749209806,
-        -0.0005341243813745677,
-        -0.003210614435374737,
-        0.5928036570549011,
-        0.0015100123127922416,
-        0.002731549320742488,
-        0.0018991812830790877,
-        0.0000493652478326112,
-        -0.0005809388239867985,
-        -0.003473253920674324,
-        0.5928736925125122,
-        0.0015514988917857409,
-        0.002922238316386938,
-        0.0021079820580780506,
-        0.000029167196771595627,
-        -0.0006329127354547381,
-        -0.0037316547241061926,
-        0.5928043127059937,
-        0.0015838317340239882,
-        0.0031066250521689653,
-        0.002319576218724251,
-        8.898059604689479e-6,
-        -0.0006800378323532641,
-        -0.004003639332950115,
-        0.5929091572761536,
-        0.0016050132689997554,
-        0.003295252099633217,
-        0.0025419394951313734,
-        -0.000013710295206692535,
-        -0.0007257868419401348,
-        -0.004269901663064957,
-        0.5930485725402832,
-        0.0016164487460628152,
-        0.0034825268667191267,
-        0.002774182241410017,
-        -0.000037491106922971085,
-        -0.0007724225870333612,
-        -0.004538912326097488,
-        0.5931875705718994,
-        0.001615947112441063,
-        0.003664180636405945,
-        0.0030124683398753405,
-        -0.00006199560448294505,
-        -0.0008142396691255271,
-        -0.00480045098811388,
-        0.5933974385261536,
-        0.0016095866449177265,
-        0.0038478909991681576,
-        0.003257976844906807,
-        -0.00008698971942067146,
-        -0.0008547022007405758,
-        -0.005078013986349106,
-        0.5936415791511536,
-        0.0015872354852035642,
-        0.004021089058369398,
-        0.0035109000746160746,
-        -0.00011568223999347538,
-        -0.0008942089625634253,
-        -0.005354971159249544,
-        0.5934317708015442,
-        0.0015591973206028342,
-        0.004198537208139896,
-        0.003773036180064082,
-        -0.00014495827781502157,
-        -0.0009303062106482685,
-        -0.005638875067234039,
-        0.5936412215232849,
-        0.0015177299501374364,
-        0.004366280045360327,
-        0.004040749743580818,
-        -0.00017588783521205187,
-        -0.0009663543896749616,
-        -0.005925025325268507,
-        0.593501627445221,
-        0.0014705799985677004,
-        0.004531881771981716,
-        0.00431715976446867,
-        -0.00021041395666543394,
-        -0.0010010612895712256,
-        -0.006207368802279234,
-        0.5935365557670593,
-        0.001413213089108467,
-        0.004692579619586468,
-        0.004599601496011019,
-        -0.0002457766968291253,
-        -0.0010327493073418736,
-        -0.006505406461656094,
-        0.5933269262313843,
-        0.0013434475986286998,
-        0.004848736338317394,
-        0.004890408832579851,
-        -0.00028711999766528606,
-        -0.00106858741492033,
-        -0.006807356141507626,
-        0.5931181311607361,
-        0.0012650334974750876,
-        0.004994421266019344,
-        0.0051898653618991375,
-        -0.00033172621624544263,
-        -0.0010990891605615616,
-        -0.0071100820787250996,
-        0.5929442048072815,
-        0.0011793962912634015,
-        0.005135250277817249,
-        0.005499166436493397,
-        -0.000378662021830678,
-        -0.0011317729949951172,
-        -0.007409243378788233,
-        0.5926647782325745,
-        0.0010838431771844625,
-        0.005268934182822704,
-        0.005817332770675421,
-        -0.0004281000583432615,
-        -0.001166283618658781,
-        -0.007717216853052378,
-        0.5926998853683472,
-        0.0009804241126403213,
-        0.0053992304019629955,
-        0.0061420174315571785,
-        -0.00048094423254951835,
-        -0.0011993126245215535,
-        -0.008030007593333721,
-        0.5924902558326721,
-        0.0008692154078744352,
-        0.005519451107829809,
-        0.006479023955762386,
-        -0.000537763989996165,
-        -0.001236212207004428,
-        -0.008356506004929543,
-        0.5929440855979919,
-        0.0007513284799642861,
-        0.00563538633286953,
-        0.006823732051998377,
-        -0.000594516284763813,
-        -0.0012742418330162764,
-        -0.008685515262186527,
-        0.5926643013954163,
-        0.0006232408923096955,
-        0.005743546411395073,
-        0.007178108207881451,
-        -0.0006553126149810851,
-        -0.0013130278093740344,
-        -0.00901535339653492,
-        0.5930123329162598,
-        0.00048613501712679863,
-        0.005843072663992643,
-        0.007545872125774622,
-        -0.0007216006051748991,
-        -0.001357274129986763,
-        -0.009348330087959766,
-        0.5926995277404785,
-        0.000340388563927263,
-        0.005928103346377611,
-        0.007921233773231506,
-        -0.0007906133541837335,
-        -0.0013983722310513258,
-        -0.00968977902084589,
-        0.5930134654045105,
-        0.00018408334290143102,
-        0.006012112833559513,
-        0.008309606462717056,
-        -0.0008611961966380477,
-        -0.00144277885556221,
-        -0.010034735314548016,
-        0.5933630466461182,
-        0.000022206833818927407,
-        0.0060895816422998905,
-        0.008708948269486427,
-        -0.0009344238787889481,
-        -0.0014912645565345883,
-        -0.01038195751607418,
-        0.5936756134033203,
-        -0.00014933485363144428,
-        0.006156634073704481,
-        0.009113290347158909,
-        -0.001009406871162355,
-        -0.0015364158898591995,
-        -0.01074444130063057,
-        0.5944079756736755,
-        -0.0003295359492767602,
-        0.006216940935701132,
-        0.009534457698464394,
-        -0.0010867133969441056,
-        -0.0015858380356803536,
-        -0.011105277575552464,
-        0.5946523547172546,
-        -0.0005184217588976026,
-        0.006267391610890627,
-        0.009968035854399204,
-        -0.00116726232226938,
-        -0.001635841210372746,
-        -0.011476362124085426,
-        0.5947223901748657,
-        -0.0007183448178693652,
-        0.006307759787887335,
-        0.010411876253783703,
-        -0.0012491614324972034,
-        -0.0016832650871947408,
-        -0.011858409270644188,
-        0.5945469737052917,
-        -0.000922650913707912,
-        0.006340802181512117,
-        0.01086830347776413,
-        -0.0013331895461305976,
-        -0.0017334959702566266,
-        -0.012237261049449444,
-        0.5951407551765442,
-        -0.001138067222200334,
-        0.006367202382534742,
-        0.011336571536958218,
-        -0.0014204677427187562,
-        -0.001783861662261188,
-        -0.012612935155630112,
-        0.5946868658065796,
-        -0.0013612991897389293,
-        0.006382528692483902,
-        0.011817813850939274,
-        -0.0015092657413333654,
-        -0.0018364684656262398,
-        -0.013003477826714516,
-        0.595176637172699,
-        -0.0015915771946310997,
-        0.0063857706263661385,
-        0.01231638714671135,
-        -0.0015994333662092686,
-        -0.0018854403169825673,
-        -0.013393816538155079,
-        0.5955926179885864,
-        -0.001827688654884696,
-        0.006383033003658056,
-        0.012827028520405293,
-        -0.0016901197377592325,
-        -0.0019397168653085828,
-        -0.01378911454230547,
-        0.5955935120582581,
-        -0.002072748262435198,
-        0.006370946299284697,
-        0.013353490270674229,
-        -0.0017832767916843295,
-        -0.001991107827052474,
-        -0.014198213815689087,
-        0.595174252986908,
-        -0.0023247504141181707,
-        0.006349846720695496,
-        0.013896841555833817,
-        -0.0018769301241263747,
-        -0.0020415729377418756,
-        -0.014593549072742462,
-        0.5950352549552917,
-        -0.0025832781102508307,
-        0.0063181049190461636,
-        0.01445536408573389,
-        -0.0019729691557586193,
-        -0.002093689749017358,
-        -0.015006774105131626,
-        0.5953840613365173
-      ],
-      "std": [
-        0.002411161782220006,
-        0.0035220407880842686,
-        0.0029013415332883596,
-        0.0013258246472105384,
-        0.0020179853308945894,
-        0.010011205449700356,
-        0.4910063147544861,
-        0.0048064119182527065,
-        0.007010283879935741,
-        0.005793850403279066,
-        0.0026270404923707247,
-        0.004016784951090813,
-        0.019959986209869385,
-        0.4910795986652374,
-        0.0072022695094347,
-        0.01047250535339117,
-        0.008696537464857101,
-        0.003923765849322081,
-        0.006003526505082846,
-        0.029928741976618767,
-        0.4911730885505676,
-        0.009601284749805927,
-        0.013944732025265694,
-        0.011596043594181538,
-        0.0051947347819805145,
-        0.007982160896062851,
-        0.039855923503637314,
-        0.49109989404678345,
-        0.011995380744338036,
-        0.017401276156306267,
-        0.01450654398649931,
-        0.006452825851738453,
-        0.009959585964679718,
-        0.049764279276132584,
-        0.4912789762020111,
-        0.014385105110704899,
-        0.020854491740465164,
-        0.017407990992069244,
-        0.007683087605983019,
-        0.011908459477126598,
-        0.059651929885149,
-        0.4912128448486328,
-        0.016765346750617027,
-        0.024284135550260544,
-        0.02030768245458603,
-        0.008882720954716206,
-        0.013833398930728436,
-        0.06945177167654037,
-        0.4911862313747406,
-        0.01913418062031269,
-        0.02769860439002514,
-        0.023200584575533867,
-        0.010069170035421848,
-        0.01573152281343937,
-        0.07920067012310028,
-        0.49123910069465637,
-        0.021495427936315536,
-        0.03108437918126583,
-        0.02608773671090603,
-        0.011218286119401455,
-        0.017600294202566147,
-        0.08888415992259979,
-        0.4912588894367218,
-        0.023848513141274452,
-        0.03446579724550247,
-        0.028954684734344482,
-        0.012335631065070629,
-        0.0194363035261631,
-        0.0984557494521141,
-        0.4911799430847168,
-        0.02618129923939705,
-        0.0377996489405632,
-        0.031812794506549835,
-        0.013426545076072216,
-        0.021240200847387314,
-        0.10793018341064453,
-        0.4912393391132355,
-        0.02850136160850525,
-        0.04113452509045601,
-        0.03465236350893974,
-        0.01448102854192257,
-        0.023004556074738503,
-        0.11730524897575378,
-        0.491252601146698,
-        0.030810443684458733,
-        0.0444553941488266,
-        0.03749171271920204,
-        0.01552011538296938,
-        0.02474648877978325,
-        0.12660986185073853,
-        0.49126577377319336,
-        0.033104438334703445,
-        0.047724947333335876,
-        0.04029623046517372,
-        0.01652422547340393,
-        0.026438293978571892,
-        0.1357518434524536,
-        0.4913119971752167,
-        0.03538680821657181,
-        0.05097673833370209,
-        0.04309060424566269,
-        0.017495229840278625,
-        0.02809188887476921,
-        0.14484061300754547,
-        0.49129876494407654,
-        0.037644777446985245,
-        0.054201558232307434,
-        0.045869018882513046,
-        0.018448922783136368,
-        0.029708286747336388,
-        0.1537874937057495,
-        0.4913118779659271,
-        0.03989279642701149,
-        0.05739012360572815,
-        0.04861682280898094,
-        0.019360067322850227,
-        0.031284283846616745,
-        0.16262821853160858,
-        0.4912920594215393,
-        0.04212509095668793,
-        0.060563746839761734,
-        0.05134649947285652,
-        0.020253626629710197,
-        0.03281664475798607,
-        0.17136116325855255,
-        0.4912656545639038,
-        0.04433578997850418,
-        0.06369910389184952,
-        0.05404900759458542,
-        0.021123351529240608,
-        0.03430699184536934,
-        0.1799684315919876,
-        0.4912393391132355,
-        0.0465271882712841,
-        0.06679350882768631,
-        0.05673054978251457,
-        0.02196338027715683,
-        0.035762034356594086,
-        0.1884494125843048,
-        0.49119946360588074,
-        0.04870201274752617,
-        0.0698690414428711,
-        0.0593833364546299,
-        0.02277866192162037,
-        0.03716905042529106,
-        0.19684050977230072,
-        0.4911530017852783,
-        0.05086364969611168,
-        0.0729081928730011,
-        0.06201274320483208,
-        0.023575762286782265,
-        0.03854267671704292,
-        0.20511800050735474,
-        0.49119293689727783,
-        0.052999626845121384,
-        0.07590420544147491,
-        0.06461741775274277,
-        0.024347545579075813,
-        0.03986750543117523,
-        0.21329139173030853,
-        0.4911530613899231,
-        0.05511622130870819,
-        0.07886126637458801,
-        0.06719071418046951,
-        0.02509898692369461,
-        0.04115770384669304,
-        0.22134073078632355,
-        0.4911796748638153,
-        0.05720451846718788,
-        0.0817820280790329,
-        0.06973543763160706,
-        0.02582353726029396,
-        0.04240873083472252,
-        0.22928737103939056,
-        0.49117299914360046,
-        0.05928165465593338,
-        0.08468781411647797,
-        0.07225947827100754,
-        0.026536913588643074,
-        0.0436241552233696,
-        0.23713935911655426,
-        0.4912128746509552,
-        0.06133345514535904,
-        0.08753105252981186,
-        0.07475195825099945,
-        0.02723076194524765,
-        0.04480902478098869,
-        0.2449100911617279,
-        0.49125248193740845,
-        0.06336378306150436,
-        0.09035338461399078,
-        0.07722371071577072,
-        0.027915935963392258,
-        0.04596531763672829,
-        0.25256893038749695,
-        0.49128544330596924,
-        0.06536917388439178,
-        0.0931331068277359,
-        0.07966616004705429,
-        0.028593653813004494,
-        0.04708719253540039,
-        0.26014769077301025,
-        0.49133822321891785,
-        0.06735322624444962,
-        0.09587426483631134,
-        0.08208146691322327,
-        0.029253413900732994,
-        0.04818820580840111,
-        0.2676284909248352,
-        0.4913316071033478,
-        0.06931550055742264,
-        0.09857368469238281,
-        0.084467314183712,
-        0.029913172125816345,
-        0.049258600920438766,
-        0.27503275871276855,
-        0.4913710951805115,
-        0.07124854624271393,
-        0.10124263167381287,
-        0.08683151751756668,
-        0.03056473471224308,
-        0.05031215026974678,
-        0.2823602259159088,
-        0.4912854731082916,
-        0.07316309213638306,
-        0.10387503355741501,
-        0.08917013555765152,
-        0.031205736100673676,
-        0.051339127123355865,
-        0.2896381616592407,
-        0.491338312625885,
-        0.0750511884689331,
-        0.10646932572126389,
-        0.09148410707712173,
-        0.03184282034635544,
-        0.05234614759683609,
-        0.2968275845050812,
-        0.49127253890037537,
-        0.07691479474306107,
-        0.10901817679405212,
-        0.0937764048576355,
-        0.032469242811203,
-        0.053335245698690414,
-        0.30398809909820557,
-        0.49133166670799255,
-        0.07874985039234161,
-        0.11152230948209763,
-        0.0960441529750824,
-        0.03308483585715294,
-        0.054303981363773346,
-        0.311068594455719,
-        0.49127233028411865,
-        0.08056764304637909,
-        0.11400332301855087,
-        0.09828827530145645,
-        0.03369200974702835,
-        0.055251531302928925,
-        0.31809186935424805,
-        0.49120599031448364,
-        0.08236116915941238,
-        0.11642979085445404,
-        0.10051009804010391,
-        0.034294869750738144,
-        0.05618305504322052,
-        0.32508960366249084,
-        0.4911465048789978,
-        0.0841296911239624,
-        0.11883242428302765,
-        0.10270429402589798,
-        0.034887079149484634,
-        0.0570959635078907,
-        0.3320097327232361,
-        0.4910062551498413,
-        0.08587083965539932,
-        0.12118621915578842,
-        0.10488012433052063,
-        0.0354686863720417,
-        0.05799192935228348,
-        0.3389015197753906,
-        0.49095919728279114,
-        0.08759026974439621,
-        0.12350363284349442,
-        0.10703437030315399,
-        0.03604672849178314,
-        0.05887560918927193,
-        0.3457629084587097,
-        0.4909456968307495,
-        0.08928582817316055,
-        0.12578748166561127,
-        0.10916976630687714,
-        0.03662176802754402,
-        0.059738170355558395,
-        0.35257115960121155,
-        0.49097952246665955,
-        0.09095519781112671,
-        0.1280231475830078,
-        0.11128115653991699,
-        0.03717926889657974,
-        0.06058768928050995,
-        0.35931187868118286,
-        0.4908647835254669,
-        0.09259875863790512,
-        0.13021688163280487,
-        0.11337121576070786,
-        0.03773583471775055,
-        0.061423756182193756,
-        0.3660118579864502,
-        0.4909525215625763,
-        0.09422165900468826,
-        0.13238278031349182,
-        0.11544124782085419,
-        0.03828861564397812,
-        0.06224775314331055,
-        0.3726641535758972,
-        0.4908578395843506,
-        0.09581982344388962,
-        0.13450181484222412,
-        0.11749458312988281,
-        0.03884110972285271,
-        0.06304997950792313,
-        0.3792511522769928,
-        0.49077698588371277,
-        0.09739767760038376,
-        0.13658517599105835,
-        0.11952484399080276,
-        0.03938090428709984,
-        0.06384975463151932,
-        0.3857972025871277,
-        0.49077683687210083,
-        0.09894638508558273,
-        0.1386304795742035,
-        0.12153798341751099,
-        0.03992167487740517,
-        0.06463739275932312,
-        0.39230042695999146,
-        0.4908582866191864,
-        0.10048258304595947,
-        0.14064013957977295,
-        0.12353785336017609,
-        0.040460873395204544,
-        0.06541262567043304,
-        0.398761510848999,
-        0.4908851981163025,
-        0.10199275612831116,
-        0.14262202382087708,
-        0.1255190521478653,
-        0.04099829122424126,
-        0.0661839172244072,
-        0.40516018867492676,
-        0.4908175766468048
-      ],
-      "q01": [
-        -0.007347712090983987,
-        -0.011486765748262403,
-        -0.005602361589856446,
-        -0.004156268547289073,
-        -0.007298904010653494,
-        -0.03456402559429407,
-        0.0,
-        -0.014532031904906033,
-        -0.023063078013062475,
-        -0.011146851527690887,
-        -0.008034631153941155,
-        -0.014402847960591318,
-        -0.0688410488665104,
-        0.0,
-        -0.02174613598883152,
-        -0.03454083656966686,
-        -0.016602809804677963,
-        -0.012004367487877608,
-        -0.02169839859008789,
-        -0.10323694312572479,
-        0.0,
-        -0.02924629563242197,
-        -0.046255945521593095,
-        -0.02208325160816312,
-        -0.01599733569249511,
-        -0.029179477728903294,
-        -0.13703150334358216,
-        0.0,
-        -0.0364012400791049,
-        -0.057891583660244945,
-        -0.027614226043969393,
-        -0.020049774289131163,
-        -0.03620416237413883,
-        -0.17225709611177442,
-        0.0,
-        -0.04366054475307465,
-        -0.06915460828691722,
-        -0.03313590524941683,
-        -0.023996917428076275,
-        -0.0432967299669981,
-        -0.20717854452133178,
-        0.0,
-        -0.0507191886767745,
-        -0.08091499265134335,
-        -0.038471172228455544,
-        -0.02769389592409134,
-        -0.05027697439491748,
-        -0.240767266279459,
-        0.0,
-        -0.05764442007690668,
-        -0.0928664004802704,
-        -0.04389051187932491,
-        -0.031378988534212116,
-        -0.05700045920312405,
-        -0.27346896691322325,
-        0.0,
-        -0.06481191511154175,
-        -0.10385700676739215,
-        -0.04937767976522446,
-        -0.034899483323097225,
-        -0.0640901755988598,
-        -0.3072322327375412,
-        0.0,
-        -0.07216909334659577,
-        -0.11465946695804596,
-        -0.05446739228963852,
-        -0.0384699240922928,
-        -0.07105188292860984,
-        -0.34030354830026627,
-        0.0,
-        -0.0791815688431263,
-        -0.12541235695779324,
-        -0.05973851401358843,
-        -0.04202334811687469,
-        -0.07781512894928456,
-        -0.37346914051771163,
-        0.0,
-        -0.08563443173170089,
-        -0.13601192309856414,
-        -0.06484265891909599,
-        -0.045433390045166,
-        -0.0843420141339302,
-        -0.40387091467380526,
-        0.0,
-        -0.09227466106414796,
-        -0.14718177989125253,
-        -0.06990853486061097,
-        -0.04866062937080862,
-        -0.09135572135448455,
-        -0.43357558735609053,
-        0.0,
-        -0.09871668579578399,
-        -0.15755651119351388,
-        -0.07494964042305946,
-        -0.05228844651579856,
-        -0.0973639042377472,
-        -0.4662614274024963,
-        0.0,
-        -0.10474277555942535,
-        -0.16660683050751687,
-        -0.07973185749202967,
-        -0.055503451734781245,
-        -0.1043376196026802,
-        -0.49795184282064436,
-        0.0,
-        -0.11146926714479923,
-        -0.17675033633708956,
-        -0.08481720062792301,
-        -0.05867235680818558,
-        -0.11146564957499505,
-        -0.5234594867229462,
-        0.0,
-        -0.1183029289394617,
-        -0.18591568176746368,
-        -0.08976946114003659,
-        -0.06189141765832901,
-        -0.11618306014537812,
-        -0.5559864712238312,
-        0.0,
-        -0.1254465562880039,
-        -0.1967654939889908,
-        -0.09452381261885166,
-        -0.06458974866867065,
-        -0.12229588437080384,
-        -0.5838854856491089,
-        0.0,
-        -0.13160774660110475,
-        -0.20561206495761872,
-        -0.09930612608790398,
-        -0.06763142464160918,
-        -0.12729683189988136,
-        -0.6118898140192032,
-        0.0,
-        -0.13866937249898909,
-        -0.21400038709640504,
-        -0.10383093198537827,
-        -0.07044867560863496,
-        -0.1330213810920715,
-        -0.6420615591049195,
-        0.0,
-        -0.14518664595484734,
-        -0.22208737790584565,
-        -0.10839852929115296,
-        -0.07363164942264558,
-        -0.13909859127998353,
-        -0.6659443435192108,
-        0.0,
-        -0.15202519185543062,
-        -0.23166744391918181,
-        -0.1132360664010048,
-        -0.07554293347597121,
-        -0.1431354057431221,
-        -0.6935547601938247,
-        0.0,
-        -0.1575157383620739,
-        -0.24115877079963685,
-        -0.11769197665154933,
-        -0.07880985802412033,
-        -0.14886759107112885,
-        -0.717457844042778,
-        0.0,
-        -0.1637449370443821,
-        -0.25002506982684136,
-        -0.12177611397504806,
-        -0.08111752626895904,
-        -0.15336844004392625,
-        -0.7450204710006714,
-        0.0,
-        -0.1702753342092037,
-        -0.25778238211274146,
-        -0.12590762196183203,
-        -0.08350524621009828,
-        -0.15739910807609558,
-        -0.7672739201784134,
-        0.0,
-        -0.17570297884345054,
-        -0.26699419630765914,
-        -0.12999144119024275,
-        -0.08586254574060437,
-        -0.15982728340625763,
-        -0.7919603824377061,
-        0.0,
-        -0.18209079703688621,
-        -0.2762185108423233,
-        -0.1341805691242218,
-        -0.08846430575847625,
-        -0.1640431535243988,
-        -0.8073872842788696,
-        0.0,
-        -0.18759027983546256,
-        -0.28415180140137675,
-        -0.13808169324100017,
-        -0.09098501863479613,
-        -0.1681320061683655,
-        -0.8263913946390152,
-        0.0,
-        -0.193171888589859,
-        -0.29068409420847896,
-        -0.14178944163620472,
-        -0.09304863989353179,
-        -0.1707449821949005,
-        -0.8507119262695311,
-        0.0,
-        -0.19844869979619978,
-        -0.2987963397145271,
-        -0.14549240586161613,
-        -0.09557797043323518,
-        -0.1739663097858429,
-        -0.8648041958332062,
-        0.0,
-        -0.20466184456944464,
-        -0.3049391875267029,
-        -0.14948453903496267,
-        -0.09817119854688644,
-        -0.17706590107083323,
-        -0.8849965750694275,
-        0.0,
-        -0.20983124628067015,
-        -0.3116496807098389,
-        -0.15297152189016341,
-        -0.10035031213760376,
-        -0.18140975254774094,
-        -0.9072247592926024,
-        0.0,
-        -0.215193039560318,
-        -0.31998531384468076,
-        -0.1565369277715683,
-        -0.102377852499485,
-        -0.1837332302033901,
-        -0.9293165444850922,
-        0.0,
-        -0.22114226788282393,
-        -0.32775930384397506,
-        -0.16037838854789732,
-        -0.10488562031984328,
-        -0.18766805601119993,
-        -0.9526331702470778,
-        0.0,
-        -0.22637617546319963,
-        -0.33561479420661927,
-        -0.16403944179415703,
-        -0.10686019870638847,
-        -0.19061433601379393,
-        -0.9705972829818725,
-        0.0,
-        -0.23192334132194517,
-        -0.3404881935119629,
-        -0.16749516949653626,
-        -0.10840021744370465,
-        -0.19327300127148628,
-        -0.9914580883026123,
-        0.0,
-        -0.23655534625053407,
-        -0.3483747916221619,
-        -0.1713618674516678,
-        -0.11093877178430556,
-        -0.19658433060646058,
-        -1.0074259456634522,
-        0.0,
-        -0.24189636796712874,
-        -0.35468626945018766,
-        -0.174676346886158,
-        -0.1137608123421669,
-        -0.1998320359289646,
-        -1.02401462726593,
-        0.0,
-        -0.24619000346064568,
-        -0.3617886746406555,
-        -0.17737202043533326,
-        -0.11608646607995032,
-        -0.20212158007621764,
-        -1.0396305271863937,
-        0.0,
-        -0.25069174761176105,
-        -0.3690223491430283,
-        -0.18019738556742668,
-        -0.11791741763353347,
-        -0.20542705215215684,
-        -1.056211252117157,
-        0.0,
-        -0.25575273798704146,
-        -0.37492762922048567,
-        -0.18413128078579902,
-        -0.11963063821196557,
-        -0.2084871412396431,
-        -1.0777596963644027,
-        0.0,
-        -0.26051444165706633,
-        -0.38124460592269893,
-        -0.18717371021509172,
-        -0.12171284379959108,
-        -0.21154849707484244,
-        -1.0918209570646285,
-        0.0,
-        -0.2645109035611153,
-        -0.38571969106197357,
-        -0.19055300951004028,
-        -0.12364264481067655,
-        -0.21497644165158272,
-        -1.1041330133199692,
-        0.0,
-        -0.26856355381011965,
-        -0.3920482913136482,
-        -0.19341499418020247,
-        -0.12596707943677904,
-        -0.21652717173099517,
-        -1.1284716510772705,
-        0.0,
-        -0.273536373257637,
-        -0.3985861983895302,
-        -0.19648525762557983,
-        -0.12863396686315537,
-        -0.2178426690995693,
-        -1.1498000781059265,
-        0.0,
-        -0.2773340993881226,
-        -0.4043089840650559,
-        -0.19916273007392882,
-        -0.1301111794948578,
-        -0.21953763974308968,
-        -1.1649874792098998,
-        0.0,
-        -0.28152843511104586,
-        -0.4105489842891693,
-        -0.20206411379575728,
-        -0.13238328326940535,
-        -0.22030045819282532,
-        -1.186898805141449,
-        0.0,
-        -0.28521604499816894,
-        -0.41719991471767426,
-        -0.20488486919403076,
-        -0.13367428585290908,
-        -0.2215240351676941,
-        -1.2064387265205383,
-        0.0,
-        -0.28908531114459035,
-        -0.4219293355822563,
-        -0.20779289478063584,
-        -0.1346039095699787,
-        -0.22363287283182146,
-        -1.225643195438385,
-        0.0,
-        -0.294126475507021,
-        -0.42881255321502687,
-        -0.21011174390912055,
-        -0.13763879417181013,
-        -0.22653749623298647,
-        -1.2400950877666475,
-        0.0
-      ],
-      "q99": [
-        0.00760505946278572,
-        0.01227881843298674,
-        0.009394833338446915,
-        0.0041071553006768215,
-        0.005846777586638929,
-        0.03394978983253241,
-        0.9998,
-        0.015092343334481124,
-        0.024545711845159537,
-        0.018547782172262668,
-        0.008136489582061764,
-        0.011534944232553239,
-        0.0680388870239258,
-        0.9998,
-        0.022567954233288764,
-        0.03644986734837294,
-        0.0278723166257143,
-        0.012022978414595133,
-        0.017261751674115658,
-        0.1018139317035675,
-        0.9998,
-        0.030176736226677894,
-        0.048319679215550415,
-        0.037043299572169786,
-        0.015823813125491137,
-        0.022935147598385802,
-        0.13534477652311327,
-        0.9998,
-        0.03749606138020754,
-        0.05974762381315232,
-        0.046320767011493444,
-        0.01970605190098286,
-        0.028365900403261188,
-        0.16783747158050544,
-        0.9998,
-        0.04486110335588456,
-        0.07147922609001398,
-        0.055334820427000525,
-        0.023239636448025694,
-        0.034125556316971783,
-        0.2017985020875931,
-        0.9998,
-        0.052173979285359384,
-        0.08346393676698208,
-        0.06446339850723744,
-        0.026847184759378445,
-        0.03955972051620485,
-        0.23437213500142096,
-        0.9998,
-        0.059150983133912094,
-        0.09537746536731717,
-        0.0733001764550805,
-        0.029705484813451755,
-        0.04516481494158506,
-        0.26646220774650575,
-        0.9998,
-        0.06645039129257202,
-        0.1076395375341177,
-        0.08239804291725158,
-        0.0328832214474678,
-        0.04995994846820831,
-        0.29610841860771175,
-        0.9998,
-        0.07374503636360169,
-        0.11985380845069887,
-        0.09145532561540604,
-        0.035987181842327104,
-        0.05490816103816032,
-        0.3266934569597244,
-        0.9998,
-        0.08122003978192807,
-        0.1319003387391567,
-        0.10024555467069149,
-        0.038626870799064655,
-        0.05953236908912657,
-        0.35580547877550117,
-        0.9998,
-        0.08822669934630395,
-        0.14304091303348543,
-        0.1088144795835018,
-        0.041924731212854394,
-        0.06486271733045579,
-        0.3876368781208992,
-        0.9998,
-        0.09569223463535309,
-        0.1534084576368332,
-        0.11751553359031677,
-        0.04469002693891522,
-        0.06991170048713685,
-        0.41592443135976787,
-        0.9998,
-        0.10324303380250932,
-        0.1640547949969769,
-        0.1258540060549974,
-        0.04742067799866201,
-        0.0741334307938814,
-        0.4491768553853035,
-        0.9998,
-        0.1099683414697647,
-        0.17510256106853483,
-        0.13403582906574008,
-        0.050331409102678304,
-        0.07791369587183,
-        0.475221610414982,
-        0.9998,
-        0.1170920331031084,
-        0.1857688842058181,
-        0.14203765797317025,
-        0.05343856880664827,
-        0.0820521941632032,
-        0.5052882493734359,
-        0.9998,
-        0.12414569870531561,
-        0.19753187570571906,
-        0.14974090793728828,
-        0.0557895161241293,
-        0.0865580612182617,
-        0.5326116788625717,
-        0.9998,
-        0.13112460071146492,
-        0.20844777574539186,
-        0.15755365291237833,
-        0.05866813796758652,
-        0.09100452079772947,
-        0.5617906215190887,
-        0.9998,
-        0.1379249924659729,
-        0.21917983024120336,
-        0.16431246396899224,
-        0.06069063851833345,
-        0.09500974670350554,
-        0.5912433532238006,
-        0.9998,
-        0.14485807346105578,
-        0.2286938194990158,
-        0.1716713245987892,
-        0.0637721403002739,
-        0.09919737915396692,
-        0.6155789003372192,
-        0.9998,
-        0.15117210847139356,
-        0.23827072948217387,
-        0.1779224501848221,
-        0.06597466281652448,
-        0.10370546174049378,
-        0.644374561071396,
-        0.9998,
-        0.15787530031800268,
-        0.24751081448793422,
-        0.18454510100185867,
-        0.06831172816753389,
-        0.10792329218983648,
-        0.6659029854536058,
-        0.9998,
-        0.1641180765330792,
-        0.2572556215524673,
-        0.19053607739508155,
-        0.07030175954103468,
-        0.11277426195144652,
-        0.686683092904091,
-        0.9998,
-        0.16966928005814552,
-        0.26658033823966976,
-        0.19653976464271544,
-        0.07203448786735533,
-        0.11643967783451081,
-        0.7125654235839844,
-        0.9998,
-        0.17621982062458996,
-        0.27701201248764995,
-        0.20197029424905777,
-        0.07416592988967896,
-        0.12035323338508608,
-        0.7360334300994871,
-        0.9998,
-        0.1819627068042755,
-        0.2855635068178177,
-        0.2084233057498932,
-        0.07603480890989306,
-        0.1241462073802948,
-        0.7568318559646605,
-        0.9998,
-        0.18748426148295405,
-        0.29386607773303985,
-        0.21488302874565124,
-        0.0777944794893265,
-        0.1279024946689606,
-        0.7800548309087754,
-        0.9998,
-        0.19402539111375805,
-        0.304579441523552,
-        0.22076346032619476,
-        0.07946800847053526,
-        0.13082660132646562,
-        0.8035489074707032,
-        0.9998,
-        0.1999460756778717,
-        0.31222576675415037,
-        0.2264517953634262,
-        0.08081280782818795,
-        0.13460262061357497,
-        0.828994060134888,
-        0.9998,
-        0.20509765752553943,
-        0.3203624894678593,
-        0.23198571143150332,
-        0.08211237077713013,
-        0.13750549141168594,
-        0.8529169931411746,
-        0.9998,
-        0.209518383550644,
-        0.3281992249488831,
-        0.23816398755311968,
-        0.08322777889966959,
-        0.14037072820067403,
-        0.8746512705802916,
-        0.9998,
-        0.21442690343856818,
-        0.33666496825218195,
-        0.24309153485894203,
-        0.08529254732131952,
-        0.14304055777192115,
-        0.8924013189554216,
-        0.9998,
-        0.21941403126716608,
-        0.3448349338173866,
-        0.2489651171147823,
-        0.08667572829127312,
-        0.14580873468518263,
-        0.9107246013164523,
-        0.9998,
-        0.22309734332561493,
-        0.3517628137350083,
-        0.2547237434387207,
-        0.08896696496009826,
-        0.1489701636433602,
-        0.9272562242746352,
-        0.9998,
-        0.22793451219797134,
-        0.3591047751426697,
-        0.2597683270871639,
-        0.09042025351524352,
-        0.1528471918106079,
-        0.9483436929702762,
-        0.9998,
-        0.23258997063636777,
-        0.36902411174774175,
-        0.26485966441631315,
-        0.09243017539381981,
-        0.15506044900417326,
-        0.9667170533418656,
-        0.9998,
-        0.23665347018241878,
-        0.3762736029624939,
-        0.27067578415870663,
-        0.09449174585938458,
-        0.15888449568748475,
-        0.9834513599395751,
-        0.9998,
-        0.24093078261613854,
-        0.3843366762638093,
-        0.2768812180876732,
-        0.09594143676757816,
-        0.16144601993560792,
-        1.0022019856452942,
-        0.9998,
-        0.2459272434949874,
-        0.3907656828403474,
-        0.2815030888676643,
-        0.09760697330236434,
-        0.16464969681501385,
-        1.019599228835106,
-        0.9998,
-        0.2511993800699711,
-        0.3962667070865631,
-        0.28746061475872997,
-        0.09892506269812584,
-        0.1685606077313423,
-        1.0345355377674106,
-        0.9998,
-        0.25586065447330475,
-        0.4064154948949814,
-        0.29352941393256193,
-        0.10010581925511364,
-        0.17137197246551505,
-        1.0455808718204502,
-        0.9998,
-        0.26066332874298104,
-        0.41295039305686965,
-        0.29917095521688464,
-        0.10042237902879714,
-        0.17334867343306548,
-        1.0644763714551928,
-        0.9998,
-        0.26424761767387395,
-        0.4171489597439766,
-        0.3047941813468933,
-        0.10030385837554934,
-        0.1771283437609672,
-        1.0775494430065153,
-        0.9998,
-        0.266823394203186,
-        0.42120621981620787,
-        0.31085526600480073,
-        0.10133219141960148,
-        0.18098006631135943,
-        1.0927770781517028,
-        0.9998,
-        0.27192984950542454,
-        0.4258830359578132,
-        0.31648782253265384,
-        0.10210902032852165,
-        0.18391375703215596,
-        1.1100240562438963,
-        0.9998,
-        0.2774573613643646,
-        0.4328473631143568,
-        0.3229170823454858,
-        0.10328457579612738,
-        0.18559744659066202,
-        1.127002547836304,
-        0.9998,
-        0.280809062719345,
-        0.43832696628570567,
-        0.32813225684762004,
-        0.10458858071565635,
-        0.18929516908526423,
-        1.1399588096141815,
-        0.9998,
-        0.28525815182924275,
-        0.44289318814277645,
-        0.3333625353813172,
-        0.1055313608527183,
-        0.1918839191913605,
-        1.1545847342491151,
-        0.9998,
-        0.2899247071444988,
-        0.4499819297909735,
-        0.33962785017490393,
-        0.10614367849826811,
-        0.19452174307107928,
-        1.1747124679088592,
-        0.9998,
-        0.29274274289608004,
-        0.4549682779312134,
-        0.3463232246458531,
-        0.10680567473173141,
-        0.19644608254432672,
-        1.185605804681778,
-        0.9998
-      ]
-    }
-  },
-  "sample_ratio": 0.05,
-  "dataset_infos": {
-    "robotgeneralist/pp__1": {
-      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
-      "num_episodes": 237,
-      "num_frames": 99363
-    },
-    "robotgeneralist/pp__1__adversarial": {
-      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
-      "num_episodes": 64,
-      "num_frames": 17906
-    },
-    "robotgeneralist/pp__2": {
-      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
-      "num_episodes": 92,
-      "num_frames": 37458
-    },
-    "robotgeneralist/pp__2__rotations": {
-      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
-      "num_episodes": 102,
-      "num_frames": 13935
-    },
-    "robotgeneralist/pp__3__with_ft": {
-      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
-      "num_episodes": 99,
-      "num_frames": 38424
-    },
-    "robotgeneralist/pp__3__with_ft__rotations": {
-      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
-      "num_episodes": 49,
-      "num_frames": 7151
-    },
-    "robotgeneralist/pp__3__with_ft__adversarial_start": {
-      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
-      "num_episodes": 147,
-      "num_frames": 21313
-    },
-    "robotgeneralist/pp__4__with_ft": {
-      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
-      "num_episodes": 94,
-      "num_frames": 33046
-    },
-    "robotgeneralist/pp__5__with_ft__adversarial_target": {
-      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
-      "num_episodes": 48,
-      "num_frames": 10615
-    },
-    "robotgeneralist/pp__6__with_ft": {
-      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
-      "num_episodes": 379,
-      "num_frames": 123611
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-1": {
-      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
-      "num_episodes": 98,
-      "num_frames": 41410
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-2": {
-      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
-      "num_episodes": 94,
-      "num_frames": 37859
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-3": {
-      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
-      "num_episodes": 127,
-      "num_frames": 53341
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-4": {
-      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
-      "num_episodes": 102,
-      "num_frames": 38263
-    }
-  }
-}
\ No newline at end of file
diff --git a/pp__ee_baseline/20000/params/_METADATA b/pp__ee_baseline/20000/params/_METADATA
deleted file mode 100644
index 9eb73624158f4514d49041e0806d2d702c62eee1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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, "write_shape": [1152]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304]}}, "('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, "write_shape": [27, 1152, 4304]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 16, 72, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [14, 14, 3, 1152]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1152, 2048]}}, "('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, "write_shape": [1, 256, 1152]}}, "('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, "write_shape": [257152, 2048]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [18, 8, 256, 2048]}}, "('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, "write_shape": [18, 8, 256, 1024]}}, "('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, "write_shape": [18, 2, 1, 2048, 256]}}, "('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, "write_shape": [18, 2, 1, 1024, 256]}}, "('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, "write_shape": [18, 8, 2048, 256]}}, "('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, "write_shape": [18, 8, 1024, 256]}}, "('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, "write_shape": [18, 2, 2048, 16384]}}, "('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, "write_shape": [18, 16384, 2048]}}, "('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, "write_shape": [18, 2, 1024, 4096]}}, "('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, "write_shape": [18, 4096, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}, "('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, "write_shape": [32]}}, "('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, "write_shape": [1024, 32]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [2048, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [1024, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/pp__ee_baseline/20000/params/_sharding b/pp__ee_baseline/20000/params/_sharding
deleted file mode 100644
index 924608abdcb99abb3c864bb9d1bfca6eb60f780e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/_sharding
+++ /dev/null
@@ -1 +0,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}]]}}"}
\ No newline at end of file
diff --git a/pp__ee_baseline/20000/params/array_metadatas/process_0 b/pp__ee_baseline/20000/params/array_metadatas/process_0
deleted file mode 100644
index 6aa9fd1decf72607133eb54664d98802bf563b5c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/array_metadatas/process_0
+++ /dev/null
@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/pp__ee_baseline/20000/params/d/4a16bad6bcafa0bc0412c8503541ce2f b/pp__ee_baseline/20000/params/d/4a16bad6bcafa0bc0412c8503541ce2f
deleted file mode 100644
index 5b6812bba489d191c0dccfeb7662810f197b8af5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/d/4a16bad6bcafa0bc0412c8503541ce2f and /dev/null differ
diff --git a/pp__ee_baseline/20000/params/manifest.ocdbt b/pp__ee_baseline/20000/params/manifest.ocdbt
deleted file mode 100644
index b407920a59dc600dca5a988df2f33221b9f7145a..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/078c08346d0aee3a9296bad5f29cfe88 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/078c08346d0aee3a9296bad5f29cfe88
deleted file mode 100644
index 2c94273dbfcd49c52cd39ee3ed835e902539281e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/078c08346d0aee3a9296bad5f29cfe88
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0873f86bb0c8726187b6ec2a93941c3d84a30fcd17b60818d7b740a0a0dd5506
-size 132963704
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/240266fe43b47dd3259d5032d9d92eda b/pp__ee_baseline/20000/params/ocdbt.process_0/d/240266fe43b47dd3259d5032d9d92eda
deleted file mode 100644
index bed854177f8b2ae66468b7e57da529bd12044689..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/ocdbt.process_0/d/240266fe43b47dd3259d5032d9d92eda and /dev/null differ
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/303318973d22eb0047e30dfc5b30e318 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/303318973d22eb0047e30dfc5b30e318
deleted file mode 100644
index 7d364417e6b07b3d2dcb6f5e1b6a4ab68e1121f1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/303318973d22eb0047e30dfc5b30e318
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ed61e98737f4888e6437d15ac811d797e4e4c1742371abcadff17b1372d320bc
-size 1120134203
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/434e40c620fe7c87cb4ccb72015d820b b/pp__ee_baseline/20000/params/ocdbt.process_0/d/434e40c620fe7c87cb4ccb72015d820b
deleted file mode 100644
index b6e76ab27e979eb474fdcd2786c25b67d64c0d99..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/434e40c620fe7c87cb4ccb72015d820b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a0fc15372640e1f4bd57922d592c79f0e966c2efa3fdefea3ac053c1bf3f6dee
-size 1120118462
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/4dc7be82ad118b2dd86ea25168fe6657 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/4dc7be82ad118b2dd86ea25168fe6657
deleted file mode 100644
index bf192773ffe8f3fdb66e44ac7fecd0d58b6d8c09..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/4dc7be82ad118b2dd86ea25168fe6657
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c2b3f97c722b61bb8a66c19cd7caf76ca6a7868a18be6202e7cd430478eca055
-size 25921638
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/4e92e2c62f917f52ea55cad280adf721 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/4e92e2c62f917f52ea55cad280adf721
deleted file mode 100644
index f19daeb6ee4088ef0df2f9d18918a38cf9a62105..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/4e92e2c62f917f52ea55cad280adf721
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:87f1fc6071c60f28d17cadae91dcab8e8c865e21f047dc4e23b4439bde0785ab
-size 34975691
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/5d86f11fa8bb76d0ea5b320fa2cb5620 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/5d86f11fa8bb76d0ea5b320fa2cb5620
deleted file mode 100644
index 30c3c1726bfa3e956696329169f44aa97e526735..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/5d86f11fa8bb76d0ea5b320fa2cb5620
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ceb2ccbb7f28d8e790304d2cfd8f3219e987246dc788709bf4a43cfae9e95ee5
-size 1056753487
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/5f1e05a82cc4d12c2d4d6e22a7da3fb2 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/5f1e05a82cc4d12c2d4d6e22a7da3fb2
deleted file mode 100644
index e300f934295512fef8a6d250fb488c3867758597..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/5f1e05a82cc4d12c2d4d6e22a7da3fb2
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:fb7f26b601475821ea2f784ee30f33251b65c48854449669ae5b8813cc8bd234
-size 70084476
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/649a9aa033541492ae99d4e3c3bed1e8 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/649a9aa033541492ae99d4e3c3bed1e8
deleted file mode 100644
index 5db456dd087aef9662b6a4101797d4d7f1814875..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/649a9aa033541492ae99d4e3c3bed1e8
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:66113a71e0e1caf1ae633b94e28ec9527455b02056ddf60c9ebb7c5022cf3a02
-size 1957442645
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/76f860dc8e46889c98c8488abed796f1 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/76f860dc8e46889c98c8488abed796f1
deleted file mode 100644
index 767969e7198aeed7748976669a0296a26abf1441..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/ocdbt.process_0/d/76f860dc8e46889c98c8488abed796f1 and /dev/null differ
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7815bab6729ca574519c4795f47c2399 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/7815bab6729ca574519c4795f47c2399
deleted file mode 100644
index a33231c11c29c7534b5b8cd3bbbc6fe63b897232..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7815bab6729ca574519c4795f47c2399
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:cfe4ed621d8128147c572fbbfcc003469d779ea3f1999b773894b2b1f53a1d00
-size 496755764
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/78db8a77fe5134106bc526808a1aab6a b/pp__ee_baseline/20000/params/ocdbt.process_0/d/78db8a77fe5134106bc526808a1aab6a
deleted file mode 100644
index cdfb6cebdff77b4a036729c92b5b3bfc878c1e48..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/78db8a77fe5134106bc526808a1aab6a
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9dbedbb3e1116d3a65a81438300a639827222361cf222918fdfbd579c7c67d6d
-size 272937963
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7bc030cd0bd6639a71eeb9ef562f618c b/pp__ee_baseline/20000/params/ocdbt.process_0/d/7bc030cd0bd6639a71eeb9ef562f618c
deleted file mode 100644
index 3cd0a2ffed92d3dbbe647e95a1369e0cbc569d05..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7bc030cd0bd6639a71eeb9ef562f618c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9014aa909dbdb588830c03791a24044c1306a3b27ee19d819a16164bae22ee2d
-size 272760410
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d266ba1935cd38cbef3082817900922 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d266ba1935cd38cbef3082817900922
deleted file mode 100644
index 647bfad48e6c63130ec040301c68674e91c41fd2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d266ba1935cd38cbef3082817900922
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:10accddcb760c6f0b5de4fb6879424b93775cbe5cd061f775ed0426abf11b809
-size 132939065
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d6d955df445be9f4d082c1c6e911319 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d6d955df445be9f4d082c1c6e911319
deleted file mode 100644
index 1eef16eb96728553f46d319fa819a1faf392ca2d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/7d6d955df445be9f4d082c1c6e911319
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a9a68982f8dc05135266b3b99229d9b7feca492002d226b62cd2187ed5d37e35
-size 280186589
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/9d96e8e7288d2766acd570176aa163e4 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/9d96e8e7288d2766acd570176aa163e4
deleted file mode 100644
index d6cc5c18d0461d9303e3521b91947cc08f843d89..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/9d96e8e7288d2766acd570176aa163e4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f6fcf18c1c7489ddbc7c3035e9b5e777765d29a9f54d088dbbaf02c94ef948f5
-size 377989
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/aa4fd585b36c1dd41609a9575c151027 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/aa4fd585b36c1dd41609a9575c151027
deleted file mode 100644
index 52b856f94c3f1e599e0e4e5367a06f279ae82d0b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/aa4fd585b36c1dd41609a9575c151027
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7af224a692eeb7479434872a2b3f8b6595c85ab13da39cf1f5757742f8e831d9
-size 1120106528
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5772fe9025a6db70f60003b76287619 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5772fe9025a6db70f60003b76287619
deleted file mode 100644
index f80d528572ba1504668dc3b914b8222193cf94e0..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5772fe9025a6db70f60003b76287619
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8f3c44cdd728e1db596fd687ee0f5212c1022a1e16139a0b6a2d7a9375c41825
-size 279249554
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5a2e26ec9382ad81e5acf204808420d b/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5a2e26ec9382ad81e5acf204808420d
deleted file mode 100644
index 2c4ba078a69033f1cd9056feec517195dc5f82dc..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/b5a2e26ec9382ad81e5acf204808420d
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c85253f92f62a05c68a46ab59b4494d3fb79c1fa03fe25e37284a57054f6fda5
-size 280037512
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/d1a146b96e6110eefcf039218eef474c b/pp__ee_baseline/20000/params/ocdbt.process_0/d/d1a146b96e6110eefcf039218eef474c
deleted file mode 100644
index f4040ce0d33575cc2fe3b7e0bf9a7a811c938395..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/d1a146b96e6110eefcf039218eef474c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bd02322d4a89485e7e0de15518c47218af69883d68748d13414fc21e69c2942b
-size 2240242301
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/dc0258ac93a056ac7ac9dee5b0bd77b3 b/pp__ee_baseline/20000/params/ocdbt.process_0/d/dc0258ac93a056ac7ac9dee5b0bd77b3
deleted file mode 100644
index 139f8bc6210514b8aeee95d5e304615a78fc6b0d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/20000/params/ocdbt.process_0/d/dc0258ac93a056ac7ac9dee5b0bd77b3
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:456dac28f3e5bf0eac1c2731ce2e38eeb0dda35ad51a2c2c4da24ef67bb85afc
-size 1120111421
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/d/e59e5689c47561586e9b7ea70d2c24ee b/pp__ee_baseline/20000/params/ocdbt.process_0/d/e59e5689c47561586e9b7ea70d2c24ee
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/ocdbt.process_0/d/e59e5689c47561586e9b7ea70d2c24ee and /dev/null differ
diff --git a/pp__ee_baseline/20000/params/ocdbt.process_0/manifest.ocdbt b/pp__ee_baseline/20000/params/ocdbt.process_0/manifest.ocdbt
deleted file mode 100644
index e19d29263389bfc7b6c19ce713c5974f5a735b5c..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/20000/params/ocdbt.process_0/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/25000/_CHECKPOINT_METADATA b/pp__ee_baseline/25000/_CHECKPOINT_METADATA
deleted file mode 100644
index 48018c9ba3d35528fbfe67ac1341ecd2889eaab4..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/_CHECKPOINT_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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": 1767242906585253896, "commit_timestamp_nsecs": 1767242950828151423, "custom_metadata": {}}
\ No newline at end of file
diff --git a/pp__ee_baseline/25000/assets/ur10e/norm_stats.json b/pp__ee_baseline/25000/assets/ur10e/norm_stats.json
deleted file mode 100644
index 8a1b886d6b668da778297b91b3d82381c2eab28a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/assets/ur10e/norm_stats.json
+++ /dev/null
@@ -1,1525 +0,0 @@
-{
-  "norm_stats": {
-    "state": {
-      "mean": [
-        0.05313729867339134,
-        0.0037223484832793474,
-        0.106409952044487,
-        0.0008666226058267057,
-        -0.00854077935218811,
-        0.01460699550807476,
-        0.5940247178077698
-      ],
-      "std": [
-        0.1176244392991066,
-        0.16962096095085144,
-        0.07773701101541519,
-        0.04537155479192734,
-        0.06616315245628357,
-        0.6038344502449036,
-        0.49107977747917175
-      ],
-      "q01": [
-        -0.2025573879778385,
-        -0.31106759839057924,
-        0.00890737680345774,
-        -0.11727181684970855,
-        -0.16711108471155167,
-        -1.423702568435669,
-        0.0
-      ],
-      "q99": [
-        0.29901714037060734,
-        0.4115018085956573,
-        0.3217185490995646,
-        0.13774808425903318,
-        0.19241279754638663,
-        1.598234023094177,
-        0.9998
-      ]
-    },
-    "actions": {
-      "mean": [
-        0.0001451610296498984,
-        0.00017207347264047712,
-        0.000048651279939804226,
-        0.00003654604734038003,
-        -0.000019767590856645256,
-        -0.00019861181499436498,
-        0.5944076776504517,
-        0.0002841090608853847,
-        0.00034722191048786044,
-        0.00011135452223243192,
-        0.00006827477045590058,
-        -0.000043589698179857805,
-        -0.0004028017574455589,
-        0.594025731086731,
-        0.00041486899135634303,
-        0.0005213047261349857,
-        0.00018569435633253306,
-        0.00009245114779332653,
-        -0.00007049143459880725,
-        -0.000616009347140789,
-        0.5935360789299011,
-        0.000542119552846998,
-        0.0006975074065849185,
-        0.00027102616149932146,
-        0.00011072693450842053,
-        -0.00010138262587133795,
-        -0.0008296052110381424,
-        0.593919575214386,
-        0.0006618681945838034,
-        0.0008754940354265273,
-        0.0003684645926114172,
-        0.00012383276771288365,
-        -0.000135598995257169,
-        -0.0010419095633551478,
-        0.5929783582687378,
-        0.0007787057547830045,
-        0.00105847988743335,
-        0.0004721089790109545,
-        0.00013335929543245584,
-        -0.00017038211808539927,
-        -0.0012720258673653007,
-        0.5933270454406738,
-        0.0008861633250489831,
-        0.001237610587850213,
-        0.0005911999032832682,
-        0.00013674379442818463,
-        -0.00020881379896309227,
-        -0.0014943619025871158,
-        0.5934669971466064,
-        0.0009874902898445725,
-        0.0014197468990460038,
-        0.0007170690805651248,
-        0.00013581194798462093,
-        -0.0002465852303430438,
-        -0.0017284756759181619,
-        0.5931888818740845,
-        0.0010826619109138846,
-        0.0016015341971069574,
-        0.0008584632305428386,
-        0.000128391373436898,
-        -0.00028913054848089814,
-        -0.0019626382272690535,
-        0.5930843949317932,
-        0.001174500328488648,
-        0.0017860672669485211,
-        0.0010054412996396422,
-        0.00012193901784485206,
-        -0.000334887852659449,
-        -0.0022029897663742304,
-        0.5935000777244568,
-        0.0012568456586450338,
-        0.0019742222502827644,
-        0.0011647609062492847,
-        0.00011021356476703659,
-        -0.00038211053470149636,
-        -0.0024467180483043194,
-        0.5931875109672546,
-        0.0013310370268300176,
-        0.0021642642095685005,
-        0.0013335321564227343,
-        0.00009949743980541825,
-        -0.0004296436673030257,
-        -0.0027013549115508795,
-        0.5931177139282227,
-        0.0014006041456013918,
-        0.002352928975597024,
-        0.0015152853447943926,
-        0.00008505478035658598,
-        -0.00048136088298633695,
-        -0.0029538467060774565,
-        0.5930479764938354,
-        0.0014608069323003292,
-        0.0025415997952222824,
-        0.001700855791568756,
-        0.00006727704749209806,
-        -0.0005341243813745677,
-        -0.003210614435374737,
-        0.5928036570549011,
-        0.0015100123127922416,
-        0.002731549320742488,
-        0.0018991812830790877,
-        0.0000493652478326112,
-        -0.0005809388239867985,
-        -0.003473253920674324,
-        0.5928736925125122,
-        0.0015514988917857409,
-        0.002922238316386938,
-        0.0021079820580780506,
-        0.000029167196771595627,
-        -0.0006329127354547381,
-        -0.0037316547241061926,
-        0.5928043127059937,
-        0.0015838317340239882,
-        0.0031066250521689653,
-        0.002319576218724251,
-        8.898059604689479e-6,
-        -0.0006800378323532641,
-        -0.004003639332950115,
-        0.5929091572761536,
-        0.0016050132689997554,
-        0.003295252099633217,
-        0.0025419394951313734,
-        -0.000013710295206692535,
-        -0.0007257868419401348,
-        -0.004269901663064957,
-        0.5930485725402832,
-        0.0016164487460628152,
-        0.0034825268667191267,
-        0.002774182241410017,
-        -0.000037491106922971085,
-        -0.0007724225870333612,
-        -0.004538912326097488,
-        0.5931875705718994,
-        0.001615947112441063,
-        0.003664180636405945,
-        0.0030124683398753405,
-        -0.00006199560448294505,
-        -0.0008142396691255271,
-        -0.00480045098811388,
-        0.5933974385261536,
-        0.0016095866449177265,
-        0.0038478909991681576,
-        0.003257976844906807,
-        -0.00008698971942067146,
-        -0.0008547022007405758,
-        -0.005078013986349106,
-        0.5936415791511536,
-        0.0015872354852035642,
-        0.004021089058369398,
-        0.0035109000746160746,
-        -0.00011568223999347538,
-        -0.0008942089625634253,
-        -0.005354971159249544,
-        0.5934317708015442,
-        0.0015591973206028342,
-        0.004198537208139896,
-        0.003773036180064082,
-        -0.00014495827781502157,
-        -0.0009303062106482685,
-        -0.005638875067234039,
-        0.5936412215232849,
-        0.0015177299501374364,
-        0.004366280045360327,
-        0.004040749743580818,
-        -0.00017588783521205187,
-        -0.0009663543896749616,
-        -0.005925025325268507,
-        0.593501627445221,
-        0.0014705799985677004,
-        0.004531881771981716,
-        0.00431715976446867,
-        -0.00021041395666543394,
-        -0.0010010612895712256,
-        -0.006207368802279234,
-        0.5935365557670593,
-        0.001413213089108467,
-        0.004692579619586468,
-        0.004599601496011019,
-        -0.0002457766968291253,
-        -0.0010327493073418736,
-        -0.006505406461656094,
-        0.5933269262313843,
-        0.0013434475986286998,
-        0.004848736338317394,
-        0.004890408832579851,
-        -0.00028711999766528606,
-        -0.00106858741492033,
-        -0.006807356141507626,
-        0.5931181311607361,
-        0.0012650334974750876,
-        0.004994421266019344,
-        0.0051898653618991375,
-        -0.00033172621624544263,
-        -0.0010990891605615616,
-        -0.0071100820787250996,
-        0.5929442048072815,
-        0.0011793962912634015,
-        0.005135250277817249,
-        0.005499166436493397,
-        -0.000378662021830678,
-        -0.0011317729949951172,
-        -0.007409243378788233,
-        0.5926647782325745,
-        0.0010838431771844625,
-        0.005268934182822704,
-        0.005817332770675421,
-        -0.0004281000583432615,
-        -0.001166283618658781,
-        -0.007717216853052378,
-        0.5926998853683472,
-        0.0009804241126403213,
-        0.0053992304019629955,
-        0.0061420174315571785,
-        -0.00048094423254951835,
-        -0.0011993126245215535,
-        -0.008030007593333721,
-        0.5924902558326721,
-        0.0008692154078744352,
-        0.005519451107829809,
-        0.006479023955762386,
-        -0.000537763989996165,
-        -0.001236212207004428,
-        -0.008356506004929543,
-        0.5929440855979919,
-        0.0007513284799642861,
-        0.00563538633286953,
-        0.006823732051998377,
-        -0.000594516284763813,
-        -0.0012742418330162764,
-        -0.008685515262186527,
-        0.5926643013954163,
-        0.0006232408923096955,
-        0.005743546411395073,
-        0.007178108207881451,
-        -0.0006553126149810851,
-        -0.0013130278093740344,
-        -0.00901535339653492,
-        0.5930123329162598,
-        0.00048613501712679863,
-        0.005843072663992643,
-        0.007545872125774622,
-        -0.0007216006051748991,
-        -0.001357274129986763,
-        -0.009348330087959766,
-        0.5926995277404785,
-        0.000340388563927263,
-        0.005928103346377611,
-        0.007921233773231506,
-        -0.0007906133541837335,
-        -0.0013983722310513258,
-        -0.00968977902084589,
-        0.5930134654045105,
-        0.00018408334290143102,
-        0.006012112833559513,
-        0.008309606462717056,
-        -0.0008611961966380477,
-        -0.00144277885556221,
-        -0.010034735314548016,
-        0.5933630466461182,
-        0.000022206833818927407,
-        0.0060895816422998905,
-        0.008708948269486427,
-        -0.0009344238787889481,
-        -0.0014912645565345883,
-        -0.01038195751607418,
-        0.5936756134033203,
-        -0.00014933485363144428,
-        0.006156634073704481,
-        0.009113290347158909,
-        -0.001009406871162355,
-        -0.0015364158898591995,
-        -0.01074444130063057,
-        0.5944079756736755,
-        -0.0003295359492767602,
-        0.006216940935701132,
-        0.009534457698464394,
-        -0.0010867133969441056,
-        -0.0015858380356803536,
-        -0.011105277575552464,
-        0.5946523547172546,
-        -0.0005184217588976026,
-        0.006267391610890627,
-        0.009968035854399204,
-        -0.00116726232226938,
-        -0.001635841210372746,
-        -0.011476362124085426,
-        0.5947223901748657,
-        -0.0007183448178693652,
-        0.006307759787887335,
-        0.010411876253783703,
-        -0.0012491614324972034,
-        -0.0016832650871947408,
-        -0.011858409270644188,
-        0.5945469737052917,
-        -0.000922650913707912,
-        0.006340802181512117,
-        0.01086830347776413,
-        -0.0013331895461305976,
-        -0.0017334959702566266,
-        -0.012237261049449444,
-        0.5951407551765442,
-        -0.001138067222200334,
-        0.006367202382534742,
-        0.011336571536958218,
-        -0.0014204677427187562,
-        -0.001783861662261188,
-        -0.012612935155630112,
-        0.5946868658065796,
-        -0.0013612991897389293,
-        0.006382528692483902,
-        0.011817813850939274,
-        -0.0015092657413333654,
-        -0.0018364684656262398,
-        -0.013003477826714516,
-        0.595176637172699,
-        -0.0015915771946310997,
-        0.0063857706263661385,
-        0.01231638714671135,
-        -0.0015994333662092686,
-        -0.0018854403169825673,
-        -0.013393816538155079,
-        0.5955926179885864,
-        -0.001827688654884696,
-        0.006383033003658056,
-        0.012827028520405293,
-        -0.0016901197377592325,
-        -0.0019397168653085828,
-        -0.01378911454230547,
-        0.5955935120582581,
-        -0.002072748262435198,
-        0.006370946299284697,
-        0.013353490270674229,
-        -0.0017832767916843295,
-        -0.001991107827052474,
-        -0.014198213815689087,
-        0.595174252986908,
-        -0.0023247504141181707,
-        0.006349846720695496,
-        0.013896841555833817,
-        -0.0018769301241263747,
-        -0.0020415729377418756,
-        -0.014593549072742462,
-        0.5950352549552917,
-        -0.0025832781102508307,
-        0.0063181049190461636,
-        0.01445536408573389,
-        -0.0019729691557586193,
-        -0.002093689749017358,
-        -0.015006774105131626,
-        0.5953840613365173
-      ],
-      "std": [
-        0.002411161782220006,
-        0.0035220407880842686,
-        0.0029013415332883596,
-        0.0013258246472105384,
-        0.0020179853308945894,
-        0.010011205449700356,
-        0.4910063147544861,
-        0.0048064119182527065,
-        0.007010283879935741,
-        0.005793850403279066,
-        0.0026270404923707247,
-        0.004016784951090813,
-        0.019959986209869385,
-        0.4910795986652374,
-        0.0072022695094347,
-        0.01047250535339117,
-        0.008696537464857101,
-        0.003923765849322081,
-        0.006003526505082846,
-        0.029928741976618767,
-        0.4911730885505676,
-        0.009601284749805927,
-        0.013944732025265694,
-        0.011596043594181538,
-        0.0051947347819805145,
-        0.007982160896062851,
-        0.039855923503637314,
-        0.49109989404678345,
-        0.011995380744338036,
-        0.017401276156306267,
-        0.01450654398649931,
-        0.006452825851738453,
-        0.009959585964679718,
-        0.049764279276132584,
-        0.4912789762020111,
-        0.014385105110704899,
-        0.020854491740465164,
-        0.017407990992069244,
-        0.007683087605983019,
-        0.011908459477126598,
-        0.059651929885149,
-        0.4912128448486328,
-        0.016765346750617027,
-        0.024284135550260544,
-        0.02030768245458603,
-        0.008882720954716206,
-        0.013833398930728436,
-        0.06945177167654037,
-        0.4911862313747406,
-        0.01913418062031269,
-        0.02769860439002514,
-        0.023200584575533867,
-        0.010069170035421848,
-        0.01573152281343937,
-        0.07920067012310028,
-        0.49123910069465637,
-        0.021495427936315536,
-        0.03108437918126583,
-        0.02608773671090603,
-        0.011218286119401455,
-        0.017600294202566147,
-        0.08888415992259979,
-        0.4912588894367218,
-        0.023848513141274452,
-        0.03446579724550247,
-        0.028954684734344482,
-        0.012335631065070629,
-        0.0194363035261631,
-        0.0984557494521141,
-        0.4911799430847168,
-        0.02618129923939705,
-        0.0377996489405632,
-        0.031812794506549835,
-        0.013426545076072216,
-        0.021240200847387314,
-        0.10793018341064453,
-        0.4912393391132355,
-        0.02850136160850525,
-        0.04113452509045601,
-        0.03465236350893974,
-        0.01448102854192257,
-        0.023004556074738503,
-        0.11730524897575378,
-        0.491252601146698,
-        0.030810443684458733,
-        0.0444553941488266,
-        0.03749171271920204,
-        0.01552011538296938,
-        0.02474648877978325,
-        0.12660986185073853,
-        0.49126577377319336,
-        0.033104438334703445,
-        0.047724947333335876,
-        0.04029623046517372,
-        0.01652422547340393,
-        0.026438293978571892,
-        0.1357518434524536,
-        0.4913119971752167,
-        0.03538680821657181,
-        0.05097673833370209,
-        0.04309060424566269,
-        0.017495229840278625,
-        0.02809188887476921,
-        0.14484061300754547,
-        0.49129876494407654,
-        0.037644777446985245,
-        0.054201558232307434,
-        0.045869018882513046,
-        0.018448922783136368,
-        0.029708286747336388,
-        0.1537874937057495,
-        0.4913118779659271,
-        0.03989279642701149,
-        0.05739012360572815,
-        0.04861682280898094,
-        0.019360067322850227,
-        0.031284283846616745,
-        0.16262821853160858,
-        0.4912920594215393,
-        0.04212509095668793,
-        0.060563746839761734,
-        0.05134649947285652,
-        0.020253626629710197,
-        0.03281664475798607,
-        0.17136116325855255,
-        0.4912656545639038,
-        0.04433578997850418,
-        0.06369910389184952,
-        0.05404900759458542,
-        0.021123351529240608,
-        0.03430699184536934,
-        0.1799684315919876,
-        0.4912393391132355,
-        0.0465271882712841,
-        0.06679350882768631,
-        0.05673054978251457,
-        0.02196338027715683,
-        0.035762034356594086,
-        0.1884494125843048,
-        0.49119946360588074,
-        0.04870201274752617,
-        0.0698690414428711,
-        0.0593833364546299,
-        0.02277866192162037,
-        0.03716905042529106,
-        0.19684050977230072,
-        0.4911530017852783,
-        0.05086364969611168,
-        0.0729081928730011,
-        0.06201274320483208,
-        0.023575762286782265,
-        0.03854267671704292,
-        0.20511800050735474,
-        0.49119293689727783,
-        0.052999626845121384,
-        0.07590420544147491,
-        0.06461741775274277,
-        0.024347545579075813,
-        0.03986750543117523,
-        0.21329139173030853,
-        0.4911530613899231,
-        0.05511622130870819,
-        0.07886126637458801,
-        0.06719071418046951,
-        0.02509898692369461,
-        0.04115770384669304,
-        0.22134073078632355,
-        0.4911796748638153,
-        0.05720451846718788,
-        0.0817820280790329,
-        0.06973543763160706,
-        0.02582353726029396,
-        0.04240873083472252,
-        0.22928737103939056,
-        0.49117299914360046,
-        0.05928165465593338,
-        0.08468781411647797,
-        0.07225947827100754,
-        0.026536913588643074,
-        0.0436241552233696,
-        0.23713935911655426,
-        0.4912128746509552,
-        0.06133345514535904,
-        0.08753105252981186,
-        0.07475195825099945,
-        0.02723076194524765,
-        0.04480902478098869,
-        0.2449100911617279,
-        0.49125248193740845,
-        0.06336378306150436,
-        0.09035338461399078,
-        0.07722371071577072,
-        0.027915935963392258,
-        0.04596531763672829,
-        0.25256893038749695,
-        0.49128544330596924,
-        0.06536917388439178,
-        0.0931331068277359,
-        0.07966616004705429,
-        0.028593653813004494,
-        0.04708719253540039,
-        0.26014769077301025,
-        0.49133822321891785,
-        0.06735322624444962,
-        0.09587426483631134,
-        0.08208146691322327,
-        0.029253413900732994,
-        0.04818820580840111,
-        0.2676284909248352,
-        0.4913316071033478,
-        0.06931550055742264,
-        0.09857368469238281,
-        0.084467314183712,
-        0.029913172125816345,
-        0.049258600920438766,
-        0.27503275871276855,
-        0.4913710951805115,
-        0.07124854624271393,
-        0.10124263167381287,
-        0.08683151751756668,
-        0.03056473471224308,
-        0.05031215026974678,
-        0.2823602259159088,
-        0.4912854731082916,
-        0.07316309213638306,
-        0.10387503355741501,
-        0.08917013555765152,
-        0.031205736100673676,
-        0.051339127123355865,
-        0.2896381616592407,
-        0.491338312625885,
-        0.0750511884689331,
-        0.10646932572126389,
-        0.09148410707712173,
-        0.03184282034635544,
-        0.05234614759683609,
-        0.2968275845050812,
-        0.49127253890037537,
-        0.07691479474306107,
-        0.10901817679405212,
-        0.0937764048576355,
-        0.032469242811203,
-        0.053335245698690414,
-        0.30398809909820557,
-        0.49133166670799255,
-        0.07874985039234161,
-        0.11152230948209763,
-        0.0960441529750824,
-        0.03308483585715294,
-        0.054303981363773346,
-        0.311068594455719,
-        0.49127233028411865,
-        0.08056764304637909,
-        0.11400332301855087,
-        0.09828827530145645,
-        0.03369200974702835,
-        0.055251531302928925,
-        0.31809186935424805,
-        0.49120599031448364,
-        0.08236116915941238,
-        0.11642979085445404,
-        0.10051009804010391,
-        0.034294869750738144,
-        0.05618305504322052,
-        0.32508960366249084,
-        0.4911465048789978,
-        0.0841296911239624,
-        0.11883242428302765,
-        0.10270429402589798,
-        0.034887079149484634,
-        0.0570959635078907,
-        0.3320097327232361,
-        0.4910062551498413,
-        0.08587083965539932,
-        0.12118621915578842,
-        0.10488012433052063,
-        0.0354686863720417,
-        0.05799192935228348,
-        0.3389015197753906,
-        0.49095919728279114,
-        0.08759026974439621,
-        0.12350363284349442,
-        0.10703437030315399,
-        0.03604672849178314,
-        0.05887560918927193,
-        0.3457629084587097,
-        0.4909456968307495,
-        0.08928582817316055,
-        0.12578748166561127,
-        0.10916976630687714,
-        0.03662176802754402,
-        0.059738170355558395,
-        0.35257115960121155,
-        0.49097952246665955,
-        0.09095519781112671,
-        0.1280231475830078,
-        0.11128115653991699,
-        0.03717926889657974,
-        0.06058768928050995,
-        0.35931187868118286,
-        0.4908647835254669,
-        0.09259875863790512,
-        0.13021688163280487,
-        0.11337121576070786,
-        0.03773583471775055,
-        0.061423756182193756,
-        0.3660118579864502,
-        0.4909525215625763,
-        0.09422165900468826,
-        0.13238278031349182,
-        0.11544124782085419,
-        0.03828861564397812,
-        0.06224775314331055,
-        0.3726641535758972,
-        0.4908578395843506,
-        0.09581982344388962,
-        0.13450181484222412,
-        0.11749458312988281,
-        0.03884110972285271,
-        0.06304997950792313,
-        0.3792511522769928,
-        0.49077698588371277,
-        0.09739767760038376,
-        0.13658517599105835,
-        0.11952484399080276,
-        0.03938090428709984,
-        0.06384975463151932,
-        0.3857972025871277,
-        0.49077683687210083,
-        0.09894638508558273,
-        0.1386304795742035,
-        0.12153798341751099,
-        0.03992167487740517,
-        0.06463739275932312,
-        0.39230042695999146,
-        0.4908582866191864,
-        0.10048258304595947,
-        0.14064013957977295,
-        0.12353785336017609,
-        0.040460873395204544,
-        0.06541262567043304,
-        0.398761510848999,
-        0.4908851981163025,
-        0.10199275612831116,
-        0.14262202382087708,
-        0.1255190521478653,
-        0.04099829122424126,
-        0.0661839172244072,
-        0.40516018867492676,
-        0.4908175766468048
-      ],
-      "q01": [
-        -0.007347712090983987,
-        -0.011486765748262403,
-        -0.005602361589856446,
-        -0.004156268547289073,
-        -0.007298904010653494,
-        -0.03456402559429407,
-        0.0,
-        -0.014532031904906033,
-        -0.023063078013062475,
-        -0.011146851527690887,
-        -0.008034631153941155,
-        -0.014402847960591318,
-        -0.0688410488665104,
-        0.0,
-        -0.02174613598883152,
-        -0.03454083656966686,
-        -0.016602809804677963,
-        -0.012004367487877608,
-        -0.02169839859008789,
-        -0.10323694312572479,
-        0.0,
-        -0.02924629563242197,
-        -0.046255945521593095,
-        -0.02208325160816312,
-        -0.01599733569249511,
-        -0.029179477728903294,
-        -0.13703150334358216,
-        0.0,
-        -0.0364012400791049,
-        -0.057891583660244945,
-        -0.027614226043969393,
-        -0.020049774289131163,
-        -0.03620416237413883,
-        -0.17225709611177442,
-        0.0,
-        -0.04366054475307465,
-        -0.06915460828691722,
-        -0.03313590524941683,
-        -0.023996917428076275,
-        -0.0432967299669981,
-        -0.20717854452133178,
-        0.0,
-        -0.0507191886767745,
-        -0.08091499265134335,
-        -0.038471172228455544,
-        -0.02769389592409134,
-        -0.05027697439491748,
-        -0.240767266279459,
-        0.0,
-        -0.05764442007690668,
-        -0.0928664004802704,
-        -0.04389051187932491,
-        -0.031378988534212116,
-        -0.05700045920312405,
-        -0.27346896691322325,
-        0.0,
-        -0.06481191511154175,
-        -0.10385700676739215,
-        -0.04937767976522446,
-        -0.034899483323097225,
-        -0.0640901755988598,
-        -0.3072322327375412,
-        0.0,
-        -0.07216909334659577,
-        -0.11465946695804596,
-        -0.05446739228963852,
-        -0.0384699240922928,
-        -0.07105188292860984,
-        -0.34030354830026627,
-        0.0,
-        -0.0791815688431263,
-        -0.12541235695779324,
-        -0.05973851401358843,
-        -0.04202334811687469,
-        -0.07781512894928456,
-        -0.37346914051771163,
-        0.0,
-        -0.08563443173170089,
-        -0.13601192309856414,
-        -0.06484265891909599,
-        -0.045433390045166,
-        -0.0843420141339302,
-        -0.40387091467380526,
-        0.0,
-        -0.09227466106414796,
-        -0.14718177989125253,
-        -0.06990853486061097,
-        -0.04866062937080862,
-        -0.09135572135448455,
-        -0.43357558735609053,
-        0.0,
-        -0.09871668579578399,
-        -0.15755651119351388,
-        -0.07494964042305946,
-        -0.05228844651579856,
-        -0.0973639042377472,
-        -0.4662614274024963,
-        0.0,
-        -0.10474277555942535,
-        -0.16660683050751687,
-        -0.07973185749202967,
-        -0.055503451734781245,
-        -0.1043376196026802,
-        -0.49795184282064436,
-        0.0,
-        -0.11146926714479923,
-        -0.17675033633708956,
-        -0.08481720062792301,
-        -0.05867235680818558,
-        -0.11146564957499505,
-        -0.5234594867229462,
-        0.0,
-        -0.1183029289394617,
-        -0.18591568176746368,
-        -0.08976946114003659,
-        -0.06189141765832901,
-        -0.11618306014537812,
-        -0.5559864712238312,
-        0.0,
-        -0.1254465562880039,
-        -0.1967654939889908,
-        -0.09452381261885166,
-        -0.06458974866867065,
-        -0.12229588437080384,
-        -0.5838854856491089,
-        0.0,
-        -0.13160774660110475,
-        -0.20561206495761872,
-        -0.09930612608790398,
-        -0.06763142464160918,
-        -0.12729683189988136,
-        -0.6118898140192032,
-        0.0,
-        -0.13866937249898909,
-        -0.21400038709640504,
-        -0.10383093198537827,
-        -0.07044867560863496,
-        -0.1330213810920715,
-        -0.6420615591049195,
-        0.0,
-        -0.14518664595484734,
-        -0.22208737790584565,
-        -0.10839852929115296,
-        -0.07363164942264558,
-        -0.13909859127998353,
-        -0.6659443435192108,
-        0.0,
-        -0.15202519185543062,
-        -0.23166744391918181,
-        -0.1132360664010048,
-        -0.07554293347597121,
-        -0.1431354057431221,
-        -0.6935547601938247,
-        0.0,
-        -0.1575157383620739,
-        -0.24115877079963685,
-        -0.11769197665154933,
-        -0.07880985802412033,
-        -0.14886759107112885,
-        -0.717457844042778,
-        0.0,
-        -0.1637449370443821,
-        -0.25002506982684136,
-        -0.12177611397504806,
-        -0.08111752626895904,
-        -0.15336844004392625,
-        -0.7450204710006714,
-        0.0,
-        -0.1702753342092037,
-        -0.25778238211274146,
-        -0.12590762196183203,
-        -0.08350524621009828,
-        -0.15739910807609558,
-        -0.7672739201784134,
-        0.0,
-        -0.17570297884345054,
-        -0.26699419630765914,
-        -0.12999144119024275,
-        -0.08586254574060437,
-        -0.15982728340625763,
-        -0.7919603824377061,
-        0.0,
-        -0.18209079703688621,
-        -0.2762185108423233,
-        -0.1341805691242218,
-        -0.08846430575847625,
-        -0.1640431535243988,
-        -0.8073872842788696,
-        0.0,
-        -0.18759027983546256,
-        -0.28415180140137675,
-        -0.13808169324100017,
-        -0.09098501863479613,
-        -0.1681320061683655,
-        -0.8263913946390152,
-        0.0,
-        -0.193171888589859,
-        -0.29068409420847896,
-        -0.14178944163620472,
-        -0.09304863989353179,
-        -0.1707449821949005,
-        -0.8507119262695311,
-        0.0,
-        -0.19844869979619978,
-        -0.2987963397145271,
-        -0.14549240586161613,
-        -0.09557797043323518,
-        -0.1739663097858429,
-        -0.8648041958332062,
-        0.0,
-        -0.20466184456944464,
-        -0.3049391875267029,
-        -0.14948453903496267,
-        -0.09817119854688644,
-        -0.17706590107083323,
-        -0.8849965750694275,
-        0.0,
-        -0.20983124628067015,
-        -0.3116496807098389,
-        -0.15297152189016341,
-        -0.10035031213760376,
-        -0.18140975254774094,
-        -0.9072247592926024,
-        0.0,
-        -0.215193039560318,
-        -0.31998531384468076,
-        -0.1565369277715683,
-        -0.102377852499485,
-        -0.1837332302033901,
-        -0.9293165444850922,
-        0.0,
-        -0.22114226788282393,
-        -0.32775930384397506,
-        -0.16037838854789732,
-        -0.10488562031984328,
-        -0.18766805601119993,
-        -0.9526331702470778,
-        0.0,
-        -0.22637617546319963,
-        -0.33561479420661927,
-        -0.16403944179415703,
-        -0.10686019870638847,
-        -0.19061433601379393,
-        -0.9705972829818725,
-        0.0,
-        -0.23192334132194517,
-        -0.3404881935119629,
-        -0.16749516949653626,
-        -0.10840021744370465,
-        -0.19327300127148628,
-        -0.9914580883026123,
-        0.0,
-        -0.23655534625053407,
-        -0.3483747916221619,
-        -0.1713618674516678,
-        -0.11093877178430556,
-        -0.19658433060646058,
-        -1.0074259456634522,
-        0.0,
-        -0.24189636796712874,
-        -0.35468626945018766,
-        -0.174676346886158,
-        -0.1137608123421669,
-        -0.1998320359289646,
-        -1.02401462726593,
-        0.0,
-        -0.24619000346064568,
-        -0.3617886746406555,
-        -0.17737202043533326,
-        -0.11608646607995032,
-        -0.20212158007621764,
-        -1.0396305271863937,
-        0.0,
-        -0.25069174761176105,
-        -0.3690223491430283,
-        -0.18019738556742668,
-        -0.11791741763353347,
-        -0.20542705215215684,
-        -1.056211252117157,
-        0.0,
-        -0.25575273798704146,
-        -0.37492762922048567,
-        -0.18413128078579902,
-        -0.11963063821196557,
-        -0.2084871412396431,
-        -1.0777596963644027,
-        0.0,
-        -0.26051444165706633,
-        -0.38124460592269893,
-        -0.18717371021509172,
-        -0.12171284379959108,
-        -0.21154849707484244,
-        -1.0918209570646285,
-        0.0,
-        -0.2645109035611153,
-        -0.38571969106197357,
-        -0.19055300951004028,
-        -0.12364264481067655,
-        -0.21497644165158272,
-        -1.1041330133199692,
-        0.0,
-        -0.26856355381011965,
-        -0.3920482913136482,
-        -0.19341499418020247,
-        -0.12596707943677904,
-        -0.21652717173099517,
-        -1.1284716510772705,
-        0.0,
-        -0.273536373257637,
-        -0.3985861983895302,
-        -0.19648525762557983,
-        -0.12863396686315537,
-        -0.2178426690995693,
-        -1.1498000781059265,
-        0.0,
-        -0.2773340993881226,
-        -0.4043089840650559,
-        -0.19916273007392882,
-        -0.1301111794948578,
-        -0.21953763974308968,
-        -1.1649874792098998,
-        0.0,
-        -0.28152843511104586,
-        -0.4105489842891693,
-        -0.20206411379575728,
-        -0.13238328326940535,
-        -0.22030045819282532,
-        -1.186898805141449,
-        0.0,
-        -0.28521604499816894,
-        -0.41719991471767426,
-        -0.20488486919403076,
-        -0.13367428585290908,
-        -0.2215240351676941,
-        -1.2064387265205383,
-        0.0,
-        -0.28908531114459035,
-        -0.4219293355822563,
-        -0.20779289478063584,
-        -0.1346039095699787,
-        -0.22363287283182146,
-        -1.225643195438385,
-        0.0,
-        -0.294126475507021,
-        -0.42881255321502687,
-        -0.21011174390912055,
-        -0.13763879417181013,
-        -0.22653749623298647,
-        -1.2400950877666475,
-        0.0
-      ],
-      "q99": [
-        0.00760505946278572,
-        0.01227881843298674,
-        0.009394833338446915,
-        0.0041071553006768215,
-        0.005846777586638929,
-        0.03394978983253241,
-        0.9998,
-        0.015092343334481124,
-        0.024545711845159537,
-        0.018547782172262668,
-        0.008136489582061764,
-        0.011534944232553239,
-        0.0680388870239258,
-        0.9998,
-        0.022567954233288764,
-        0.03644986734837294,
-        0.0278723166257143,
-        0.012022978414595133,
-        0.017261751674115658,
-        0.1018139317035675,
-        0.9998,
-        0.030176736226677894,
-        0.048319679215550415,
-        0.037043299572169786,
-        0.015823813125491137,
-        0.022935147598385802,
-        0.13534477652311327,
-        0.9998,
-        0.03749606138020754,
-        0.05974762381315232,
-        0.046320767011493444,
-        0.01970605190098286,
-        0.028365900403261188,
-        0.16783747158050544,
-        0.9998,
-        0.04486110335588456,
-        0.07147922609001398,
-        0.055334820427000525,
-        0.023239636448025694,
-        0.034125556316971783,
-        0.2017985020875931,
-        0.9998,
-        0.052173979285359384,
-        0.08346393676698208,
-        0.06446339850723744,
-        0.026847184759378445,
-        0.03955972051620485,
-        0.23437213500142096,
-        0.9998,
-        0.059150983133912094,
-        0.09537746536731717,
-        0.0733001764550805,
-        0.029705484813451755,
-        0.04516481494158506,
-        0.26646220774650575,
-        0.9998,
-        0.06645039129257202,
-        0.1076395375341177,
-        0.08239804291725158,
-        0.0328832214474678,
-        0.04995994846820831,
-        0.29610841860771175,
-        0.9998,
-        0.07374503636360169,
-        0.11985380845069887,
-        0.09145532561540604,
-        0.035987181842327104,
-        0.05490816103816032,
-        0.3266934569597244,
-        0.9998,
-        0.08122003978192807,
-        0.1319003387391567,
-        0.10024555467069149,
-        0.038626870799064655,
-        0.05953236908912657,
-        0.35580547877550117,
-        0.9998,
-        0.08822669934630395,
-        0.14304091303348543,
-        0.1088144795835018,
-        0.041924731212854394,
-        0.06486271733045579,
-        0.3876368781208992,
-        0.9998,
-        0.09569223463535309,
-        0.1534084576368332,
-        0.11751553359031677,
-        0.04469002693891522,
-        0.06991170048713685,
-        0.41592443135976787,
-        0.9998,
-        0.10324303380250932,
-        0.1640547949969769,
-        0.1258540060549974,
-        0.04742067799866201,
-        0.0741334307938814,
-        0.4491768553853035,
-        0.9998,
-        0.1099683414697647,
-        0.17510256106853483,
-        0.13403582906574008,
-        0.050331409102678304,
-        0.07791369587183,
-        0.475221610414982,
-        0.9998,
-        0.1170920331031084,
-        0.1857688842058181,
-        0.14203765797317025,
-        0.05343856880664827,
-        0.0820521941632032,
-        0.5052882493734359,
-        0.9998,
-        0.12414569870531561,
-        0.19753187570571906,
-        0.14974090793728828,
-        0.0557895161241293,
-        0.0865580612182617,
-        0.5326116788625717,
-        0.9998,
-        0.13112460071146492,
-        0.20844777574539186,
-        0.15755365291237833,
-        0.05866813796758652,
-        0.09100452079772947,
-        0.5617906215190887,
-        0.9998,
-        0.1379249924659729,
-        0.21917983024120336,
-        0.16431246396899224,
-        0.06069063851833345,
-        0.09500974670350554,
-        0.5912433532238006,
-        0.9998,
-        0.14485807346105578,
-        0.2286938194990158,
-        0.1716713245987892,
-        0.0637721403002739,
-        0.09919737915396692,
-        0.6155789003372192,
-        0.9998,
-        0.15117210847139356,
-        0.23827072948217387,
-        0.1779224501848221,
-        0.06597466281652448,
-        0.10370546174049378,
-        0.644374561071396,
-        0.9998,
-        0.15787530031800268,
-        0.24751081448793422,
-        0.18454510100185867,
-        0.06831172816753389,
-        0.10792329218983648,
-        0.6659029854536058,
-        0.9998,
-        0.1641180765330792,
-        0.2572556215524673,
-        0.19053607739508155,
-        0.07030175954103468,
-        0.11277426195144652,
-        0.686683092904091,
-        0.9998,
-        0.16966928005814552,
-        0.26658033823966976,
-        0.19653976464271544,
-        0.07203448786735533,
-        0.11643967783451081,
-        0.7125654235839844,
-        0.9998,
-        0.17621982062458996,
-        0.27701201248764995,
-        0.20197029424905777,
-        0.07416592988967896,
-        0.12035323338508608,
-        0.7360334300994871,
-        0.9998,
-        0.1819627068042755,
-        0.2855635068178177,
-        0.2084233057498932,
-        0.07603480890989306,
-        0.1241462073802948,
-        0.7568318559646605,
-        0.9998,
-        0.18748426148295405,
-        0.29386607773303985,
-        0.21488302874565124,
-        0.0777944794893265,
-        0.1279024946689606,
-        0.7800548309087754,
-        0.9998,
-        0.19402539111375805,
-        0.304579441523552,
-        0.22076346032619476,
-        0.07946800847053526,
-        0.13082660132646562,
-        0.8035489074707032,
-        0.9998,
-        0.1999460756778717,
-        0.31222576675415037,
-        0.2264517953634262,
-        0.08081280782818795,
-        0.13460262061357497,
-        0.828994060134888,
-        0.9998,
-        0.20509765752553943,
-        0.3203624894678593,
-        0.23198571143150332,
-        0.08211237077713013,
-        0.13750549141168594,
-        0.8529169931411746,
-        0.9998,
-        0.209518383550644,
-        0.3281992249488831,
-        0.23816398755311968,
-        0.08322777889966959,
-        0.14037072820067403,
-        0.8746512705802916,
-        0.9998,
-        0.21442690343856818,
-        0.33666496825218195,
-        0.24309153485894203,
-        0.08529254732131952,
-        0.14304055777192115,
-        0.8924013189554216,
-        0.9998,
-        0.21941403126716608,
-        0.3448349338173866,
-        0.2489651171147823,
-        0.08667572829127312,
-        0.14580873468518263,
-        0.9107246013164523,
-        0.9998,
-        0.22309734332561493,
-        0.3517628137350083,
-        0.2547237434387207,
-        0.08896696496009826,
-        0.1489701636433602,
-        0.9272562242746352,
-        0.9998,
-        0.22793451219797134,
-        0.3591047751426697,
-        0.2597683270871639,
-        0.09042025351524352,
-        0.1528471918106079,
-        0.9483436929702762,
-        0.9998,
-        0.23258997063636777,
-        0.36902411174774175,
-        0.26485966441631315,
-        0.09243017539381981,
-        0.15506044900417326,
-        0.9667170533418656,
-        0.9998,
-        0.23665347018241878,
-        0.3762736029624939,
-        0.27067578415870663,
-        0.09449174585938458,
-        0.15888449568748475,
-        0.9834513599395751,
-        0.9998,
-        0.24093078261613854,
-        0.3843366762638093,
-        0.2768812180876732,
-        0.09594143676757816,
-        0.16144601993560792,
-        1.0022019856452942,
-        0.9998,
-        0.2459272434949874,
-        0.3907656828403474,
-        0.2815030888676643,
-        0.09760697330236434,
-        0.16464969681501385,
-        1.019599228835106,
-        0.9998,
-        0.2511993800699711,
-        0.3962667070865631,
-        0.28746061475872997,
-        0.09892506269812584,
-        0.1685606077313423,
-        1.0345355377674106,
-        0.9998,
-        0.25586065447330475,
-        0.4064154948949814,
-        0.29352941393256193,
-        0.10010581925511364,
-        0.17137197246551505,
-        1.0455808718204502,
-        0.9998,
-        0.26066332874298104,
-        0.41295039305686965,
-        0.29917095521688464,
-        0.10042237902879714,
-        0.17334867343306548,
-        1.0644763714551928,
-        0.9998,
-        0.26424761767387395,
-        0.4171489597439766,
-        0.3047941813468933,
-        0.10030385837554934,
-        0.1771283437609672,
-        1.0775494430065153,
-        0.9998,
-        0.266823394203186,
-        0.42120621981620787,
-        0.31085526600480073,
-        0.10133219141960148,
-        0.18098006631135943,
-        1.0927770781517028,
-        0.9998,
-        0.27192984950542454,
-        0.4258830359578132,
-        0.31648782253265384,
-        0.10210902032852165,
-        0.18391375703215596,
-        1.1100240562438963,
-        0.9998,
-        0.2774573613643646,
-        0.4328473631143568,
-        0.3229170823454858,
-        0.10328457579612738,
-        0.18559744659066202,
-        1.127002547836304,
-        0.9998,
-        0.280809062719345,
-        0.43832696628570567,
-        0.32813225684762004,
-        0.10458858071565635,
-        0.18929516908526423,
-        1.1399588096141815,
-        0.9998,
-        0.28525815182924275,
-        0.44289318814277645,
-        0.3333625353813172,
-        0.1055313608527183,
-        0.1918839191913605,
-        1.1545847342491151,
-        0.9998,
-        0.2899247071444988,
-        0.4499819297909735,
-        0.33962785017490393,
-        0.10614367849826811,
-        0.19452174307107928,
-        1.1747124679088592,
-        0.9998,
-        0.29274274289608004,
-        0.4549682779312134,
-        0.3463232246458531,
-        0.10680567473173141,
-        0.19644608254432672,
-        1.185605804681778,
-        0.9998
-      ]
-    }
-  },
-  "sample_ratio": 0.05,
-  "dataset_infos": {
-    "robotgeneralist/pp__1": {
-      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
-      "num_episodes": 237,
-      "num_frames": 99363
-    },
-    "robotgeneralist/pp__1__adversarial": {
-      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
-      "num_episodes": 64,
-      "num_frames": 17906
-    },
-    "robotgeneralist/pp__2": {
-      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
-      "num_episodes": 92,
-      "num_frames": 37458
-    },
-    "robotgeneralist/pp__2__rotations": {
-      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
-      "num_episodes": 102,
-      "num_frames": 13935
-    },
-    "robotgeneralist/pp__3__with_ft": {
-      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
-      "num_episodes": 99,
-      "num_frames": 38424
-    },
-    "robotgeneralist/pp__3__with_ft__rotations": {
-      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
-      "num_episodes": 49,
-      "num_frames": 7151
-    },
-    "robotgeneralist/pp__3__with_ft__adversarial_start": {
-      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
-      "num_episodes": 147,
-      "num_frames": 21313
-    },
-    "robotgeneralist/pp__4__with_ft": {
-      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
-      "num_episodes": 94,
-      "num_frames": 33046
-    },
-    "robotgeneralist/pp__5__with_ft__adversarial_target": {
-      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
-      "num_episodes": 48,
-      "num_frames": 10615
-    },
-    "robotgeneralist/pp__6__with_ft": {
-      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
-      "num_episodes": 379,
-      "num_frames": 123611
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-1": {
-      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
-      "num_episodes": 98,
-      "num_frames": 41410
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-2": {
-      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
-      "num_episodes": 94,
-      "num_frames": 37859
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-3": {
-      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
-      "num_episodes": 127,
-      "num_frames": 53341
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-4": {
-      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
-      "num_episodes": 102,
-      "num_frames": 38263
-    }
-  }
-}
\ No newline at end of file
diff --git a/pp__ee_baseline/25000/params/_METADATA b/pp__ee_baseline/25000/params/_METADATA
deleted file mode 100644
index 9eb73624158f4514d49041e0806d2d702c62eee1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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, "write_shape": [1152]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304]}}, "('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, "write_shape": [27, 1152, 4304]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 16, 72, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [14, 14, 3, 1152]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1152, 2048]}}, "('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, "write_shape": [1, 256, 1152]}}, "('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, "write_shape": [257152, 2048]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [18, 8, 256, 2048]}}, "('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, "write_shape": [18, 8, 256, 1024]}}, "('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, "write_shape": [18, 2, 1, 2048, 256]}}, "('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, "write_shape": [18, 2, 1, 1024, 256]}}, "('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, "write_shape": [18, 8, 2048, 256]}}, "('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, "write_shape": [18, 8, 1024, 256]}}, "('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, "write_shape": [18, 2, 2048, 16384]}}, "('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, "write_shape": [18, 16384, 2048]}}, "('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, "write_shape": [18, 2, 1024, 4096]}}, "('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, "write_shape": [18, 4096, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}, "('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, "write_shape": [32]}}, "('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, "write_shape": [1024, 32]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [2048, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [1024, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/pp__ee_baseline/25000/params/_sharding b/pp__ee_baseline/25000/params/_sharding
deleted file mode 100644
index 924608abdcb99abb3c864bb9d1bfca6eb60f780e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/_sharding
+++ /dev/null
@@ -1 +0,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}]]}}"}
\ No newline at end of file
diff --git a/pp__ee_baseline/25000/params/array_metadatas/process_0 b/pp__ee_baseline/25000/params/array_metadatas/process_0
deleted file mode 100644
index 6aa9fd1decf72607133eb54664d98802bf563b5c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/array_metadatas/process_0
+++ /dev/null
@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/pp__ee_baseline/25000/params/d/41f6c4f83b1eb80410d89ad963f0a39f b/pp__ee_baseline/25000/params/d/41f6c4f83b1eb80410d89ad963f0a39f
deleted file mode 100644
index 0eb61ba5e70a45994a0b51f7b247c9109164eca5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/d/41f6c4f83b1eb80410d89ad963f0a39f and /dev/null differ
diff --git a/pp__ee_baseline/25000/params/manifest.ocdbt b/pp__ee_baseline/25000/params/manifest.ocdbt
deleted file mode 100644
index 471de7e3c618f077a6019d7a7edc47d20d9f47f5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/0e1b99ca04307b6d19b3bdfc27357085 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/0e1b99ca04307b6d19b3bdfc27357085
deleted file mode 100644
index 497be7c5adfbef1224fbccea5256a7630ee700ef..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/0e1b99ca04307b6d19b3bdfc27357085
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e22991f62580aa9b5ed12c12118775df43b677be971fc9d4586be9fb94b6980e
-size 132846806
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/0f3a141236b8e149552f1a977a6b16c0 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/0f3a141236b8e149552f1a977a6b16c0
deleted file mode 100644
index 115b5942ce202eac392deea36594cf6fcd392922..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/0f3a141236b8e149552f1a977a6b16c0
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:67df6cdd60d8ba009531522bbb7fa50831974bf931a51525ff22df15550d9c22
-size 272906229
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/35c2c1d489133606b988c1cc29af68fc b/pp__ee_baseline/25000/params/ocdbt.process_0/d/35c2c1d489133606b988c1cc29af68fc
deleted file mode 100644
index a838165ce936c5b2aa32dcc9fe28a960d8d38b9e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/35c2c1d489133606b988c1cc29af68fc
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:10122b68e2b7f64430e161e9f47d36068202625e60fdb92621acacaf5583151a
-size 1120108367
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/405e0e508866b12d4e747360349fa197 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/405e0e508866b12d4e747360349fa197
deleted file mode 100644
index 1deb39271c040954f157251c27980c9aa1aa6ebb..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/405e0e508866b12d4e747360349fa197
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a00f76cb76f5f92ab9c04926a39d68e8a279f347c27949c00de74e985bec6a82
-size 139916180
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/445d32db4bb6d08103f59ba26ae50d65 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/445d32db4bb6d08103f59ba26ae50d65
deleted file mode 100644
index 4f63957473fc957c450a405d63b850656188c3d1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/445d32db4bb6d08103f59ba26ae50d65
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:543a8594aa04e2644145d02eda6256b7b1ab06f342266d3c70128064aee0a8ee
-size 34974331
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/616dc1787a328eb4d4b068e99908820e b/pp__ee_baseline/25000/params/ocdbt.process_0/d/616dc1787a328eb4d4b068e99908820e
deleted file mode 100644
index b3f30d2151758d7410e8824097328cd34ccfd7fa..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/ocdbt.process_0/d/616dc1787a328eb4d4b068e99908820e and /dev/null differ
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/6a459fc596203be9bd9fe7390b8831c6 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/6a459fc596203be9bd9fe7390b8831c6
deleted file mode 100644
index 2fa372020fa02a4969314a6be08e401f4906d756..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/6a459fc596203be9bd9fe7390b8831c6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f0a46cc79e1c7591be274bba1ce04c2af1b0024b5e897b4f98834650ecd84f9f
-size 496759554
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/7164d558c2571f920e27cb4267c3f271 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/7164d558c2571f920e27cb4267c3f271
deleted file mode 100644
index 1eab164aab812b8975c172205f2b5e3acd122c82..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/7164d558c2571f920e27cb4267c3f271
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d8cbaa207c3f38825967c21b5df138e446e90478314fa3e297415be61aa09bf6
-size 279250123
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/8a39bc8646e20830ac636d9e521e5f52 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/8a39bc8646e20830ac636d9e521e5f52
deleted file mode 100644
index ffc57c5cefe3160b4f5ec2c8030ab168d9324180..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/ocdbt.process_0/d/8a39bc8646e20830ac636d9e521e5f52 and /dev/null differ
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/8ed420382c28bfa98785f5f86d2333c4 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/8ed420382c28bfa98785f5f86d2333c4
deleted file mode 100644
index 4c0a969f61507aa0504dbcc13a8829f51b348144..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/8ed420382c28bfa98785f5f86d2333c4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a8d83570dfcd53eff8f04b272254c5e1070f140abbfce9d1125c724e48fef65a
-size 280038428
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/90b7387f30cc88038864ef386a4abc3f b/pp__ee_baseline/25000/params/ocdbt.process_0/d/90b7387f30cc88038864ef386a4abc3f
deleted file mode 100644
index 98544838d055e5b4ca7012ea94b86a1cbeafc7ef..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/90b7387f30cc88038864ef386a4abc3f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8115faa56b1900a6ba0b8095bc9efb32e8209131b8c93fd17fb03b21e5e08eea
-size 1957442664
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/979dfcb66887eb5146badd8d8d00d6e6 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/979dfcb66887eb5146badd8d8d00d6e6
deleted file mode 100644
index c78aceee8456a0da49b2fcf564e245d28cc5b44f..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/979dfcb66887eb5146badd8d8d00d6e6
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ac7180500266d1d8c0a2131b367f5cab9f4f78d32e4b0c3f9f8cc193fe57ba3b
-size 1120134161
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/a29e284285f8c69956ab85176e46c026 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/a29e284285f8c69956ab85176e46c026
deleted file mode 100644
index 57b5256970ee6fc9d5f59955e944a273ac2dd86c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/a29e284285f8c69956ab85176e46c026
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8145a4e1d59abf3b08270ba679561fb0c8608c6d7698aa654cedda88fed6cdf6
-size 2240248593
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/a3ee5200478f84ca9ef02f1a3217f55f b/pp__ee_baseline/25000/params/ocdbt.process_0/d/a3ee5200478f84ca9ef02f1a3217f55f
deleted file mode 100644
index bacec68fd04beae1803b46e8ce34781b82a2f606..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/a3ee5200478f84ca9ef02f1a3217f55f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:027dc8ae8a359f58c0a51c35d2c3c868f3da089cdc19447f5398ea95ac10d61c
-size 70083154
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/af0d2902ce6046296d6b8087cbf63993 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/af0d2902ce6046296d6b8087cbf63993
deleted file mode 100644
index 90fbb0575063e9958a84c67e5156c22c02804b22..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/af0d2902ce6046296d6b8087cbf63993
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8d2cab09d128d5496b73afb7381c766517066db6f2d496a11d0d5da82ea94c77
-size 280188037
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/b1436f9c1d4fb318d0aaa830169be260 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/b1436f9c1d4fb318d0aaa830169be260
deleted file mode 100644
index 113eba7b4d9220c135b0a18979dc376b5e0f8d0f..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/b1436f9c1d4fb318d0aaa830169be260
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:65be1c5eddfb172da53adb371376c66cbddeaa1bbe4a611837fe59e1fe9cf7fa
-size 2240230455
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/bcb3b4efaf6455fcb1bed3025961e342 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/bcb3b4efaf6455fcb1bed3025961e342
deleted file mode 100644
index 15f7ee6d3906da8addcc24a4dd19baf7a3e749ae..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/bcb3b4efaf6455fcb1bed3025961e342
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:40fdeb131c58b0271b80a75de5d62bcb74382ecf9bcb1b13256ed0b5329b028e
-size 497035911
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/c90675a326382ef18c6f526046bfa514 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/c90675a326382ef18c6f526046bfa514
deleted file mode 100644
index 74c93e6eea2d6ed921c5c5bc514ed191830c99ce..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/c90675a326382ef18c6f526046bfa514
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:041c1ab08955b44185abf0e12c0df9863f1627c9999181234fc92f5b3722de13
-size 362558
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/d4839b4fb6f955277ba4aa02682b0ecf b/pp__ee_baseline/25000/params/ocdbt.process_0/d/d4839b4fb6f955277ba4aa02682b0ecf
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/ocdbt.process_0/d/d4839b4fb6f955277ba4aa02682b0ecf and /dev/null differ
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/ebe0c1cc8071bbf942be64db013f1ef7 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/ebe0c1cc8071bbf942be64db013f1ef7
deleted file mode 100644
index cb6a6e0a3bddceb4791210b529a77a9fc42fa7d6..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/ebe0c1cc8071bbf942be64db013f1ef7
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:23eb64d58154db81e9706a979dd1532cfaedb0ca56f2cb7b402ba8d8d711c24f
-size 559722117
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/efc7e3ff17c82858829bc4f238b8562c b/pp__ee_baseline/25000/params/ocdbt.process_0/d/efc7e3ff17c82858829bc4f238b8562c
deleted file mode 100644
index 72f8858b7636404811f0b7ce3ffb6825b2f0d492..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/efc7e3ff17c82858829bc4f238b8562c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:678d1750fb20d29cd20b6ead5a46828421fbb964f2af7fd5c061f07adc622e63
-size 17061339
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/f6694e9c3c94df6c8a96828254b0b9f0 b/pp__ee_baseline/25000/params/ocdbt.process_0/d/f6694e9c3c94df6c8a96828254b0b9f0
deleted file mode 100644
index ae817d19e339e34f29649468df7b970477b60f54..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/f6694e9c3c94df6c8a96828254b0b9f0
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b6a8ea67fced8a20c6154dc8f70c0e271839ac4928355bf62f9090d2f85ea36f
-size 265931749
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/d/fd5accf503d4f0be6ef08037608d9e3a b/pp__ee_baseline/25000/params/ocdbt.process_0/d/fd5accf503d4f0be6ef08037608d9e3a
deleted file mode 100644
index 83fadaabf8fbe020cd212215025b432502c6376e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/25000/params/ocdbt.process_0/d/fd5accf503d4f0be6ef08037608d9e3a
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:aaae9eef598c5fb0ef490a1361c55c0ed0dd4988e68e63c1f10d49cf15e3b243
-size 8881359
diff --git a/pp__ee_baseline/25000/params/ocdbt.process_0/manifest.ocdbt b/pp__ee_baseline/25000/params/ocdbt.process_0/manifest.ocdbt
deleted file mode 100644
index 1ce85287492347aa1f59d5cbb8d8b4480619c94b..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/25000/params/ocdbt.process_0/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/5000/_CHECKPOINT_METADATA b/pp__ee_baseline/5000/_CHECKPOINT_METADATA
deleted file mode 100644
index 26e84d50e09ca781130d2ca457333370f1873bf3..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/_CHECKPOINT_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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": 1768740317811847756, "commit_timestamp_nsecs": 1768740369148769257, "custom_metadata": {}}
\ No newline at end of file
diff --git a/pp__ee_baseline/5000/assets/ur10e/norm_stats.json b/pp__ee_baseline/5000/assets/ur10e/norm_stats.json
deleted file mode 100644
index 8a1b886d6b668da778297b91b3d82381c2eab28a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/assets/ur10e/norm_stats.json
+++ /dev/null
@@ -1,1525 +0,0 @@
-{
-  "norm_stats": {
-    "state": {
-      "mean": [
-        0.05313729867339134,
-        0.0037223484832793474,
-        0.106409952044487,
-        0.0008666226058267057,
-        -0.00854077935218811,
-        0.01460699550807476,
-        0.5940247178077698
-      ],
-      "std": [
-        0.1176244392991066,
-        0.16962096095085144,
-        0.07773701101541519,
-        0.04537155479192734,
-        0.06616315245628357,
-        0.6038344502449036,
-        0.49107977747917175
-      ],
-      "q01": [
-        -0.2025573879778385,
-        -0.31106759839057924,
-        0.00890737680345774,
-        -0.11727181684970855,
-        -0.16711108471155167,
-        -1.423702568435669,
-        0.0
-      ],
-      "q99": [
-        0.29901714037060734,
-        0.4115018085956573,
-        0.3217185490995646,
-        0.13774808425903318,
-        0.19241279754638663,
-        1.598234023094177,
-        0.9998
-      ]
-    },
-    "actions": {
-      "mean": [
-        0.0001451610296498984,
-        0.00017207347264047712,
-        0.000048651279939804226,
-        0.00003654604734038003,
-        -0.000019767590856645256,
-        -0.00019861181499436498,
-        0.5944076776504517,
-        0.0002841090608853847,
-        0.00034722191048786044,
-        0.00011135452223243192,
-        0.00006827477045590058,
-        -0.000043589698179857805,
-        -0.0004028017574455589,
-        0.594025731086731,
-        0.00041486899135634303,
-        0.0005213047261349857,
-        0.00018569435633253306,
-        0.00009245114779332653,
-        -0.00007049143459880725,
-        -0.000616009347140789,
-        0.5935360789299011,
-        0.000542119552846998,
-        0.0006975074065849185,
-        0.00027102616149932146,
-        0.00011072693450842053,
-        -0.00010138262587133795,
-        -0.0008296052110381424,
-        0.593919575214386,
-        0.0006618681945838034,
-        0.0008754940354265273,
-        0.0003684645926114172,
-        0.00012383276771288365,
-        -0.000135598995257169,
-        -0.0010419095633551478,
-        0.5929783582687378,
-        0.0007787057547830045,
-        0.00105847988743335,
-        0.0004721089790109545,
-        0.00013335929543245584,
-        -0.00017038211808539927,
-        -0.0012720258673653007,
-        0.5933270454406738,
-        0.0008861633250489831,
-        0.001237610587850213,
-        0.0005911999032832682,
-        0.00013674379442818463,
-        -0.00020881379896309227,
-        -0.0014943619025871158,
-        0.5934669971466064,
-        0.0009874902898445725,
-        0.0014197468990460038,
-        0.0007170690805651248,
-        0.00013581194798462093,
-        -0.0002465852303430438,
-        -0.0017284756759181619,
-        0.5931888818740845,
-        0.0010826619109138846,
-        0.0016015341971069574,
-        0.0008584632305428386,
-        0.000128391373436898,
-        -0.00028913054848089814,
-        -0.0019626382272690535,
-        0.5930843949317932,
-        0.001174500328488648,
-        0.0017860672669485211,
-        0.0010054412996396422,
-        0.00012193901784485206,
-        -0.000334887852659449,
-        -0.0022029897663742304,
-        0.5935000777244568,
-        0.0012568456586450338,
-        0.0019742222502827644,
-        0.0011647609062492847,
-        0.00011021356476703659,
-        -0.00038211053470149636,
-        -0.0024467180483043194,
-        0.5931875109672546,
-        0.0013310370268300176,
-        0.0021642642095685005,
-        0.0013335321564227343,
-        0.00009949743980541825,
-        -0.0004296436673030257,
-        -0.0027013549115508795,
-        0.5931177139282227,
-        0.0014006041456013918,
-        0.002352928975597024,
-        0.0015152853447943926,
-        0.00008505478035658598,
-        -0.00048136088298633695,
-        -0.0029538467060774565,
-        0.5930479764938354,
-        0.0014608069323003292,
-        0.0025415997952222824,
-        0.001700855791568756,
-        0.00006727704749209806,
-        -0.0005341243813745677,
-        -0.003210614435374737,
-        0.5928036570549011,
-        0.0015100123127922416,
-        0.002731549320742488,
-        0.0018991812830790877,
-        0.0000493652478326112,
-        -0.0005809388239867985,
-        -0.003473253920674324,
-        0.5928736925125122,
-        0.0015514988917857409,
-        0.002922238316386938,
-        0.0021079820580780506,
-        0.000029167196771595627,
-        -0.0006329127354547381,
-        -0.0037316547241061926,
-        0.5928043127059937,
-        0.0015838317340239882,
-        0.0031066250521689653,
-        0.002319576218724251,
-        8.898059604689479e-6,
-        -0.0006800378323532641,
-        -0.004003639332950115,
-        0.5929091572761536,
-        0.0016050132689997554,
-        0.003295252099633217,
-        0.0025419394951313734,
-        -0.000013710295206692535,
-        -0.0007257868419401348,
-        -0.004269901663064957,
-        0.5930485725402832,
-        0.0016164487460628152,
-        0.0034825268667191267,
-        0.002774182241410017,
-        -0.000037491106922971085,
-        -0.0007724225870333612,
-        -0.004538912326097488,
-        0.5931875705718994,
-        0.001615947112441063,
-        0.003664180636405945,
-        0.0030124683398753405,
-        -0.00006199560448294505,
-        -0.0008142396691255271,
-        -0.00480045098811388,
-        0.5933974385261536,
-        0.0016095866449177265,
-        0.0038478909991681576,
-        0.003257976844906807,
-        -0.00008698971942067146,
-        -0.0008547022007405758,
-        -0.005078013986349106,
-        0.5936415791511536,
-        0.0015872354852035642,
-        0.004021089058369398,
-        0.0035109000746160746,
-        -0.00011568223999347538,
-        -0.0008942089625634253,
-        -0.005354971159249544,
-        0.5934317708015442,
-        0.0015591973206028342,
-        0.004198537208139896,
-        0.003773036180064082,
-        -0.00014495827781502157,
-        -0.0009303062106482685,
-        -0.005638875067234039,
-        0.5936412215232849,
-        0.0015177299501374364,
-        0.004366280045360327,
-        0.004040749743580818,
-        -0.00017588783521205187,
-        -0.0009663543896749616,
-        -0.005925025325268507,
-        0.593501627445221,
-        0.0014705799985677004,
-        0.004531881771981716,
-        0.00431715976446867,
-        -0.00021041395666543394,
-        -0.0010010612895712256,
-        -0.006207368802279234,
-        0.5935365557670593,
-        0.001413213089108467,
-        0.004692579619586468,
-        0.004599601496011019,
-        -0.0002457766968291253,
-        -0.0010327493073418736,
-        -0.006505406461656094,
-        0.5933269262313843,
-        0.0013434475986286998,
-        0.004848736338317394,
-        0.004890408832579851,
-        -0.00028711999766528606,
-        -0.00106858741492033,
-        -0.006807356141507626,
-        0.5931181311607361,
-        0.0012650334974750876,
-        0.004994421266019344,
-        0.0051898653618991375,
-        -0.00033172621624544263,
-        -0.0010990891605615616,
-        -0.0071100820787250996,
-        0.5929442048072815,
-        0.0011793962912634015,
-        0.005135250277817249,
-        0.005499166436493397,
-        -0.000378662021830678,
-        -0.0011317729949951172,
-        -0.007409243378788233,
-        0.5926647782325745,
-        0.0010838431771844625,
-        0.005268934182822704,
-        0.005817332770675421,
-        -0.0004281000583432615,
-        -0.001166283618658781,
-        -0.007717216853052378,
-        0.5926998853683472,
-        0.0009804241126403213,
-        0.0053992304019629955,
-        0.0061420174315571785,
-        -0.00048094423254951835,
-        -0.0011993126245215535,
-        -0.008030007593333721,
-        0.5924902558326721,
-        0.0008692154078744352,
-        0.005519451107829809,
-        0.006479023955762386,
-        -0.000537763989996165,
-        -0.001236212207004428,
-        -0.008356506004929543,
-        0.5929440855979919,
-        0.0007513284799642861,
-        0.00563538633286953,
-        0.006823732051998377,
-        -0.000594516284763813,
-        -0.0012742418330162764,
-        -0.008685515262186527,
-        0.5926643013954163,
-        0.0006232408923096955,
-        0.005743546411395073,
-        0.007178108207881451,
-        -0.0006553126149810851,
-        -0.0013130278093740344,
-        -0.00901535339653492,
-        0.5930123329162598,
-        0.00048613501712679863,
-        0.005843072663992643,
-        0.007545872125774622,
-        -0.0007216006051748991,
-        -0.001357274129986763,
-        -0.009348330087959766,
-        0.5926995277404785,
-        0.000340388563927263,
-        0.005928103346377611,
-        0.007921233773231506,
-        -0.0007906133541837335,
-        -0.0013983722310513258,
-        -0.00968977902084589,
-        0.5930134654045105,
-        0.00018408334290143102,
-        0.006012112833559513,
-        0.008309606462717056,
-        -0.0008611961966380477,
-        -0.00144277885556221,
-        -0.010034735314548016,
-        0.5933630466461182,
-        0.000022206833818927407,
-        0.0060895816422998905,
-        0.008708948269486427,
-        -0.0009344238787889481,
-        -0.0014912645565345883,
-        -0.01038195751607418,
-        0.5936756134033203,
-        -0.00014933485363144428,
-        0.006156634073704481,
-        0.009113290347158909,
-        -0.001009406871162355,
-        -0.0015364158898591995,
-        -0.01074444130063057,
-        0.5944079756736755,
-        -0.0003295359492767602,
-        0.006216940935701132,
-        0.009534457698464394,
-        -0.0010867133969441056,
-        -0.0015858380356803536,
-        -0.011105277575552464,
-        0.5946523547172546,
-        -0.0005184217588976026,
-        0.006267391610890627,
-        0.009968035854399204,
-        -0.00116726232226938,
-        -0.001635841210372746,
-        -0.011476362124085426,
-        0.5947223901748657,
-        -0.0007183448178693652,
-        0.006307759787887335,
-        0.010411876253783703,
-        -0.0012491614324972034,
-        -0.0016832650871947408,
-        -0.011858409270644188,
-        0.5945469737052917,
-        -0.000922650913707912,
-        0.006340802181512117,
-        0.01086830347776413,
-        -0.0013331895461305976,
-        -0.0017334959702566266,
-        -0.012237261049449444,
-        0.5951407551765442,
-        -0.001138067222200334,
-        0.006367202382534742,
-        0.011336571536958218,
-        -0.0014204677427187562,
-        -0.001783861662261188,
-        -0.012612935155630112,
-        0.5946868658065796,
-        -0.0013612991897389293,
-        0.006382528692483902,
-        0.011817813850939274,
-        -0.0015092657413333654,
-        -0.0018364684656262398,
-        -0.013003477826714516,
-        0.595176637172699,
-        -0.0015915771946310997,
-        0.0063857706263661385,
-        0.01231638714671135,
-        -0.0015994333662092686,
-        -0.0018854403169825673,
-        -0.013393816538155079,
-        0.5955926179885864,
-        -0.001827688654884696,
-        0.006383033003658056,
-        0.012827028520405293,
-        -0.0016901197377592325,
-        -0.0019397168653085828,
-        -0.01378911454230547,
-        0.5955935120582581,
-        -0.002072748262435198,
-        0.006370946299284697,
-        0.013353490270674229,
-        -0.0017832767916843295,
-        -0.001991107827052474,
-        -0.014198213815689087,
-        0.595174252986908,
-        -0.0023247504141181707,
-        0.006349846720695496,
-        0.013896841555833817,
-        -0.0018769301241263747,
-        -0.0020415729377418756,
-        -0.014593549072742462,
-        0.5950352549552917,
-        -0.0025832781102508307,
-        0.0063181049190461636,
-        0.01445536408573389,
-        -0.0019729691557586193,
-        -0.002093689749017358,
-        -0.015006774105131626,
-        0.5953840613365173
-      ],
-      "std": [
-        0.002411161782220006,
-        0.0035220407880842686,
-        0.0029013415332883596,
-        0.0013258246472105384,
-        0.0020179853308945894,
-        0.010011205449700356,
-        0.4910063147544861,
-        0.0048064119182527065,
-        0.007010283879935741,
-        0.005793850403279066,
-        0.0026270404923707247,
-        0.004016784951090813,
-        0.019959986209869385,
-        0.4910795986652374,
-        0.0072022695094347,
-        0.01047250535339117,
-        0.008696537464857101,
-        0.003923765849322081,
-        0.006003526505082846,
-        0.029928741976618767,
-        0.4911730885505676,
-        0.009601284749805927,
-        0.013944732025265694,
-        0.011596043594181538,
-        0.0051947347819805145,
-        0.007982160896062851,
-        0.039855923503637314,
-        0.49109989404678345,
-        0.011995380744338036,
-        0.017401276156306267,
-        0.01450654398649931,
-        0.006452825851738453,
-        0.009959585964679718,
-        0.049764279276132584,
-        0.4912789762020111,
-        0.014385105110704899,
-        0.020854491740465164,
-        0.017407990992069244,
-        0.007683087605983019,
-        0.011908459477126598,
-        0.059651929885149,
-        0.4912128448486328,
-        0.016765346750617027,
-        0.024284135550260544,
-        0.02030768245458603,
-        0.008882720954716206,
-        0.013833398930728436,
-        0.06945177167654037,
-        0.4911862313747406,
-        0.01913418062031269,
-        0.02769860439002514,
-        0.023200584575533867,
-        0.010069170035421848,
-        0.01573152281343937,
-        0.07920067012310028,
-        0.49123910069465637,
-        0.021495427936315536,
-        0.03108437918126583,
-        0.02608773671090603,
-        0.011218286119401455,
-        0.017600294202566147,
-        0.08888415992259979,
-        0.4912588894367218,
-        0.023848513141274452,
-        0.03446579724550247,
-        0.028954684734344482,
-        0.012335631065070629,
-        0.0194363035261631,
-        0.0984557494521141,
-        0.4911799430847168,
-        0.02618129923939705,
-        0.0377996489405632,
-        0.031812794506549835,
-        0.013426545076072216,
-        0.021240200847387314,
-        0.10793018341064453,
-        0.4912393391132355,
-        0.02850136160850525,
-        0.04113452509045601,
-        0.03465236350893974,
-        0.01448102854192257,
-        0.023004556074738503,
-        0.11730524897575378,
-        0.491252601146698,
-        0.030810443684458733,
-        0.0444553941488266,
-        0.03749171271920204,
-        0.01552011538296938,
-        0.02474648877978325,
-        0.12660986185073853,
-        0.49126577377319336,
-        0.033104438334703445,
-        0.047724947333335876,
-        0.04029623046517372,
-        0.01652422547340393,
-        0.026438293978571892,
-        0.1357518434524536,
-        0.4913119971752167,
-        0.03538680821657181,
-        0.05097673833370209,
-        0.04309060424566269,
-        0.017495229840278625,
-        0.02809188887476921,
-        0.14484061300754547,
-        0.49129876494407654,
-        0.037644777446985245,
-        0.054201558232307434,
-        0.045869018882513046,
-        0.018448922783136368,
-        0.029708286747336388,
-        0.1537874937057495,
-        0.4913118779659271,
-        0.03989279642701149,
-        0.05739012360572815,
-        0.04861682280898094,
-        0.019360067322850227,
-        0.031284283846616745,
-        0.16262821853160858,
-        0.4912920594215393,
-        0.04212509095668793,
-        0.060563746839761734,
-        0.05134649947285652,
-        0.020253626629710197,
-        0.03281664475798607,
-        0.17136116325855255,
-        0.4912656545639038,
-        0.04433578997850418,
-        0.06369910389184952,
-        0.05404900759458542,
-        0.021123351529240608,
-        0.03430699184536934,
-        0.1799684315919876,
-        0.4912393391132355,
-        0.0465271882712841,
-        0.06679350882768631,
-        0.05673054978251457,
-        0.02196338027715683,
-        0.035762034356594086,
-        0.1884494125843048,
-        0.49119946360588074,
-        0.04870201274752617,
-        0.0698690414428711,
-        0.0593833364546299,
-        0.02277866192162037,
-        0.03716905042529106,
-        0.19684050977230072,
-        0.4911530017852783,
-        0.05086364969611168,
-        0.0729081928730011,
-        0.06201274320483208,
-        0.023575762286782265,
-        0.03854267671704292,
-        0.20511800050735474,
-        0.49119293689727783,
-        0.052999626845121384,
-        0.07590420544147491,
-        0.06461741775274277,
-        0.024347545579075813,
-        0.03986750543117523,
-        0.21329139173030853,
-        0.4911530613899231,
-        0.05511622130870819,
-        0.07886126637458801,
-        0.06719071418046951,
-        0.02509898692369461,
-        0.04115770384669304,
-        0.22134073078632355,
-        0.4911796748638153,
-        0.05720451846718788,
-        0.0817820280790329,
-        0.06973543763160706,
-        0.02582353726029396,
-        0.04240873083472252,
-        0.22928737103939056,
-        0.49117299914360046,
-        0.05928165465593338,
-        0.08468781411647797,
-        0.07225947827100754,
-        0.026536913588643074,
-        0.0436241552233696,
-        0.23713935911655426,
-        0.4912128746509552,
-        0.06133345514535904,
-        0.08753105252981186,
-        0.07475195825099945,
-        0.02723076194524765,
-        0.04480902478098869,
-        0.2449100911617279,
-        0.49125248193740845,
-        0.06336378306150436,
-        0.09035338461399078,
-        0.07722371071577072,
-        0.027915935963392258,
-        0.04596531763672829,
-        0.25256893038749695,
-        0.49128544330596924,
-        0.06536917388439178,
-        0.0931331068277359,
-        0.07966616004705429,
-        0.028593653813004494,
-        0.04708719253540039,
-        0.26014769077301025,
-        0.49133822321891785,
-        0.06735322624444962,
-        0.09587426483631134,
-        0.08208146691322327,
-        0.029253413900732994,
-        0.04818820580840111,
-        0.2676284909248352,
-        0.4913316071033478,
-        0.06931550055742264,
-        0.09857368469238281,
-        0.084467314183712,
-        0.029913172125816345,
-        0.049258600920438766,
-        0.27503275871276855,
-        0.4913710951805115,
-        0.07124854624271393,
-        0.10124263167381287,
-        0.08683151751756668,
-        0.03056473471224308,
-        0.05031215026974678,
-        0.2823602259159088,
-        0.4912854731082916,
-        0.07316309213638306,
-        0.10387503355741501,
-        0.08917013555765152,
-        0.031205736100673676,
-        0.051339127123355865,
-        0.2896381616592407,
-        0.491338312625885,
-        0.0750511884689331,
-        0.10646932572126389,
-        0.09148410707712173,
-        0.03184282034635544,
-        0.05234614759683609,
-        0.2968275845050812,
-        0.49127253890037537,
-        0.07691479474306107,
-        0.10901817679405212,
-        0.0937764048576355,
-        0.032469242811203,
-        0.053335245698690414,
-        0.30398809909820557,
-        0.49133166670799255,
-        0.07874985039234161,
-        0.11152230948209763,
-        0.0960441529750824,
-        0.03308483585715294,
-        0.054303981363773346,
-        0.311068594455719,
-        0.49127233028411865,
-        0.08056764304637909,
-        0.11400332301855087,
-        0.09828827530145645,
-        0.03369200974702835,
-        0.055251531302928925,
-        0.31809186935424805,
-        0.49120599031448364,
-        0.08236116915941238,
-        0.11642979085445404,
-        0.10051009804010391,
-        0.034294869750738144,
-        0.05618305504322052,
-        0.32508960366249084,
-        0.4911465048789978,
-        0.0841296911239624,
-        0.11883242428302765,
-        0.10270429402589798,
-        0.034887079149484634,
-        0.0570959635078907,
-        0.3320097327232361,
-        0.4910062551498413,
-        0.08587083965539932,
-        0.12118621915578842,
-        0.10488012433052063,
-        0.0354686863720417,
-        0.05799192935228348,
-        0.3389015197753906,
-        0.49095919728279114,
-        0.08759026974439621,
-        0.12350363284349442,
-        0.10703437030315399,
-        0.03604672849178314,
-        0.05887560918927193,
-        0.3457629084587097,
-        0.4909456968307495,
-        0.08928582817316055,
-        0.12578748166561127,
-        0.10916976630687714,
-        0.03662176802754402,
-        0.059738170355558395,
-        0.35257115960121155,
-        0.49097952246665955,
-        0.09095519781112671,
-        0.1280231475830078,
-        0.11128115653991699,
-        0.03717926889657974,
-        0.06058768928050995,
-        0.35931187868118286,
-        0.4908647835254669,
-        0.09259875863790512,
-        0.13021688163280487,
-        0.11337121576070786,
-        0.03773583471775055,
-        0.061423756182193756,
-        0.3660118579864502,
-        0.4909525215625763,
-        0.09422165900468826,
-        0.13238278031349182,
-        0.11544124782085419,
-        0.03828861564397812,
-        0.06224775314331055,
-        0.3726641535758972,
-        0.4908578395843506,
-        0.09581982344388962,
-        0.13450181484222412,
-        0.11749458312988281,
-        0.03884110972285271,
-        0.06304997950792313,
-        0.3792511522769928,
-        0.49077698588371277,
-        0.09739767760038376,
-        0.13658517599105835,
-        0.11952484399080276,
-        0.03938090428709984,
-        0.06384975463151932,
-        0.3857972025871277,
-        0.49077683687210083,
-        0.09894638508558273,
-        0.1386304795742035,
-        0.12153798341751099,
-        0.03992167487740517,
-        0.06463739275932312,
-        0.39230042695999146,
-        0.4908582866191864,
-        0.10048258304595947,
-        0.14064013957977295,
-        0.12353785336017609,
-        0.040460873395204544,
-        0.06541262567043304,
-        0.398761510848999,
-        0.4908851981163025,
-        0.10199275612831116,
-        0.14262202382087708,
-        0.1255190521478653,
-        0.04099829122424126,
-        0.0661839172244072,
-        0.40516018867492676,
-        0.4908175766468048
-      ],
-      "q01": [
-        -0.007347712090983987,
-        -0.011486765748262403,
-        -0.005602361589856446,
-        -0.004156268547289073,
-        -0.007298904010653494,
-        -0.03456402559429407,
-        0.0,
-        -0.014532031904906033,
-        -0.023063078013062475,
-        -0.011146851527690887,
-        -0.008034631153941155,
-        -0.014402847960591318,
-        -0.0688410488665104,
-        0.0,
-        -0.02174613598883152,
-        -0.03454083656966686,
-        -0.016602809804677963,
-        -0.012004367487877608,
-        -0.02169839859008789,
-        -0.10323694312572479,
-        0.0,
-        -0.02924629563242197,
-        -0.046255945521593095,
-        -0.02208325160816312,
-        -0.01599733569249511,
-        -0.029179477728903294,
-        -0.13703150334358216,
-        0.0,
-        -0.0364012400791049,
-        -0.057891583660244945,
-        -0.027614226043969393,
-        -0.020049774289131163,
-        -0.03620416237413883,
-        -0.17225709611177442,
-        0.0,
-        -0.04366054475307465,
-        -0.06915460828691722,
-        -0.03313590524941683,
-        -0.023996917428076275,
-        -0.0432967299669981,
-        -0.20717854452133178,
-        0.0,
-        -0.0507191886767745,
-        -0.08091499265134335,
-        -0.038471172228455544,
-        -0.02769389592409134,
-        -0.05027697439491748,
-        -0.240767266279459,
-        0.0,
-        -0.05764442007690668,
-        -0.0928664004802704,
-        -0.04389051187932491,
-        -0.031378988534212116,
-        -0.05700045920312405,
-        -0.27346896691322325,
-        0.0,
-        -0.06481191511154175,
-        -0.10385700676739215,
-        -0.04937767976522446,
-        -0.034899483323097225,
-        -0.0640901755988598,
-        -0.3072322327375412,
-        0.0,
-        -0.07216909334659577,
-        -0.11465946695804596,
-        -0.05446739228963852,
-        -0.0384699240922928,
-        -0.07105188292860984,
-        -0.34030354830026627,
-        0.0,
-        -0.0791815688431263,
-        -0.12541235695779324,
-        -0.05973851401358843,
-        -0.04202334811687469,
-        -0.07781512894928456,
-        -0.37346914051771163,
-        0.0,
-        -0.08563443173170089,
-        -0.13601192309856414,
-        -0.06484265891909599,
-        -0.045433390045166,
-        -0.0843420141339302,
-        -0.40387091467380526,
-        0.0,
-        -0.09227466106414796,
-        -0.14718177989125253,
-        -0.06990853486061097,
-        -0.04866062937080862,
-        -0.09135572135448455,
-        -0.43357558735609053,
-        0.0,
-        -0.09871668579578399,
-        -0.15755651119351388,
-        -0.07494964042305946,
-        -0.05228844651579856,
-        -0.0973639042377472,
-        -0.4662614274024963,
-        0.0,
-        -0.10474277555942535,
-        -0.16660683050751687,
-        -0.07973185749202967,
-        -0.055503451734781245,
-        -0.1043376196026802,
-        -0.49795184282064436,
-        0.0,
-        -0.11146926714479923,
-        -0.17675033633708956,
-        -0.08481720062792301,
-        -0.05867235680818558,
-        -0.11146564957499505,
-        -0.5234594867229462,
-        0.0,
-        -0.1183029289394617,
-        -0.18591568176746368,
-        -0.08976946114003659,
-        -0.06189141765832901,
-        -0.11618306014537812,
-        -0.5559864712238312,
-        0.0,
-        -0.1254465562880039,
-        -0.1967654939889908,
-        -0.09452381261885166,
-        -0.06458974866867065,
-        -0.12229588437080384,
-        -0.5838854856491089,
-        0.0,
-        -0.13160774660110475,
-        -0.20561206495761872,
-        -0.09930612608790398,
-        -0.06763142464160918,
-        -0.12729683189988136,
-        -0.6118898140192032,
-        0.0,
-        -0.13866937249898909,
-        -0.21400038709640504,
-        -0.10383093198537827,
-        -0.07044867560863496,
-        -0.1330213810920715,
-        -0.6420615591049195,
-        0.0,
-        -0.14518664595484734,
-        -0.22208737790584565,
-        -0.10839852929115296,
-        -0.07363164942264558,
-        -0.13909859127998353,
-        -0.6659443435192108,
-        0.0,
-        -0.15202519185543062,
-        -0.23166744391918181,
-        -0.1132360664010048,
-        -0.07554293347597121,
-        -0.1431354057431221,
-        -0.6935547601938247,
-        0.0,
-        -0.1575157383620739,
-        -0.24115877079963685,
-        -0.11769197665154933,
-        -0.07880985802412033,
-        -0.14886759107112885,
-        -0.717457844042778,
-        0.0,
-        -0.1637449370443821,
-        -0.25002506982684136,
-        -0.12177611397504806,
-        -0.08111752626895904,
-        -0.15336844004392625,
-        -0.7450204710006714,
-        0.0,
-        -0.1702753342092037,
-        -0.25778238211274146,
-        -0.12590762196183203,
-        -0.08350524621009828,
-        -0.15739910807609558,
-        -0.7672739201784134,
-        0.0,
-        -0.17570297884345054,
-        -0.26699419630765914,
-        -0.12999144119024275,
-        -0.08586254574060437,
-        -0.15982728340625763,
-        -0.7919603824377061,
-        0.0,
-        -0.18209079703688621,
-        -0.2762185108423233,
-        -0.1341805691242218,
-        -0.08846430575847625,
-        -0.1640431535243988,
-        -0.8073872842788696,
-        0.0,
-        -0.18759027983546256,
-        -0.28415180140137675,
-        -0.13808169324100017,
-        -0.09098501863479613,
-        -0.1681320061683655,
-        -0.8263913946390152,
-        0.0,
-        -0.193171888589859,
-        -0.29068409420847896,
-        -0.14178944163620472,
-        -0.09304863989353179,
-        -0.1707449821949005,
-        -0.8507119262695311,
-        0.0,
-        -0.19844869979619978,
-        -0.2987963397145271,
-        -0.14549240586161613,
-        -0.09557797043323518,
-        -0.1739663097858429,
-        -0.8648041958332062,
-        0.0,
-        -0.20466184456944464,
-        -0.3049391875267029,
-        -0.14948453903496267,
-        -0.09817119854688644,
-        -0.17706590107083323,
-        -0.8849965750694275,
-        0.0,
-        -0.20983124628067015,
-        -0.3116496807098389,
-        -0.15297152189016341,
-        -0.10035031213760376,
-        -0.18140975254774094,
-        -0.9072247592926024,
-        0.0,
-        -0.215193039560318,
-        -0.31998531384468076,
-        -0.1565369277715683,
-        -0.102377852499485,
-        -0.1837332302033901,
-        -0.9293165444850922,
-        0.0,
-        -0.22114226788282393,
-        -0.32775930384397506,
-        -0.16037838854789732,
-        -0.10488562031984328,
-        -0.18766805601119993,
-        -0.9526331702470778,
-        0.0,
-        -0.22637617546319963,
-        -0.33561479420661927,
-        -0.16403944179415703,
-        -0.10686019870638847,
-        -0.19061433601379393,
-        -0.9705972829818725,
-        0.0,
-        -0.23192334132194517,
-        -0.3404881935119629,
-        -0.16749516949653626,
-        -0.10840021744370465,
-        -0.19327300127148628,
-        -0.9914580883026123,
-        0.0,
-        -0.23655534625053407,
-        -0.3483747916221619,
-        -0.1713618674516678,
-        -0.11093877178430556,
-        -0.19658433060646058,
-        -1.0074259456634522,
-        0.0,
-        -0.24189636796712874,
-        -0.35468626945018766,
-        -0.174676346886158,
-        -0.1137608123421669,
-        -0.1998320359289646,
-        -1.02401462726593,
-        0.0,
-        -0.24619000346064568,
-        -0.3617886746406555,
-        -0.17737202043533326,
-        -0.11608646607995032,
-        -0.20212158007621764,
-        -1.0396305271863937,
-        0.0,
-        -0.25069174761176105,
-        -0.3690223491430283,
-        -0.18019738556742668,
-        -0.11791741763353347,
-        -0.20542705215215684,
-        -1.056211252117157,
-        0.0,
-        -0.25575273798704146,
-        -0.37492762922048567,
-        -0.18413128078579902,
-        -0.11963063821196557,
-        -0.2084871412396431,
-        -1.0777596963644027,
-        0.0,
-        -0.26051444165706633,
-        -0.38124460592269893,
-        -0.18717371021509172,
-        -0.12171284379959108,
-        -0.21154849707484244,
-        -1.0918209570646285,
-        0.0,
-        -0.2645109035611153,
-        -0.38571969106197357,
-        -0.19055300951004028,
-        -0.12364264481067655,
-        -0.21497644165158272,
-        -1.1041330133199692,
-        0.0,
-        -0.26856355381011965,
-        -0.3920482913136482,
-        -0.19341499418020247,
-        -0.12596707943677904,
-        -0.21652717173099517,
-        -1.1284716510772705,
-        0.0,
-        -0.273536373257637,
-        -0.3985861983895302,
-        -0.19648525762557983,
-        -0.12863396686315537,
-        -0.2178426690995693,
-        -1.1498000781059265,
-        0.0,
-        -0.2773340993881226,
-        -0.4043089840650559,
-        -0.19916273007392882,
-        -0.1301111794948578,
-        -0.21953763974308968,
-        -1.1649874792098998,
-        0.0,
-        -0.28152843511104586,
-        -0.4105489842891693,
-        -0.20206411379575728,
-        -0.13238328326940535,
-        -0.22030045819282532,
-        -1.186898805141449,
-        0.0,
-        -0.28521604499816894,
-        -0.41719991471767426,
-        -0.20488486919403076,
-        -0.13367428585290908,
-        -0.2215240351676941,
-        -1.2064387265205383,
-        0.0,
-        -0.28908531114459035,
-        -0.4219293355822563,
-        -0.20779289478063584,
-        -0.1346039095699787,
-        -0.22363287283182146,
-        -1.225643195438385,
-        0.0,
-        -0.294126475507021,
-        -0.42881255321502687,
-        -0.21011174390912055,
-        -0.13763879417181013,
-        -0.22653749623298647,
-        -1.2400950877666475,
-        0.0
-      ],
-      "q99": [
-        0.00760505946278572,
-        0.01227881843298674,
-        0.009394833338446915,
-        0.0041071553006768215,
-        0.005846777586638929,
-        0.03394978983253241,
-        0.9998,
-        0.015092343334481124,
-        0.024545711845159537,
-        0.018547782172262668,
-        0.008136489582061764,
-        0.011534944232553239,
-        0.0680388870239258,
-        0.9998,
-        0.022567954233288764,
-        0.03644986734837294,
-        0.0278723166257143,
-        0.012022978414595133,
-        0.017261751674115658,
-        0.1018139317035675,
-        0.9998,
-        0.030176736226677894,
-        0.048319679215550415,
-        0.037043299572169786,
-        0.015823813125491137,
-        0.022935147598385802,
-        0.13534477652311327,
-        0.9998,
-        0.03749606138020754,
-        0.05974762381315232,
-        0.046320767011493444,
-        0.01970605190098286,
-        0.028365900403261188,
-        0.16783747158050544,
-        0.9998,
-        0.04486110335588456,
-        0.07147922609001398,
-        0.055334820427000525,
-        0.023239636448025694,
-        0.034125556316971783,
-        0.2017985020875931,
-        0.9998,
-        0.052173979285359384,
-        0.08346393676698208,
-        0.06446339850723744,
-        0.026847184759378445,
-        0.03955972051620485,
-        0.23437213500142096,
-        0.9998,
-        0.059150983133912094,
-        0.09537746536731717,
-        0.0733001764550805,
-        0.029705484813451755,
-        0.04516481494158506,
-        0.26646220774650575,
-        0.9998,
-        0.06645039129257202,
-        0.1076395375341177,
-        0.08239804291725158,
-        0.0328832214474678,
-        0.04995994846820831,
-        0.29610841860771175,
-        0.9998,
-        0.07374503636360169,
-        0.11985380845069887,
-        0.09145532561540604,
-        0.035987181842327104,
-        0.05490816103816032,
-        0.3266934569597244,
-        0.9998,
-        0.08122003978192807,
-        0.1319003387391567,
-        0.10024555467069149,
-        0.038626870799064655,
-        0.05953236908912657,
-        0.35580547877550117,
-        0.9998,
-        0.08822669934630395,
-        0.14304091303348543,
-        0.1088144795835018,
-        0.041924731212854394,
-        0.06486271733045579,
-        0.3876368781208992,
-        0.9998,
-        0.09569223463535309,
-        0.1534084576368332,
-        0.11751553359031677,
-        0.04469002693891522,
-        0.06991170048713685,
-        0.41592443135976787,
-        0.9998,
-        0.10324303380250932,
-        0.1640547949969769,
-        0.1258540060549974,
-        0.04742067799866201,
-        0.0741334307938814,
-        0.4491768553853035,
-        0.9998,
-        0.1099683414697647,
-        0.17510256106853483,
-        0.13403582906574008,
-        0.050331409102678304,
-        0.07791369587183,
-        0.475221610414982,
-        0.9998,
-        0.1170920331031084,
-        0.1857688842058181,
-        0.14203765797317025,
-        0.05343856880664827,
-        0.0820521941632032,
-        0.5052882493734359,
-        0.9998,
-        0.12414569870531561,
-        0.19753187570571906,
-        0.14974090793728828,
-        0.0557895161241293,
-        0.0865580612182617,
-        0.5326116788625717,
-        0.9998,
-        0.13112460071146492,
-        0.20844777574539186,
-        0.15755365291237833,
-        0.05866813796758652,
-        0.09100452079772947,
-        0.5617906215190887,
-        0.9998,
-        0.1379249924659729,
-        0.21917983024120336,
-        0.16431246396899224,
-        0.06069063851833345,
-        0.09500974670350554,
-        0.5912433532238006,
-        0.9998,
-        0.14485807346105578,
-        0.2286938194990158,
-        0.1716713245987892,
-        0.0637721403002739,
-        0.09919737915396692,
-        0.6155789003372192,
-        0.9998,
-        0.15117210847139356,
-        0.23827072948217387,
-        0.1779224501848221,
-        0.06597466281652448,
-        0.10370546174049378,
-        0.644374561071396,
-        0.9998,
-        0.15787530031800268,
-        0.24751081448793422,
-        0.18454510100185867,
-        0.06831172816753389,
-        0.10792329218983648,
-        0.6659029854536058,
-        0.9998,
-        0.1641180765330792,
-        0.2572556215524673,
-        0.19053607739508155,
-        0.07030175954103468,
-        0.11277426195144652,
-        0.686683092904091,
-        0.9998,
-        0.16966928005814552,
-        0.26658033823966976,
-        0.19653976464271544,
-        0.07203448786735533,
-        0.11643967783451081,
-        0.7125654235839844,
-        0.9998,
-        0.17621982062458996,
-        0.27701201248764995,
-        0.20197029424905777,
-        0.07416592988967896,
-        0.12035323338508608,
-        0.7360334300994871,
-        0.9998,
-        0.1819627068042755,
-        0.2855635068178177,
-        0.2084233057498932,
-        0.07603480890989306,
-        0.1241462073802948,
-        0.7568318559646605,
-        0.9998,
-        0.18748426148295405,
-        0.29386607773303985,
-        0.21488302874565124,
-        0.0777944794893265,
-        0.1279024946689606,
-        0.7800548309087754,
-        0.9998,
-        0.19402539111375805,
-        0.304579441523552,
-        0.22076346032619476,
-        0.07946800847053526,
-        0.13082660132646562,
-        0.8035489074707032,
-        0.9998,
-        0.1999460756778717,
-        0.31222576675415037,
-        0.2264517953634262,
-        0.08081280782818795,
-        0.13460262061357497,
-        0.828994060134888,
-        0.9998,
-        0.20509765752553943,
-        0.3203624894678593,
-        0.23198571143150332,
-        0.08211237077713013,
-        0.13750549141168594,
-        0.8529169931411746,
-        0.9998,
-        0.209518383550644,
-        0.3281992249488831,
-        0.23816398755311968,
-        0.08322777889966959,
-        0.14037072820067403,
-        0.8746512705802916,
-        0.9998,
-        0.21442690343856818,
-        0.33666496825218195,
-        0.24309153485894203,
-        0.08529254732131952,
-        0.14304055777192115,
-        0.8924013189554216,
-        0.9998,
-        0.21941403126716608,
-        0.3448349338173866,
-        0.2489651171147823,
-        0.08667572829127312,
-        0.14580873468518263,
-        0.9107246013164523,
-        0.9998,
-        0.22309734332561493,
-        0.3517628137350083,
-        0.2547237434387207,
-        0.08896696496009826,
-        0.1489701636433602,
-        0.9272562242746352,
-        0.9998,
-        0.22793451219797134,
-        0.3591047751426697,
-        0.2597683270871639,
-        0.09042025351524352,
-        0.1528471918106079,
-        0.9483436929702762,
-        0.9998,
-        0.23258997063636777,
-        0.36902411174774175,
-        0.26485966441631315,
-        0.09243017539381981,
-        0.15506044900417326,
-        0.9667170533418656,
-        0.9998,
-        0.23665347018241878,
-        0.3762736029624939,
-        0.27067578415870663,
-        0.09449174585938458,
-        0.15888449568748475,
-        0.9834513599395751,
-        0.9998,
-        0.24093078261613854,
-        0.3843366762638093,
-        0.2768812180876732,
-        0.09594143676757816,
-        0.16144601993560792,
-        1.0022019856452942,
-        0.9998,
-        0.2459272434949874,
-        0.3907656828403474,
-        0.2815030888676643,
-        0.09760697330236434,
-        0.16464969681501385,
-        1.019599228835106,
-        0.9998,
-        0.2511993800699711,
-        0.3962667070865631,
-        0.28746061475872997,
-        0.09892506269812584,
-        0.1685606077313423,
-        1.0345355377674106,
-        0.9998,
-        0.25586065447330475,
-        0.4064154948949814,
-        0.29352941393256193,
-        0.10010581925511364,
-        0.17137197246551505,
-        1.0455808718204502,
-        0.9998,
-        0.26066332874298104,
-        0.41295039305686965,
-        0.29917095521688464,
-        0.10042237902879714,
-        0.17334867343306548,
-        1.0644763714551928,
-        0.9998,
-        0.26424761767387395,
-        0.4171489597439766,
-        0.3047941813468933,
-        0.10030385837554934,
-        0.1771283437609672,
-        1.0775494430065153,
-        0.9998,
-        0.266823394203186,
-        0.42120621981620787,
-        0.31085526600480073,
-        0.10133219141960148,
-        0.18098006631135943,
-        1.0927770781517028,
-        0.9998,
-        0.27192984950542454,
-        0.4258830359578132,
-        0.31648782253265384,
-        0.10210902032852165,
-        0.18391375703215596,
-        1.1100240562438963,
-        0.9998,
-        0.2774573613643646,
-        0.4328473631143568,
-        0.3229170823454858,
-        0.10328457579612738,
-        0.18559744659066202,
-        1.127002547836304,
-        0.9998,
-        0.280809062719345,
-        0.43832696628570567,
-        0.32813225684762004,
-        0.10458858071565635,
-        0.18929516908526423,
-        1.1399588096141815,
-        0.9998,
-        0.28525815182924275,
-        0.44289318814277645,
-        0.3333625353813172,
-        0.1055313608527183,
-        0.1918839191913605,
-        1.1545847342491151,
-        0.9998,
-        0.2899247071444988,
-        0.4499819297909735,
-        0.33962785017490393,
-        0.10614367849826811,
-        0.19452174307107928,
-        1.1747124679088592,
-        0.9998,
-        0.29274274289608004,
-        0.4549682779312134,
-        0.3463232246458531,
-        0.10680567473173141,
-        0.19644608254432672,
-        1.185605804681778,
-        0.9998
-      ]
-    }
-  },
-  "sample_ratio": 0.05,
-  "dataset_infos": {
-    "robotgeneralist/pp__1": {
-      "commit_hash": "f1c11292bf450c33ec285da885c5ed1a4d0e4b9b",
-      "num_episodes": 237,
-      "num_frames": 99363
-    },
-    "robotgeneralist/pp__1__adversarial": {
-      "commit_hash": "a8b6024ff9f5261e87d553f35f07f16487adabc1",
-      "num_episodes": 64,
-      "num_frames": 17906
-    },
-    "robotgeneralist/pp__2": {
-      "commit_hash": "f244ff639c1a837fdec7990c601bf5ed4fbb56a9",
-      "num_episodes": 92,
-      "num_frames": 37458
-    },
-    "robotgeneralist/pp__2__rotations": {
-      "commit_hash": "9b8e02337fce6bccda7adcc4aae1edcbe5b3de41",
-      "num_episodes": 102,
-      "num_frames": 13935
-    },
-    "robotgeneralist/pp__3__with_ft": {
-      "commit_hash": "68023819f92217d1239bf1dc1229b73a32580513",
-      "num_episodes": 99,
-      "num_frames": 38424
-    },
-    "robotgeneralist/pp__3__with_ft__rotations": {
-      "commit_hash": "062188051c5f78b03d51848dc0c03f93bc4a78c5",
-      "num_episodes": 49,
-      "num_frames": 7151
-    },
-    "robotgeneralist/pp__3__with_ft__adversarial_start": {
-      "commit_hash": "d0e27c579029e82f626c9ba8817ca43c610fa28e",
-      "num_episodes": 147,
-      "num_frames": 21313
-    },
-    "robotgeneralist/pp__4__with_ft": {
-      "commit_hash": "0d64bf5c9bac34d9cce59175a0becf822366bc53",
-      "num_episodes": 94,
-      "num_frames": 33046
-    },
-    "robotgeneralist/pp__5__with_ft__adversarial_target": {
-      "commit_hash": "8629d43a975c254367b84f491481fa8574416c6e",
-      "num_episodes": 48,
-      "num_frames": 10615
-    },
-    "robotgeneralist/pp__6__with_ft": {
-      "commit_hash": "e2a7c2ded8aed6663aa71fd9a1e6708d636071b1",
-      "num_episodes": 379,
-      "num_frames": 123611
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-1": {
-      "commit_hash": "defdc7ebf6871fc057ef30d21a2f71bfe3a40ce4",
-      "num_episodes": 98,
-      "num_frames": 41410
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-2": {
-      "commit_hash": "73551fdf12a436f983537d0454d9a6046a51c2a7",
-      "num_episodes": 94,
-      "num_frames": 37859
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-3": {
-      "commit_hash": "ae7abf3c61cb6d4f3096dc30b67731305c50d59c",
-      "num_episodes": 127,
-      "num_frames": 53341
-    },
-    "robotgeneralist/precise-packing-2025-10-22-part-4": {
-      "commit_hash": "5d68ce10ce1c5a492bfc760012aba4793a5d170b",
-      "num_episodes": 102,
-      "num_frames": 38263
-    }
-  }
-}
\ No newline at end of file
diff --git a/pp__ee_baseline/5000/params/_METADATA b/pp__ee_baseline/5000/params/_METADATA
deleted file mode 100644
index 9eb73624158f4514d49041e0806d2d702c62eee1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/_METADATA
+++ /dev/null
@@ -1 +0,0 @@
-{"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, "write_shape": [1152]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304]}}, "('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, "write_shape": [27, 1152, 4304]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 4304, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 1152]}}, "('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, "write_shape": [27, 16, 72, 1152]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [27, 16, 72]}}, "('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, "write_shape": [27, 1152, 16, 72]}}, "('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, "write_shape": [1152]}}, "('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, "write_shape": [14, 14, 3, 1152]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1152, 2048]}}, "('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, "write_shape": [1, 256, 1152]}}, "('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, "write_shape": [257152, 2048]}}, "('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, "write_shape": [2048]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [18, 8, 256, 2048]}}, "('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, "write_shape": [18, 8, 256, 1024]}}, "('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, "write_shape": [18, 2, 1, 2048, 256]}}, "('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, "write_shape": [18, 2, 1, 1024, 256]}}, "('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, "write_shape": [18, 8, 2048, 256]}}, "('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, "write_shape": [18, 8, 1024, 256]}}, "('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, "write_shape": [18, 2, 2048, 16384]}}, "('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, "write_shape": [18, 16384, 2048]}}, "('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, "write_shape": [18, 2, 1024, 4096]}}, "('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, "write_shape": [18, 4096, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [18, 2048]}}, "('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, "write_shape": [18, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}, "('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, "write_shape": [32]}}, "('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, "write_shape": [1024, 32]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [2048, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [1024, 1024]}}, "('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, "write_shape": [1024]}}, "('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, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/pp__ee_baseline/5000/params/_sharding b/pp__ee_baseline/5000/params/_sharding
deleted file mode 100644
index 924608abdcb99abb3c864bb9d1bfca6eb60f780e..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/_sharding
+++ /dev/null
@@ -1 +0,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}]]}}"}
\ No newline at end of file
diff --git a/pp__ee_baseline/5000/params/array_metadatas/process_0 b/pp__ee_baseline/5000/params/array_metadatas/process_0
deleted file mode 100644
index 6aa9fd1decf72607133eb54664d98802bf563b5c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/array_metadatas/process_0
+++ /dev/null
@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/pp__ee_baseline/5000/params/d/0d624ba20c775e5366d00f868da2785d b/pp__ee_baseline/5000/params/d/0d624ba20c775e5366d00f868da2785d
deleted file mode 100644
index a28c319afedc8582a0a61ad9b0a5844e38497f30..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/d/0d624ba20c775e5366d00f868da2785d and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/d/272f5dd8e3f2cc688f2390e612b06ad4 b/pp__ee_baseline/5000/params/d/272f5dd8e3f2cc688f2390e612b06ad4
deleted file mode 100644
index 0e3fdbe8ca5f70d77c9f96056d9010762b183c7d..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/d/272f5dd8e3f2cc688f2390e612b06ad4 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/manifest.ocdbt b/pp__ee_baseline/5000/params/manifest.ocdbt
deleted file mode 100644
index f76f2532607e2ce2afa92056cc638a620bf4db2b..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/01e0f3530fe883986482ed66efbfbf3c b/pp__ee_baseline/5000/params/ocdbt.process_0/d/01e0f3530fe883986482ed66efbfbf3c
deleted file mode 100644
index 5d0e02fed181162c85c5b701fedae0dd79041e37..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/01e0f3530fe883986482ed66efbfbf3c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0822c12d5019e6fd87e29543c3997a5f38b9babf9febb35db3963fb21fca0f7c
-size 26051997
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/091feb92cd5fe1bc24048612ed6073d8 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/091feb92cd5fe1bc24048612ed6073d8
deleted file mode 100644
index 0911dc95f15bad2fc2f6ef8b2010d7487c0a482c..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/091feb92cd5fe1bc24048612ed6073d8
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3c34f6a0793a49bb754060114a872fbeeaa267f048edd3eabca91130a6c22ec3
-size 280188447
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/136ad58a740fc189e433469579b311e9 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/136ad58a740fc189e433469579b311e9
deleted file mode 100644
index c94fb4e8281ed5e74e863e620e8fdbb4a60ad42d..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/136ad58a740fc189e433469579b311e9 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/152d849dc69362e369e705290b7219c5 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/152d849dc69362e369e705290b7219c5
deleted file mode 100644
index eacd3c44dd08188372288df7f7b763b724855fd4..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/152d849dc69362e369e705290b7219c5
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bf31719ae79f02a1c27fd18d9b9219f26ca748153e5b0921547d0f6a423bde16
-size 545658956
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/16844be3148f756dd83e62b229214cd0 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/16844be3148f756dd83e62b229214cd0
deleted file mode 100644
index 3b765dae0415e048c52799262cd34b40ae50f8f2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/16844be3148f756dd83e62b229214cd0
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f9ac3c5857bb474fa4d22ffddfd5c111c5133732112aed80e640bb8141d07684
-size 1957442684
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/20f5e60ff55481fbb393b56016f2cf42 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/20f5e60ff55481fbb393b56016f2cf42
deleted file mode 100644
index 2a70538817c1baa8385dba77564573d6708e9ff7..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/20f5e60ff55481fbb393b56016f2cf42
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1e0b48476921f7539f5f918b395154bd1647e785cb865c9f48fa895b77053dd2
-size 139912514
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/26c41476e8dcd0ed98e27f854ea2bdf2 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/26c41476e8dcd0ed98e27f854ea2bdf2
deleted file mode 100644
index d44c01f04950bccb1e78d853ca27039fa0d35c34..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/26c41476e8dcd0ed98e27f854ea2bdf2
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4028e412320b36b0e99c32e8147f282177c9141749244208d669ae63e951f15b
-size 1120112042
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/2809120a934c5368a82267613f7eb65f b/pp__ee_baseline/5000/params/ocdbt.process_0/d/2809120a934c5368a82267613f7eb65f
deleted file mode 100644
index bbd0352b6e9520e90ee29a7e6d305b9ea42771da..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/2809120a934c5368a82267613f7eb65f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9d92c1cdef3d84cafa0377b570d62149c7a9e2757baa758f7212fcb385adc645
-size 34975155
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/372dbf2c9d1e27bd959bcfa5abfbc0ad b/pp__ee_baseline/5000/params/ocdbt.process_0/d/372dbf2c9d1e27bd959bcfa5abfbc0ad
deleted file mode 100644
index 5b929e736231420502e0b657e009c344dd9c1a17..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/372dbf2c9d1e27bd959bcfa5abfbc0ad and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/387a628d90eeda4cdf24aba5c776fa9b b/pp__ee_baseline/5000/params/ocdbt.process_0/d/387a628d90eeda4cdf24aba5c776fa9b
deleted file mode 100644
index 3e49aaaa1586e3b9ad37aa464584d43bd4669ad2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/387a628d90eeda4cdf24aba5c776fa9b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e2b0629d3be5ab8344d898d95f1053e5b809f1ad624aa4bf635a7a696d444bc3
-size 279236681
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/41dac0d0689c501a2caa61a788d0d7fc b/pp__ee_baseline/5000/params/ocdbt.process_0/d/41dac0d0689c501a2caa61a788d0d7fc
deleted file mode 100644
index 7e07e5262fd3ed7b7c38fb6f6ddad91a9ac77b79..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/41dac0d0689c501a2caa61a788d0d7fc
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:99b261aa5b198ab7ce9db5efc6f69a4c0bb5811c8e976d9b05ec17a6a72c30da
-size 122062
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/4714b5da5731936f43357e22ad45c8da b/pp__ee_baseline/5000/params/ocdbt.process_0/d/4714b5da5731936f43357e22ad45c8da
deleted file mode 100644
index b74398bb6240a43dd65dbcba7787dafba318abef..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/4714b5da5731936f43357e22ad45c8da
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d54d81493327899e069671edaad393d5dd8ab1dd0511643efdfc4f9128a399e3
-size 34975690
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/4977635ce3f9431137f847b675fe9127 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/4977635ce3f9431137f847b675fe9127
deleted file mode 100644
index c4bb03976fe6b7f31ca882e322c318c478a97620..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/4977635ce3f9431137f847b675fe9127
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:78977c6d57c64efb0ef184e6f98a4617363fa93c00a82c3e0a1187dc9e61949f
-size 117983
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/505303792b8f36a97c41157e8986c41b b/pp__ee_baseline/5000/params/ocdbt.process_0/d/505303792b8f36a97c41157e8986c41b
deleted file mode 100644
index 99b41ad24f0fe357019601c1c20ae918aef300bb..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/505303792b8f36a97c41157e8986c41b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:548c1dd67295271f86265e62afe6aee26d5dd5af1250d6bd632df5422dc6a639
-size 559731189
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/5adacdf0a500609f3ecb114a3c0af22e b/pp__ee_baseline/5000/params/ocdbt.process_0/d/5adacdf0a500609f3ecb114a3c0af22e
deleted file mode 100644
index 069121c04a8b94642276f01a7c2bde6297e37513..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/5adacdf0a500609f3ecb114a3c0af22e and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/5cc07f257c8b9c964b3b25be9219678d b/pp__ee_baseline/5000/params/ocdbt.process_0/d/5cc07f257c8b9c964b3b25be9219678d
deleted file mode 100644
index 895165ec92261fdf4baa07961681208d1de0f9cb..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/5cc07f257c8b9c964b3b25be9219678d and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/620737092048de6f7232e2be686807fb b/pp__ee_baseline/5000/params/ocdbt.process_0/d/620737092048de6f7232e2be686807fb
deleted file mode 100644
index 7d5d9412d74b89afa080b00d10439a1f2620bcb8..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/620737092048de6f7232e2be686807fb
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:eabfe94059b9ccc6ff21998db1fb327f84a19145f63c9991eb3cde304ab18e37
-size 132968547
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/6343c989767dfda020ac6c5aa33edf96 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/6343c989767dfda020ac6c5aa33edf96
deleted file mode 100644
index 18f98df64b52a9514440ab8d3e90410ac0bdbb46..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/6343c989767dfda020ac6c5aa33edf96
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bbf5096379f5e4d057bf487b4d69ee583ae76deed579761ad21cb6e59e79568a
-size 1957442621
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/680aa7f1b42db1fb4349832e05ae0dad b/pp__ee_baseline/5000/params/ocdbt.process_0/d/680aa7f1b42db1fb4349832e05ae0dad
deleted file mode 100644
index 3ebcb31605f262669c027795943aa26248e75f24..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/680aa7f1b42db1fb4349832e05ae0dad
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b17295974843bb2c17b5e6abe82b2a62bda197a1aa45d5802fc053cd183f7c57
-size 2240237241
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/69d5ef58e134b7de8c1d18e7d2de4b23 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/69d5ef58e134b7de8c1d18e7d2de4b23
deleted file mode 100644
index 7cc94866208872d0aad7ae82ba521a96a97a2629..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/69d5ef58e134b7de8c1d18e7d2de4b23 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/764dcce523c3443d46304a4ef35b6471 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/764dcce523c3443d46304a4ef35b6471
deleted file mode 100644
index adfe8c7aed2c550b3123df1a8d6757b0eedfe752..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/764dcce523c3443d46304a4ef35b6471
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8df1dd41972cacbb92bb896994d635336fbe134b9de3bdeedcb390f29d4c462a
-size 132969793
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/777df4d0d4c05dd906855bd52bea189e b/pp__ee_baseline/5000/params/ocdbt.process_0/d/777df4d0d4c05dd906855bd52bea189e
deleted file mode 100644
index e254bf1dd4a04a60a066facdfa2371dab1b10899..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/777df4d0d4c05dd906855bd52bea189e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:647c21919e0bf7666e85699e8dc470a98af89252e6ba0969e0f4aa1689fdd044
-size 70086687
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/7ea5a7ca0cf497a0860daf93ef3ddf9c b/pp__ee_baseline/5000/params/ocdbt.process_0/d/7ea5a7ca0cf497a0860daf93ef3ddf9c
deleted file mode 100644
index b120411e4a08ac68ca144c348726629a29064cd1..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/7ea5a7ca0cf497a0860daf93ef3ddf9c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a947ae1a57c3fe1f53c42657304eaf58ca742d77ddb3aaa3eaac135e6436eb5e
-size 280190943
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/811207a7cd9ad977706f67cfcc6dd9c4 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/811207a7cd9ad977706f67cfcc6dd9c4
deleted file mode 100644
index d0f79820b75b341e4331664649fa215bb9e9d450..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/811207a7cd9ad977706f67cfcc6dd9c4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8ce0e31e95e9b740214a3e06147329cc3d28fd099329f367b2828a97b7583f9a
-size 559732373
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/818efefb0862ebfc20a1d00926bcefa8 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/818efefb0862ebfc20a1d00926bcefa8
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/818efefb0862ebfc20a1d00926bcefa8 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/8b721937a83d2ee7a10a7dd7730cecb2 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/8b721937a83d2ee7a10a7dd7730cecb2
deleted file mode 100644
index ac3749b4459f99564b427b5a52ebed7f038f5b95..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/8b721937a83d2ee7a10a7dd7730cecb2
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8f23e6e1c50469418ff3c353c5615077dc0fff7caa05ffd5f1ab9ed80d2bc524
-size 203021339
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/8f50fa56180994b6d666988ad000393a b/pp__ee_baseline/5000/params/ocdbt.process_0/d/8f50fa56180994b6d666988ad000393a
deleted file mode 100644
index ecf60b47541338868130420001f2c52d88f01cf5..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/8f50fa56180994b6d666988ad000393a and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/91fb95bce57976a54ef2f4169a774197 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/91fb95bce57976a54ef2f4169a774197
deleted file mode 100644
index fbc39793736f34f132a44a4563929e8683c32317..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/91fb95bce57976a54ef2f4169a774197 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a2d7dbb58755073c358cc161939d3db9 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/a2d7dbb58755073c358cc161939d3db9
deleted file mode 100644
index 29c017f93d3e98caf1fb0f530348a02bc8b9fb45..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a2d7dbb58755073c358cc161939d3db9
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4233b0cd0fe4e0417c28bfd497e82407754c8fee0c8c790097ad01f2b168fcf5
-size 279235231
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7350e2e718ad5dea870c1cc1b41d377 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7350e2e718ad5dea870c1cc1b41d377
deleted file mode 100644
index 486de2e639b38c4b82739f91ca3ab3d87f0aa93b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7350e2e718ad5dea870c1cc1b41d377
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7128149510c1b0f8b46edb7c3306d15342ef62be214348742f735f890190f20e
-size 139973807
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7aec432342f6c246df9218d07f7cd7f b/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7aec432342f6c246df9218d07f7cd7f
deleted file mode 100644
index 8393dc474da31851de9d51b205ef791387378e8b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/a7aec432342f6c246df9218d07f7cd7f
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1a7f31e7f6e665308269937b25e95f85af090e80da25c5188eceb50af2ede224
-size 132846844
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/b00fe577b925219985168204a01f0f77 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/b00fe577b925219985168204a01f0f77
deleted file mode 100644
index f54f3a8a436576640347c7cac825e19ea08e9b66..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/b00fe577b925219985168204a01f0f77
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a0c4daf089b540570c11b81f74e47e4280234c5114232d18a31e01a767578c2b
-size 132967024
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/b8974e3cf6e81673db5634dc9068e338 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/b8974e3cf6e81673db5634dc9068e338
deleted file mode 100644
index 3aa938a11c70e30f0625d32ed5ed15146678e853..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/b8974e3cf6e81673db5634dc9068e338
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5f5283d6b9cbd5a78e23ed29da09e8f30c6b703026b32c6ea342e1ac70a69c63
-size 121830
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c065fde40fb813bf40c51ce0f0420970 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/c065fde40fb813bf40c51ce0f0420970
deleted file mode 100644
index 6f3a8c8b01f1eaae23ad09c740dccdda8ec35ece..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c065fde40fb813bf40c51ce0f0420970
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c2a22de8f9d6870c4e2664170230e36ced9f8313003e1a9521b8f15c2655ae4a
-size 280037836
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c0aba72871ac2a6ab22033491f15512b b/pp__ee_baseline/5000/params/ocdbt.process_0/d/c0aba72871ac2a6ab22033491f15512b
deleted file mode 100644
index c5f1da0a7e993262314a89983579cb7467922a92..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c0aba72871ac2a6ab22033491f15512b
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6f2339b1bb813656744c11f03a0b21992ecc998a9df3dd9e03aa745a2879d3e3
-size 497031628
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c7145e07a788653b4c57654c8967e796 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/c7145e07a788653b4c57654c8967e796
deleted file mode 100644
index 7d908162e97c188a43f99ce4830715f29ca92c80..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/c7145e07a788653b4c57654c8967e796
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7b7acdf7ca04606fd94858da563c84f9211fb8bda8bfa34b23c9fa2532d5f555
-size 497034808
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d0646980a9322a12cd5b98b542f3e225 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/d0646980a9322a12cd5b98b542f3e225
deleted file mode 100644
index 9644a109cc5bf83d99f7df20701b86911985877a..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d0646980a9322a12cd5b98b542f3e225
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:020d3927725fc9f81a1fbd9a88faad48c0eda7f53a1706bc5eac02151943d1a6
-size 2240238287
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d671f8eab5297646965c6d938999220c b/pp__ee_baseline/5000/params/ocdbt.process_0/d/d671f8eab5297646965c6d938999220c
deleted file mode 100644
index 19c4acdfe9d67b71142e4b4764776a8a900d7ca3..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d671f8eab5297646965c6d938999220c
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:902fd17f1e65eb2df0a9804219a3fc5f231a1ecce7baeda07dae8d3c433d3f56
-size 2240237020
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d69c9b569d41231113afc2b71ddbb816 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/d69c9b569d41231113afc2b71ddbb816
deleted file mode 100644
index 39069a05b12845c53d863efaadbabf135c5efbcd..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d69c9b569d41231113afc2b71ddbb816
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:27121c3bcd58cf869cb60af60982a851d0b9ee517fe6ee1cfa540b94f16037d1
-size 496754771
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d7c1d538d78b8e51d894d0a379b768a0 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/d7c1d538d78b8e51d894d0a379b768a0
deleted file mode 100644
index ffc57c5cefe3160b4f5ec2c8030ab168d9324180..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/d7c1d538d78b8e51d894d0a379b768a0 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/dcc80efc3fa7d93d8a950293a48a5fcc b/pp__ee_baseline/5000/params/ocdbt.process_0/d/dcc80efc3fa7d93d8a950293a48a5fcc
deleted file mode 100644
index b8370bfb4f8d7dd377ac2b4d5b57a9d00005b336..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/dcc80efc3fa7d93d8a950293a48a5fcc
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:270d80e0807b61367586d2eb94a0511de52102537253b4a855407d7e90898299
-size 2240219617
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/dd53aa6912126879ddef3b744aa12a5c b/pp__ee_baseline/5000/params/ocdbt.process_0/d/dd53aa6912126879ddef3b744aa12a5c
deleted file mode 100644
index 7e10ba956fa757d7599b479ce408807eebefc1a3..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/dd53aa6912126879ddef3b744aa12a5c and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/df8661c636ac76802fc5eaae4c3c7576 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/df8661c636ac76802fc5eaae4c3c7576
deleted file mode 100644
index 32673a4f8e96a4331527f3505a78fd829f83294b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/df8661c636ac76802fc5eaae4c3c7576
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:74db0f8fd5317ab1910c14c170171cb75069d9302c4e6bdb54bcb7a77b456ca7
-size 496754711
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e025c85ce83708124508c236729b5277 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/e025c85ce83708124508c236729b5277
deleted file mode 100644
index cc4d3d11d4986dadaa2f478706e869ee24423028..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e025c85ce83708124508c236729b5277
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e75e8151b615d431f640d5bb5ce0990c7478506d61ba3989dc06d81703cc5148
-size 26176077
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e55295e59db570c9ab43ce082fb67de4 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/e55295e59db570c9ab43ce082fb67de4
deleted file mode 100644
index 7d0ce1532bbdfeaf392034710ed5567ccb23b685..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e55295e59db570c9ab43ce082fb67de4
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3fc8d133dbba72532521e98b9d06e782c8b0384dd603f253032486030396c863
-size 132967707
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e92a50a91e4c5bd11e8577f75087c6fb b/pp__ee_baseline/5000/params/ocdbt.process_0/d/e92a50a91e4c5bd11e8577f75087c6fb
deleted file mode 100644
index 441ae92a7980e3d5084d228db6a15fb867bd121b..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/e92a50a91e4c5bd11e8577f75087c6fb
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0c759c65e4b2ac70de76e73aabafefabebb21c4395e6389e4680cda2bfef41ef
-size 1120107404
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/ebd7e4fda61d3f32be8c9f289a7ca992 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/ebd7e4fda61d3f32be8c9f289a7ca992
deleted file mode 100644
index bd865b3e932bcf29615819f5d0b12923d2866736..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/ebd7e4fda61d3f32be8c9f289a7ca992
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e2c48e212fc64d396bf64fd3e81d3e32749826dbffe17697137acc9a6135fa85
-size 280037709
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/ec4d73754b512fd97572ccfb8a3842a6 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/ec4d73754b512fd97572ccfb8a3842a6
deleted file mode 100644
index 46cd46020c3fc37eed8126482d2295a61a10ce74..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/ec4d73754b512fd97572ccfb8a3842a6 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f51991e4beace02c96afc071c17c0efc b/pp__ee_baseline/5000/params/ocdbt.process_0/d/f51991e4beace02c96afc071c17c0efc
deleted file mode 100644
index 6e0606b86bad14899fd21dc09aa95e73a9c9592d..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f51991e4beace02c96afc071c17c0efc
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6bfd591c6cf3ede9f9453116e1164637442814418a012c4de3a71b3a3d879373
-size 1120128481
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f57b15ac83d595dd49b2bef8e25a975e b/pp__ee_baseline/5000/params/ocdbt.process_0/d/f57b15ac83d595dd49b2bef8e25a975e
deleted file mode 100644
index cff30ba77ad5b9ad836500dcbf9b42287ae806f2..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f57b15ac83d595dd49b2bef8e25a975e
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4bfeeb8ee2b3cc9dfcfc007b578d3e5b56ed3625844e94536867d3d643cd7824
-size 1120114490
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f5fa49d7193f8da0dd0de2baf104ec23 b/pp__ee_baseline/5000/params/ocdbt.process_0/d/f5fa49d7193f8da0dd0de2baf104ec23
deleted file mode 100644
index ff99512cbc02fb16e6df4474c10c00cb18732639..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/d/f5fa49d7193f8da0dd0de2baf104ec23 and /dev/null differ
diff --git a/pp__ee_baseline/5000/params/ocdbt.process_0/manifest.ocdbt b/pp__ee_baseline/5000/params/ocdbt.process_0/manifest.ocdbt
deleted file mode 100644
index 5c9ca2b5ddc51ce9fcd5a135e4d5b061abc2a9d3..0000000000000000000000000000000000000000
Binary files a/pp__ee_baseline/5000/params/ocdbt.process_0/manifest.ocdbt and /dev/null differ
diff --git a/pp__ee_baseline/config.yaml b/pp__ee_baseline/config.yaml
deleted file mode 100644
index 5b258ea870fba2affb76ea46f03e63c405d146e6..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/config.yaml
+++ /dev/null
@@ -1,103 +0,0 @@
-!!python/object:openpi.training.config.TrainConfig
-name: precise_packing_baseline
-project_name: openpi
-exp_name: pp__ee_baseline
-model: !!python/object:openpi.models.pi0_config.Pi0Config
-    action_dim: 32
-    action_horizon: 50
-    max_token_len: 48
-    dtype: bfloat16
-    paligemma_variant: gemma_2b
-    action_expert_variant: gemma_300m
-    img_variant: So400m/14
-    action_loss_horizon: null
-    pi05: false
-    discrete_state_input: false
-weight_loader: !!python/object:openpi.training.weight_loaders.CheckpointWeightLoader
-    local_dir: null
-    repo_id: robotgeneralist/openpi_checkpoint_mirrors
-    revision: main
-    checkpoint_path: pi0_base
-pytorch_weight_path: null
-pytorch_training_precision: bfloat16
-lr_schedule: !!python/object:openpi.training.optimizer.CosineDecaySchedule
-    warmup_steps: 1000
-    peak_lr: 2.5e-05
-    decay_steps: 30000
-    decay_lr: 2.5e-06
-optimizer: !!python/object:openpi.training.optimizer.AdamW
-    b1: 0.9
-    b2: 0.95
-    eps: 1.0e-08
-    weight_decay: 1.0e-10
-    clip_gradient_norm: 1.0
-ema_decay: 0.99
-freeze_filter: !!python/name:flax.nnx.filterlib.Nothing ''
-data: !!python/object:openpi.training.config.LeRobotNomagicURXDataConfig
-    repo_ids: !!python/tuple
-    - robotgeneralist/pp__1
-    - robotgeneralist/pp__1__adversarial
-    - robotgeneralist/pp__2
-    - robotgeneralist/pp__2__rotations
-    - robotgeneralist/pp__3__with_ft
-    - robotgeneralist/pp__3__with_ft__rotations
-    - robotgeneralist/pp__3__with_ft__adversarial_start
-    - robotgeneralist/pp__4__with_ft
-    - robotgeneralist/pp__5__with_ft__adversarial_target
-    - robotgeneralist/pp__6__with_ft
-    - robotgeneralist/precise-packing-2025-10-22-part-1
-    - robotgeneralist/precise-packing-2025-10-22-part-2
-    - robotgeneralist/precise-packing-2025-10-22-part-3
-    - robotgeneralist/precise-packing-2025-10-22-part-4
-    assets: !!python/object:openpi.training.config.AssetsConfig
-        assets_dir: null
-        asset_id: ur10e
-    norm_stats_sample_ratio: 0.05
-    base_config: !!python/object:openpi.training.config.DataConfig
-        repo_ids: null
-        asset_id: null
-        norm_stats_data: null
-        norm_stats_sample_ratio: 1.0
-        norm_stats_dir: null
-        tolerance_s: 0.0001
-        per_action_index_norm_stats: false
-        repack_transforms: !!python/object:openpi.transforms.Group
-            inputs: !!python/tuple []
-            outputs: !!python/tuple []
-        data_transforms: !!python/object:openpi.transforms.Group
-            inputs: !!python/tuple []
-            outputs: !!python/tuple []
-        model_transforms: !!python/object:openpi.transforms.Group
-            inputs: !!python/tuple []
-            outputs: !!python/tuple []
-        use_quantile_norm: false
-        action_sequence_keys: !!python/tuple
-        - actions
-        fourier_config: null
-        prompt_from_task: true
-        rlds_data_dir: null
-        action_space: null
-        filter_dict_path: null
-    default_prompt: null
-    use_poses: true
-    end_effector_frame: true
-    per_action_index_norm_stats: true
-    fourier_config: null
-    tolerance_s: 0.0001
-assets_base_dir: ./assets
-checkpoint_base_dir: /storage_nvme_1/robotgeneralist/checkpoints
-seed: 42
-batch_size: 32
-num_workers: 2
-num_train_steps: 15001
-log_interval: 100
-save_interval: 1000
-keep_period: 5000
-overwrite: false
-resume: true
-wandb_enabled: true
-policy_metadata: null
-fsdp_devices: 1
-test_split_ratio: 0.05
-test_eval_interval: 1000
-test_eval_use_train_mode: false
diff --git a/pp__ee_baseline/wandb_id.txt b/pp__ee_baseline/wandb_id.txt
deleted file mode 100644
index ee5d1fc03aa95f904c684d8ac4f1cc2d4705f255..0000000000000000000000000000000000000000
--- a/pp__ee_baseline/wandb_id.txt
+++ /dev/null
@@ -1 +0,0 @@
-dwxxl3uw
\ No newline at end of file