Delete pp__ee_baseline

pp__ee_baseline/10000/_CHECKPOINT_METADATA

@@ -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": {}}

pp__ee_baseline/10000/assets/ur10e/norm_stats.json

@@ -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
-    }
-  }
-}

pp__ee_baseline/10000/params/_METADATA

@@ -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}

pp__ee_baseline/10000/params/_sharding

@@ -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}]]}}"}

pp__ee_baseline/10000/params/array_metadatas/process_0

@@ -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}}]}

pp__ee_baseline/10000/params/ocdbt.process_0/d/019a25f7c816d93d1532cffe87ca4fba

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:14314f9310db685f4d3fc618d4a2c6b93ca3f4a695088f85ac759032c76e7f89
-size 496756117

pp__ee_baseline/10000/params/ocdbt.process_0/d/0481fb299f75ed688ffd20ac1469caf2

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a7abdec44c731ed70672074a0d0625d6734a8a615d411c2ac1358d097a8eb106
-size 126325

pp__ee_baseline/10000/params/ocdbt.process_0/d/087551093b64c3dfcfafa3e4ed4da6e4

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f223b4b43eada56595cd73dfee8ad74bf3110871ca6b5aa61d742a5b0381b8c8
-size 132966908

pp__ee_baseline/10000/params/ocdbt.process_0/d/09a9331e13384e739fcbeb1a48f8f5fa

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b8115cdc272d324b37d65d459a1f40821bfc93798b2d39739a6965a82a2bf67a
-size 265933406

pp__ee_baseline/10000/params/ocdbt.process_0/d/12b81d51f17757c52b8603ce12bae39c

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7562bd122479ee341c6057bf3a824df1b04953e870a58a3e3a507b67c4cca033
-size 2240220466

pp__ee_baseline/10000/params/ocdbt.process_0/d/19aa52193d5a2eab86bd9aeccc15d7f6

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5213104da9a2949ae259e291f88ba6719d898bba1531e4064d5dd5072f6c67a6
-size 132846192

pp__ee_baseline/10000/params/ocdbt.process_0/d/1b5628f075cbe1dc1493c4b980515dbd

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:55cded5c66245c84035aa24c66ad4ef494af51c1ccb01ae88d3ecb773ff3b0bb
-size 420161312

pp__ee_baseline/10000/params/ocdbt.process_0/d/1cd349cbead5749a4627d7691d161319

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:855eafa051ea84cce1210b0ac7465c11050edff00cc0aace38245f05d1fb50d5
-size 497034683

pp__ee_baseline/10000/params/ocdbt.process_0/d/24698c6b3ec1ba38b1228e81eb1db105

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c506e39fb6f91cb39d2060d2b42060a2449b2507b4d35d026169bcd2442d38c2
-size 21823301

pp__ee_baseline/10000/params/ocdbt.process_0/d/331d51f3288166627f64764d6803b0bd

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4004af616b0b79a98e5e1821c789f88876d74a98d13dc3b7a76d733f94a9d052
-size 2240246012

pp__ee_baseline/10000/params/ocdbt.process_0/d/39c6cfbe2087f2d95106a08dd4156e25

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:80211421a54a866851a2a79e211f6a55d563878d37278710ec3d73f8a65f46fe
-size 210053134

pp__ee_baseline/10000/params/ocdbt.process_0/d/3fdfa849a83f6bf866207b002311048d

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:edb18f05db613d4224aed78e7f22718d46e0a9f423779179a5364a36392d2163
-size 258879

pp__ee_baseline/10000/params/ocdbt.process_0/d/40f16d0adbb2baab0da6e8bb0d87e2f0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7f4eff48c68dbbdf59b89d1eb5cae2b7662488c30235aed88f555fd69dce2afb
-size 1120134600

pp__ee_baseline/10000/params/ocdbt.process_0/d/45dd7b8a685a334d5f60d1d8d00295a1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:24d64d14967d06d580b66033cd948d4ac0e4c8b47524eeea3dc915b0ffaf5a74
-size 139914364

pp__ee_baseline/10000/params/ocdbt.process_0/d/4f0b92e628e23d7c371c772ef5ca1246

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8c4e376d59e1fea14f0a8bdf41b1237834f0203da4c055758079a76ae9d94db1
-size 559275070

pp__ee_baseline/10000/params/ocdbt.process_0/d/51a121cf733d9bf4384fe5150c563646

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:74818d180db5debb2ffbc51dabd371971f12f36efae0b1a35d91842f1547fd70
-size 559726481

pp__ee_baseline/10000/params/ocdbt.process_0/d/5f290fa84d42df457b9d8258fd5e4484

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ccc9fa94b7a28a79042578896191c821ffdb15414836089547aefdc018a1b19c
-size 1120110723

pp__ee_baseline/10000/params/ocdbt.process_0/d/64f354c787df6b5c565108c5db5e5eb0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1273eb30699f3f825ab9fb7d705361225bba8f91a269725b8583f6f1687c573b
-size 34975093

pp__ee_baseline/10000/params/ocdbt.process_0/d/6684d4aa0a30c667c448a2ce8539eb32

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8da663270a5117fc74b0e5f68652dea464b9c828d625f51f8363d89896b3f277
-size 1957442747

pp__ee_baseline/10000/params/ocdbt.process_0/d/6b1caa67c5e2e1dac4a31947b3df3f2e

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:439e1269164348713e99b9b4fd1aa77b3f51f66551b56066e18dafc7324a7ec4
-size 1120132587

pp__ee_baseline/10000/params/ocdbt.process_0/d/7006c014b0daa53c04f1493aa21c17a5

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:632a29a5a6d13bc3f1f8e9b547d6820c3e3c915e346bb7ba5b1124500a60cab2
-size 2240219627

pp__ee_baseline/10000/params/ocdbt.process_0/d/707dcc33103413a3305af91394b0ba44

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f947983d364378798c9db81d51bfd0f79b8e8473b04f9f73d83a319890882dc2
-size 1120113348

pp__ee_baseline/10000/params/ocdbt.process_0/d/71edca3654852fa173676d75b09e18d4

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a6dd324f4db35fe3c054dadba8826356ed25e0e2e250a0fc069332693cd6fe7b
-size 497031537

pp__ee_baseline/10000/params/ocdbt.process_0/d/77521ce735d761eea532dc7d664286c6

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b5f1973c8c43cfde6ec89315359186cc275435c5fe5c8493f54eeaa0f1560377
-size 1957442621

pp__ee_baseline/10000/params/ocdbt.process_0/d/7c97f4dc8dcabd69aba7f65bfe810ed2

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2b414884d3ebc3b4642045cdceb437c17ae30439fd61b6523b779fa3a7cd0920
-size 132936139

pp__ee_baseline/10000/params/ocdbt.process_0/d/80bd6202a3fce6885076f5e442529f58

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:997031a6b24f3a57c97802d9799e7211f3788c530c1e6828332291c810fe8cce
-size 34975312

pp__ee_baseline/10000/params/ocdbt.process_0/d/84f6a79ed1ad4a2765fb9c4c2d0523be

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e140372b71d30d3aa306cd20eb76892ddfa763cb465de00449f49860ce30f8a2
-size 559731600

pp__ee_baseline/10000/params/ocdbt.process_0/d/89ace7fe8fc3849793247be9d2798c7d

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:64e24e7d7e5aca72c302bddf8cc01d0b6e54e6896d394e9a7a3bda5fe8bb18b5
-size 139914428

pp__ee_baseline/10000/params/ocdbt.process_0/d/8abe8a9d9cbaa63161d184bd45d3684d

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:03f8d519f1cc5a2f7f2ed4537926a245f4a4ab1503f53b9d8cf61fa20fb52c0b
-size 132937141

pp__ee_baseline/10000/params/ocdbt.process_0/d/8ac47f752ed1e41142b2cc41bdd3824e

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6dfed161aacec3ee366d239317328e881c7ff590b1bbc84bf2920d67d2bbc444
-size 279242282

pp__ee_baseline/10000/params/ocdbt.process_0/d/93f7ce1f72c301b5a7e6efdf51768941

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b4259584767c31bd8d243e005b79d246e20d763b07b0816b875aa8f29d551840
-size 132967967

pp__ee_baseline/10000/params/ocdbt.process_0/d/a032e292a41fa056512a40d6b7065f5f

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:428707c0ca3a010ca25b608b0a7d2f10ef45d681ed69bebaf0cbbe3a294cdfa9
-size 8878986

pp__ee_baseline/10000/params/ocdbt.process_0/d/ab75d254589876883e492d81b296d044

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7785cfc27e6206a77a16d5184cf5098a1cb86a731aaf28dad0269b176e6f3f61
-size 16809448

pp__ee_baseline/10000/params/ocdbt.process_0/d/abb20658929263a54d4472dbc6d5e946

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3eca8f6259e99df25d6d9ed9693d7095a076cea11d8875901946d11085c3506c
-size 132846482

pp__ee_baseline/10000/params/ocdbt.process_0/d/c097b4f81f2052b9c41827a0d2615524

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3e2e09db0f7af143d14e013c0ce7193e90f2b134d7f07e5780c3761662f514ce
-size 4353995