diff --git a/.gitattributes b/.gitattributes index ba512abb0b3bf0f9e8c3c1369ae9932d27eee295..92f700fa6215b7eaa053c5df1d15034b4a858486 100644 --- a/.gitattributes +++ b/.gitattributes @@ -142,3 +142,28 @@ robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/cff2ef8538e74c56d998d41e robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/59e1f689146ae7f6e8a7bcfa5081eb69 filter=lfs diff=lfs merge=lfs -text robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/1ece9c71bcfb811727f86642540e443c filter=lfs diff=lfs merge=lfs -text robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/d98a87efc206def7500304e6a7d1fd4f filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/5a0478d092e8c38d6dc2286a32764170 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/220c9c33cb2564892b2a61129cd932bd filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/4fd7e8b1e5e6559336e2f2979ea4ad9f filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/00af7efa6854d2978cea03d9cba221b9 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/3c278f1430593277115ac72b6462768c filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6c882add699c3c540d47c1b320e168c3 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/07fb6438160dd3701f5530cefdd0457b filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9869d9caca47dfda0dc5b4040ed37d3 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/f8b773fe4952a7147d6c60500fd33ea0 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/8eaed967dc675b0e6fcc0ea4664a7ceb filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/62f883e4860df4efd5d871cb9b4f8ff9 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5a0a05f1cb40745f53331701f9f1dcd2 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/b4103fba59981246d8aa41973f5db056 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/d37d4bf12d12967462dd75a12ba07fda filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/76b44384dc110b74eabaf89b6c912045 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/544b5cbd6788cb5274852a3f5522ae51 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/293f5e08ce6095836eeee01ab8ff7f8d filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/9dd81c05d6a91aef3a037fccd41de5f1 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/103260ea4f910af5238783686c1e8b0e filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8096bcfab7faa6bfccb00ee709811345 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/873ddb891e2a9959613902029998f221 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/45cf9cd4afbc860169fd850ac59b9c45 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/1afbaa46434c0fd6a54f7832317a5f7c filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/a8a1680dfcbb3985189eeabd625e2cdf filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/12300672920ad9e3cba201fdb87ec852 filter=lfs diff=lfs merge=lfs -text diff --git a/robocasa_test_all_atomic/20000/_CHECKPOINT_METADATA b/robocasa_test_all_atomic/20000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..f75862780518e24afd14df527078db1dcb990679 --- /dev/null +++ b/robocasa_test_all_atomic/20000/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1775512802157034325, "commit_timestamp_nsecs": 1775513172313119259, "custom_metadata": {}} \ No newline at end of file diff --git a/robocasa_test_all_atomic/20000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json b/robocasa_test_all_atomic/20000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..a61f23a33c416ca9fb46fde8f50af856841be99a --- /dev/null +++ b/robocasa_test_all_atomic/20000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json @@ -0,0 +1,464 @@ +{ + "norm_stats": { + "state": { + "mean": [ + 2.637775182723999, + -2.0912017822265625, + 0.7007339596748352, + 0.0, + 0.0, + 0.24000521004199982, + 0.6076071262359619, + 0.258401483297348, + -0.018448200076818466, + 0.44676560163497925, + -0.26384344696998596, + 0.03587118536233902, + -0.07241316884756088, + 0.13728724420070648, + 0.03210829943418503, + -0.03232070058584213 + ], + "std": [ + 1.6751232147216797, + 1.3338454961776733, + 0.0, + 0.0, + 0.0, + 0.6626312732696533, + 0.3660806715488434, + 0.16573558747768402, + 0.2436121702194214, + 0.2089885175228119, + 0.8243448734283447, + 0.29920220375061035, + 0.3282162845134735, + 0.1679685264825821, + 0.012291335500776768, + 0.012060738168656826 + ], + "q01": [ + -1.218368652534485, + -5.65514452624917, + 0.6999506360054016, + 0.0, + 0.0, + -1.0, + 0.0, + -0.13958552385568623, + -0.5809329965710639, + -0.1109548017382622, + -0.9983834087371826, + -0.6892688079476357, + -0.745897297000885, + 0.0007815830585608637, + -0.000049362923577427054, + -0.04081762075200677 + ], + "q99": [ + 6.3111974027633675, + -0.46855325088500965, + 0.7051179245471955, + 0.0, + 0.0, + 0.9996, + 0.9998, + 0.7251633455753326, + 0.5839828933954241, + 0.8639214414954184, + 0.9963686748266221, + 0.7791513680815696, + 0.726318956565857, + 0.6806885522310082, + 0.040635910079255703, + 0.00001817415971308578 + ] + }, + "actions": { + "mean": [ + 0.026344629004597664, + 0.0013832177501171827, + -0.0021026874892413616, + 0.0, + -0.8568487763404846, + -0.009222988970577717, + -0.0020607062615454197, + -0.051031820476055145, + 0.008847072720527649, + -0.018305813893675804, + -0.0024289467837661505, + -0.330047070980072 + ], + "std": [ + 0.16712690889835358, + 0.1392730474472046, + 0.12041067332029343, + 0.0, + 0.5155678391456604, + 0.4523046910762787, + 0.38781222701072693, + 0.4078434407711029, + 0.11646749824285507, + 0.13211272656917572, + 0.1214357316493988, + 0.9439644813537598 + ], + "q01": [ + -0.0716, + -0.5571999999999999, + -0.5371999999999999, + 0.0, + -1.0, + -1.0, + -1.0, + -1.0, + -0.346, + -0.45719999999999994, + -0.4032, + -1.0 + ], + "q99": [ + 0.9996, + 0.6356000000000002, + 0.4828000000000001, + 0.0, + 0.9996, + 0.9996, + 0.9996, + 0.9996, + 0.4456, + 0.3740000000000001, + 0.3884000000000001, + 0.9996 + ] + } + }, + "sample_ratio": 0.05, + "dataset_infos": { + "robotgeneralist/AdjustToasterOvenTemperature_pretrain": { + "commit_hash": "24364b1b9664fe7a58ecfce7e61cac66bcde1209", + "num_episodes": 107, + "num_frames": 21328 + }, + "robotgeneralist/AdjustWaterTemperature_pretrain": { + "commit_hash": "f1bcd907c0510f687521a44b98edaba314548a95", + "num_episodes": 106, + "num_frames": 20953 + }, + "robotgeneralist/CheesyBread_pretrain": { + "commit_hash": "d59106dabf8827be89ea3513399b6a48c8cbf814", + "num_episodes": 101, + "num_frames": 31141 + }, + "robotgeneralist/CloseBlenderLid_pretrain": { + "commit_hash": "68fb8f0d7495e9da1382c803467c488b97b55bf0", + "num_episodes": 106, + "num_frames": 36933 + }, + "robotgeneralist/CloseCabinet_pretrain": { + "commit_hash": "b58bca4c52b1673fb6a20fd0a15e9031f5b3eb72", + "num_episodes": 105, + "num_frames": 27754 + }, + "robotgeneralist/CloseDishwasher_pretrain": { + "commit_hash": "659c06a9486fb1a0eaf37c25b35af0b3e8ff2d4e", + "num_episodes": 111, + "num_frames": 15781 + }, + "robotgeneralist/CloseDrawer_pretrain": { + "commit_hash": "ab571fcde78de907f8154bcad20101423a8ee503", + "num_episodes": 110, + "num_frames": 15670 + }, + "robotgeneralist/CloseElectricKettleLid_pretrain": { + "commit_hash": "b4b4c5967fe877790ac86c5bf9e8a0216e229105", + "num_episodes": 102, + "num_frames": 7530 + }, + "robotgeneralist/CloseFridge_pretrain": { + "commit_hash": "fb04ab4ebc39a0a23dc27140f4f7701b91b7313b", + "num_episodes": 106, + "num_frames": 26888 + }, + "robotgeneralist/CloseFridgeDrawer_pretrain": { + "commit_hash": "ed9f1b467346f600ae6f755ebc7ad0e1dd841dda", + "num_episodes": 111, + "num_frames": 14946 + }, + "robotgeneralist/CloseMicrowave_pretrain": { + "commit_hash": "61e4ea6d29e504329df7147ca563d34324ae7aa5", + "num_episodes": 105, + "num_frames": 20075 + }, + "robotgeneralist/CloseOven_pretrain": { + "commit_hash": "61b80872b74919655d4fed2fc4e8667ce248fa48", + "num_episodes": 107, + "num_frames": 18230 + }, + "robotgeneralist/CloseStandMixerHead_pretrain": { + "commit_hash": "855c0a58a8396f8855eb8e213e6bed53dcde551a", + "num_episodes": 110, + "num_frames": 11593 + }, + "robotgeneralist/CloseToasterOvenDoor_pretrain": { + "commit_hash": "3c16da6f363ad8fd2bcd811efd4a87c6ed61a82c", + "num_episodes": 110, + "num_frames": 19815 + }, + "robotgeneralist/CoffeeServeMug_pretrain": { + "commit_hash": "23157f307108f6e352b98b536e941d1086c3d8af", + "num_episodes": 108, + "num_frames": 16921 + }, + "robotgeneralist/CoffeeSetupMug_pretrain": { + "commit_hash": "76e51e454262ddeb046b015486c1fe4602a53b49", + "num_episodes": 105, + "num_frames": 23636 + }, + "robotgeneralist/LowerHeat_pretrain": { + "commit_hash": "891f233633a61ce11f4fefe95a3b5081657c0227", + "num_episodes": 110, + "num_frames": 31174 + }, + "robotgeneralist/MakeIcedCoffee_pretrain": { + "commit_hash": "aee5552cade8433a5c9282851250ee0a7c6c4717", + "num_episodes": 102, + "num_frames": 29048 + }, + "robotgeneralist/NavigateKitchen_pretrain": { + "commit_hash": "9df39eed58424fbe7945717c54549e5308c9fcc6", + "num_episodes": 503, + "num_frames": 79550 + }, + "robotgeneralist/OpenBlenderLid_pretrain": { + "commit_hash": "302d625654bdb678e877dff9b6a0d9c9dcc3fc15", + "num_episodes": 106, + "num_frames": 20124 + }, + "robotgeneralist/OpenCabinet_pretrain": { + "commit_hash": "ee50aad81bc9a115a5f00885d45046d6539646cf", + "num_episodes": 107, + "num_frames": 37492 + }, + "robotgeneralist/OpenDishwasher_pretrain": { + "commit_hash": "7dd0903b5ddf482ddedbd0f013fe5223118d47b2", + "num_episodes": 111, + "num_frames": 18086 + }, + "robotgeneralist/OpenDrawer_pretrain": { + "commit_hash": "f2abb58f7fdf0157aeeaa645791a460de44be1d8", + "num_episodes": 102, + "num_frames": 20488 + }, + "robotgeneralist/OpenElectricKettleLid_pretrain": { + "commit_hash": "5416f530c440f61466d02f6ce0bf549973c0db95", + "num_episodes": 110, + "num_frames": 10928 + }, + "robotgeneralist/OpenFridge_pretrain": { + "commit_hash": "5869139ca48dbe66f7d8dabe51d4ecaf4fb71c9d", + "num_episodes": 105, + "num_frames": 33138 + }, + "robotgeneralist/OpenFridgeDrawer_pretrain": { + "commit_hash": "ceda3eb991f74682c8bc679c348402ea4587f5c4", + "num_episodes": 107, + "num_frames": 18517 + }, + "robotgeneralist/OpenMicrowave_pretrain": { + "commit_hash": "6222929e60bbee3d135cfdc27d6663e3fc535ead", + "num_episodes": 105, + "num_frames": 26017 + }, + "robotgeneralist/OpenOven_pretrain": { + "commit_hash": "b8e0e83300e3754fdd21de46680cb2852d8905c9", + "num_episodes": 108, + "num_frames": 15555 + }, + "robotgeneralist/OpenStandMixerHead_pretrain": { + "commit_hash": "43c2fe25f56fc1904f12db1b66855c1dd422463a", + "num_episodes": 109, + "num_frames": 13411 + }, + "robotgeneralist/OpenToasterOvenDoor_pretrain": { + "commit_hash": "ad2164f3590a05d06ba5cd2aea8216ba73619cb6", + "num_episodes": 105, + "num_frames": 15469 + }, + "robotgeneralist/PackDessert_pretrain": { + "commit_hash": "63d48fe23f8fb5d5a9f603b62c38f22ad0d5a418", + "num_episodes": 108, + "num_frames": 27994 + }, + "robotgeneralist/PickPlaceCabinetToCounter_pretrain": { + "commit_hash": "a338e9f408bad681248461fe42deffb77e0ab67a", + "num_episodes": 106, + "num_frames": 20201 + }, + "robotgeneralist/PickPlaceCounterToBlender_pretrain": { + "commit_hash": "6a2aba5d6a1fe63f65f7ad40d5098e8d68b08e72", + "num_episodes": 106, + "num_frames": 38892 + }, + "robotgeneralist/PickPlaceCounterToCabinet_pretrain": { + "commit_hash": "628a13d9c15c967e6133d394913b0dfe55e8fa8e", + "num_episodes": 108, + "num_frames": 24225 + }, + "robotgeneralist/PickPlaceCounterToDrawer_pretrain": { + "commit_hash": "35ac4c271761916cde68099be0fad4f59ad68dba", + "num_episodes": 107, + "num_frames": 28225 + }, + "robotgeneralist/PickPlaceCounterToMicrowave_pretrain": { + "commit_hash": "c9720edf6282bcf99f2a76505804f81d4ac97e92", + "num_episodes": 110, + "num_frames": 42012 + }, + "robotgeneralist/PickPlaceCounterToOven_pretrain": { + "commit_hash": "f1d9b1c899de4abec2f3a11b5abd0db519f555d7", + "num_episodes": 111, + "num_frames": 32014 + }, + "robotgeneralist/PickPlaceCounterToSink_pretrain": { + "commit_hash": "8642bb8c7db762fe4b82fd0b0e26847223f83de0", + "num_episodes": 108, + "num_frames": 22410 + }, + "robotgeneralist/PickPlaceCounterToStandMixer_pretrain": { + "commit_hash": "76ca30ee317ae85519afb52a0bec1ee8d116a732", + "num_episodes": 107, + "num_frames": 25467 + }, + "robotgeneralist/PickPlaceCounterToStove_pretrain": { + "commit_hash": "bf5a50869a7ecc1050f878001b39c6cb3bb0bfe5", + "num_episodes": 108, + "num_frames": 24039 + }, + "robotgeneralist/PickPlaceCounterToToasterOven_pretrain": { + "commit_hash": "515f871b5d014f070bb221d7cd9568508db890ef", + "num_episodes": 108, + "num_frames": 24313 + }, + "robotgeneralist/PickPlaceDrawerToCounter_pretrain": { + "commit_hash": "cf62a0e481b087d531fc8cc6b49f1d697ac82d9e", + "num_episodes": 103, + "num_frames": 31819 + }, + "robotgeneralist/PickPlaceFridgeDrawerToShelf_pretrain": { + "commit_hash": "b944b9dd65fb12c6253e3254f0acd7bfb7af730f", + "num_episodes": 106, + "num_frames": 26396 + }, + "robotgeneralist/PickPlaceFridgeShelfToDrawer_pretrain": { + "commit_hash": "5381c4c9ff0e1a09cf963d537836e1060ef75565", + "num_episodes": 111, + "num_frames": 27047 + }, + "robotgeneralist/PickPlaceMicrowaveToCounter_pretrain": { + "commit_hash": "de6cc72aa95974b46fd965d66ab60629c68503a3", + "num_episodes": 112, + "num_frames": 38729 + }, + "robotgeneralist/PickPlaceSinkToCounter_pretrain": { + "commit_hash": "854d27642b1810e7576cefe465c181bbee5dfcb3", + "num_episodes": 108, + "num_frames": 26397 + }, + "robotgeneralist/PickPlaceStoveToCounter_pretrain": { + "commit_hash": "fca1c62a4f719a24a0e8b587c0422e37c7ef5756", + "num_episodes": 109, + "num_frames": 23003 + }, + "robotgeneralist/PickPlaceToasterOvenToCounter_pretrain": { + "commit_hash": "18f6af5e40443639cb1af2e32cb991d8813e016e", + "num_episodes": 106, + "num_frames": 19323 + }, + "robotgeneralist/PickPlaceToasterToCounter_pretrain": { + "commit_hash": "f748ad7b5f13a1faec1fb21a8640903a344298ba", + "num_episodes": 105, + "num_frames": 26907 + }, + "robotgeneralist/PreheatOven_pretrain": { + "commit_hash": "fe6d8d7f32aabfb9605daec9d6dc9e5552222f43", + "num_episodes": 107, + "num_frames": 21102 + }, + "robotgeneralist/SlideDishwasherRack_pretrain": { + "commit_hash": "ee2be96cab8d45b0713a4d3f4c77b7d086b593e7", + "num_episodes": 100, + "num_frames": 19052 + }, + "robotgeneralist/SlideOvenRack_pretrain": { + "commit_hash": "ec897192c4630505dbf7ecafc87672d945b5806b", + "num_episodes": 113, + "num_frames": 23958 + }, + "robotgeneralist/SlideToasterOvenRack_pretrain": { + "commit_hash": "90c2db6d3c847d2a25ced10ba441cd79845218d3", + "num_episodes": 105, + "num_frames": 11496 + }, + "robotgeneralist/StartCoffeeMachine_pretrain": { + "commit_hash": "62b11dc4fa1459dffac7c20f596a282181f8ebba", + "num_episodes": 108, + "num_frames": 13722 + }, + "robotgeneralist/TurnOffMicrowave_pretrain": { + "commit_hash": "b40bb9ef7cdb8dd9c5da4487b3d4e538016d28ee", + "num_episodes": 108, + "num_frames": 15233 + }, + "robotgeneralist/TurnOffSinkFaucet_pretrain": { + "commit_hash": "4726894e94f3b4baf80a96cc9041eb95fe399fb1", + "num_episodes": 106, + "num_frames": 12309 + }, + "robotgeneralist/TurnOffStove_pretrain": { + "commit_hash": "c5029366ce8dd5becc73980076d4e6106d76dc76", + "num_episodes": 109, + "num_frames": 32741 + }, + "robotgeneralist/TurnOnBlender_pretrain": { + "commit_hash": "aaa1396e005c3d83f3fa320f3b66a8c240a1e86d", + "num_episodes": 107, + "num_frames": 11698 + }, + "robotgeneralist/TurnOnElectricKettle_pretrain": { + "commit_hash": "01adea2d774b56c2f269380b938dc822bd446552", + "num_episodes": 108, + "num_frames": 12460 + }, + "robotgeneralist/TurnOnMicrowave_pretrain": { + "commit_hash": "0fd65a1926ad490725cee73933917d7a9985cf57", + "num_episodes": 107, + "num_frames": 14010 + }, + "robotgeneralist/TurnOnSinkFaucet_pretrain": { + "commit_hash": "d4efe49ea58fde3cef15480faf663b17af6481f4", + "num_episodes": 107, + "num_frames": 23795 + }, + "robotgeneralist/TurnOnStove_pretrain": { + "commit_hash": "782bec492f509094e3dd61938b4d906db62a1321", + "num_episodes": 104, + "num_frames": 17910 + }, + "robotgeneralist/TurnOnToaster_pretrain": { + "commit_hash": "5b00e0998c7270bc7b1e106b6c0acd78ee15eac3", + "num_episodes": 108, + "num_frames": 10042 + }, + "robotgeneralist/TurnOnToasterOven_pretrain": { + "commit_hash": "f6808d51f8efc6191cf5d1d2d8e795d68657567a", + "num_episodes": 103, + "num_frames": 17051 + }, + "robotgeneralist/TurnSinkSpout_pretrain": { + "commit_hash": "504c7c807a1cf800867437f50b6fb08410121807", + "num_episodes": 109, + "num_frames": 11130 + } + } +} \ No newline at end of file diff --git a/robocasa_test_all_atomic/20000/params/_METADATA b/robocasa_test_all_atomic/20000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..9eb73624158f4514d49041e0806d2d702c62eee1 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [14, 14, 3, 1152]}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152, 2048]}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 256, 1152]}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [257152, 2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 2048, 16384]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 16384, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1024, 4096]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 4096, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 32]}}, "('params', 'action_time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048, 1024]}}, "('params', 'action_time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 1024]}}, "('params', 'state_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'state_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/robocasa_test_all_atomic/20000/params/_sharding b/robocasa_test_all_atomic/20000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..924608abdcb99abb3c864bb9d1bfca6eb60f780e --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/_sharding @@ -0,0 +1 @@ +{"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}"} \ No newline at end of file diff --git a/robocasa_test_all_atomic/20000/params/array_metadatas/process_0 b/robocasa_test_all_atomic/20000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..6aa9fd1decf72607133eb54664d98802bf563b5c --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/array_metadatas/process_0 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]} \ No newline at end of file diff --git a/robocasa_test_all_atomic/20000/params/d/7c9795d8a8b83a6a9fbc19fc88a9e654 b/robocasa_test_all_atomic/20000/params/d/7c9795d8a8b83a6a9fbc19fc88a9e654 new file mode 100644 index 0000000000000000000000000000000000000000..93bc68aa745342929a7e0d32fb368f2de9140493 Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/d/7c9795d8a8b83a6a9fbc19fc88a9e654 differ diff --git a/robocasa_test_all_atomic/20000/params/manifest.ocdbt b/robocasa_test_all_atomic/20000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..8281955cbf8f817155fa67dd8d89120a7130cc8b Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/00af7efa6854d2978cea03d9cba221b9 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/00af7efa6854d2978cea03d9cba221b9 new file mode 100644 index 0000000000000000000000000000000000000000..afa812790b5ad4f9471f4b8b1b7000e20c8a216b --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/00af7efa6854d2978cea03d9cba221b9 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:53ccd38e05fd9f4c0c5af4205af9b098461bb71ea0d21eb381c93acfac9f5996 +size 43860164 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/07fb6438160dd3701f5530cefdd0457b b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/07fb6438160dd3701f5530cefdd0457b new file mode 100644 index 0000000000000000000000000000000000000000..5317a2c5cc33a742712c3e9a76e4f4976afb1c85 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/07fb6438160dd3701f5530cefdd0457b @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:645e07f39c8bb243dec2e0f7ca7b402510013dbf8630fd50607cbafc2e61c74b +size 265904310 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/220c9c33cb2564892b2a61129cd932bd b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/220c9c33cb2564892b2a61129cd932bd new file mode 100644 index 0000000000000000000000000000000000000000..1aae014c126293ed35a442c4f0215bfc4e970a14 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/220c9c33cb2564892b2a61129cd932bd @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:5c3be08d5b4c3e3d2a6b61bd93aa37de0e7a77eed38739e8ada81191c312557e +size 1118979 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/28d2fdb42be2a56c2e06673acaf62454 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/28d2fdb42be2a56c2e06673acaf62454 new file mode 100644 index 0000000000000000000000000000000000000000..ffc57c5cefe3160b4f5ec2c8030ab168d9324180 Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/28d2fdb42be2a56c2e06673acaf62454 differ diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/293f5e08ce6095836eeee01ab8ff7f8d b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/293f5e08ce6095836eeee01ab8ff7f8d new file mode 100644 index 0000000000000000000000000000000000000000..c794b5d9e953853c9ce641e23b3ae69a0ad4ffdc --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/293f5e08ce6095836eeee01ab8ff7f8d @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:ed19d1b9535120c915f05b2eaec8a3b90a9616dc0d4e8df935594cb2210e952f +size 2240263007 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/3c278f1430593277115ac72b6462768c b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/3c278f1430593277115ac72b6462768c new file mode 100644 index 0000000000000000000000000000000000000000..7126da4169d9f8f38b2e22710d41c4d7bd1c2808 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/3c278f1430593277115ac72b6462768c @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:60bb514733bf32ef4a8420bfdff0783d4ed2acbf8fd9b15747610fc1d3cf6ea4 +size 70081751 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/4fd7e8b1e5e6559336e2f2979ea4ad9f b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/4fd7e8b1e5e6559336e2f2979ea4ad9f new file mode 100644 index 0000000000000000000000000000000000000000..398b00cba9634df489f2ae819871442b17b797ea --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/4fd7e8b1e5e6559336e2f2979ea4ad9f @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f0c97456be75eee44467388af5411e7c2d27ff5a7b6c770f9676a35401a77fad +size 16293369 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/544b5cbd6788cb5274852a3f5522ae51 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/544b5cbd6788cb5274852a3f5522ae51 new file mode 100644 index 0000000000000000000000000000000000000000..1937bdd99e3d31a79a72edf58aa2056b8ded7cd1 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/544b5cbd6788cb5274852a3f5522ae51 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1c8a1077881cc13107b5f586b139cd9ddc69ccf7195ebe8bb0346e8499229e0a +size 2240225294 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5a0a05f1cb40745f53331701f9f1dcd2 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5a0a05f1cb40745f53331701f9f1dcd2 new file mode 100644 index 0000000000000000000000000000000000000000..35f9ff3f3d0c291a7114dc379e684ad07363f7a4 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5a0a05f1cb40745f53331701f9f1dcd2 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a16440397c96e2139baa97baa78abaccbea9b1a42be9bd6b61506424ea95bf1e +size 497034634 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5b74442612bfab52132de9e5d473514c b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5b74442612bfab52132de9e5d473514c new file mode 100644 index 0000000000000000000000000000000000000000..7110809726f86b147cf9cffcade682fd92de7d76 Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/5b74442612bfab52132de9e5d473514c differ diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/62f883e4860df4efd5d871cb9b4f8ff9 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/62f883e4860df4efd5d871cb9b4f8ff9 new file mode 100644 index 0000000000000000000000000000000000000000..d3fc7458b64fe10fd6f560daa342fffbca337f9b --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/62f883e4860df4efd5d871cb9b4f8ff9 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c899c275e56c29238df0da56bd4112c465f3ba5abb02233092d1b40b1a51b4f +size 496759947 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6b17d7839b157db09b830be34ffaa9be b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6b17d7839b157db09b830be34ffaa9be new file mode 100644 index 0000000000000000000000000000000000000000..a3c08a1b186dbb588f31a9458025546e8327399a Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6b17d7839b157db09b830be34ffaa9be differ diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6c882add699c3c540d47c1b320e168c3 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6c882add699c3c540d47c1b320e168c3 new file mode 100644 index 0000000000000000000000000000000000000000..5f35626d06a43958c41cf2ad9e9da6e235a5472d --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/6c882add699c3c540d47c1b320e168c3 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6ada0e9ba79ac2f1ffcfd0ed938ee19b2af99c5453fbc08682a7aafe75171062 +size 132844984 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/76b44384dc110b74eabaf89b6c912045 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/76b44384dc110b74eabaf89b6c912045 new file mode 100644 index 0000000000000000000000000000000000000000..4fb1c437762e9c2f0d40fdfad2a9cb9a29380ace --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/76b44384dc110b74eabaf89b6c912045 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:666bb285a82eac7e0608a4a086d01c8898a2d6e1b827bb61f98e84373c57a7ab +size 1120110654 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/8eaed967dc675b0e6fcc0ea4664a7ceb b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/8eaed967dc675b0e6fcc0ea4664a7ceb new file mode 100644 index 0000000000000000000000000000000000000000..ce54053cee5b1ae840946f8d799e1d3b069d561f --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/8eaed967dc675b0e6fcc0ea4664a7ceb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0273be9ac9e2c371e57c40457dc74732cfe8b64b6f5742d7c106266e14c24c3e +size 280188353 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/9dd81c05d6a91aef3a037fccd41de5f1 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/9dd81c05d6a91aef3a037fccd41de5f1 new file mode 100644 index 0000000000000000000000000000000000000000..ea03c9151315f64a21c291e5ad8eec6dacc961f8 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/9dd81c05d6a91aef3a037fccd41de5f1 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a84e009ba043e2f775abe6c0f8482efd229b3a81e73278a8364c72d39a69c010 +size 1957442346 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9869d9caca47dfda0dc5b4040ed37d3 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9869d9caca47dfda0dc5b4040ed37d3 new file mode 100644 index 0000000000000000000000000000000000000000..aede9e5d2c4a7b51d434de3d35ad87ead09e1a17 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9869d9caca47dfda0dc5b4040ed37d3 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f72c1e5eac37a111453904becc44550912c22be13264468d1cc0f5da6261370 +size 280036813 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9fa6d29a3627b2776b80e4c5d34e6ac b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9fa6d29a3627b2776b80e4c5d34e6ac new file mode 100644 index 0000000000000000000000000000000000000000..46cd46020c3fc37eed8126482d2295a61a10ce74 Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/a9fa6d29a3627b2776b80e4c5d34e6ac differ diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/b4103fba59981246d8aa41973f5db056 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/b4103fba59981246d8aa41973f5db056 new file mode 100644 index 0000000000000000000000000000000000000000..c0ac7bdc63c058370c9f45613f2329786bf42e2e --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/b4103fba59981246d8aa41973f5db056 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:531e197ef3a9bb048436bc88df2aaff4dd842851a5d8b1090dbbb016210df686 +size 559726121 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/d37d4bf12d12967462dd75a12ba07fda b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/d37d4bf12d12967462dd75a12ba07fda new file mode 100644 index 0000000000000000000000000000000000000000..225d126e381f6a50f869420280cf821efd7fb26b --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/d37d4bf12d12967462dd75a12ba07fda @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:67727c7e04afd0ff980ed583a668ddd555d69c4d3d71a3613a512e704804139a +size 1120131322 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/f8b773fe4952a7147d6c60500fd33ea0 b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/f8b773fe4952a7147d6c60500fd33ea0 new file mode 100644 index 0000000000000000000000000000000000000000..240548427170870fc2301b41dbfed65fb8cbf287 --- /dev/null +++ b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/d/f8b773fe4952a7147d6c60500fd33ea0 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:24b5b0389dd6b475b8eed9a2c85e619f5ef27591eb8195364a50c8eae786cfdf +size 692093133 diff --git a/robocasa_test_all_atomic/20000/params/ocdbt.process_0/manifest.ocdbt b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..e76c40e39c376371471daca9ad22da8228232c75 Binary files /dev/null and b/robocasa_test_all_atomic/20000/params/ocdbt.process_0/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/25000/_CHECKPOINT_METADATA b/robocasa_test_all_atomic/25000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..d4890083815acb83da755746ad1610dfe4f881c7 --- /dev/null +++ b/robocasa_test_all_atomic/25000/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": {"assets": "openpi.training.checkpoints.CallbackHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1775520536081222343, "commit_timestamp_nsecs": 1775520579025909545, "custom_metadata": {}} \ No newline at end of file diff --git a/robocasa_test_all_atomic/25000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json b/robocasa_test_all_atomic/25000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..a61f23a33c416ca9fb46fde8f50af856841be99a --- /dev/null +++ b/robocasa_test_all_atomic/25000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json @@ -0,0 +1,464 @@ +{ + "norm_stats": { + "state": { + "mean": [ + 2.637775182723999, + -2.0912017822265625, + 0.7007339596748352, + 0.0, + 0.0, + 0.24000521004199982, + 0.6076071262359619, + 0.258401483297348, + -0.018448200076818466, + 0.44676560163497925, + -0.26384344696998596, + 0.03587118536233902, + -0.07241316884756088, + 0.13728724420070648, + 0.03210829943418503, + -0.03232070058584213 + ], + "std": [ + 1.6751232147216797, + 1.3338454961776733, + 0.0, + 0.0, + 0.0, + 0.6626312732696533, + 0.3660806715488434, + 0.16573558747768402, + 0.2436121702194214, + 0.2089885175228119, + 0.8243448734283447, + 0.29920220375061035, + 0.3282162845134735, + 0.1679685264825821, + 0.012291335500776768, + 0.012060738168656826 + ], + "q01": [ + -1.218368652534485, + -5.65514452624917, + 0.6999506360054016, + 0.0, + 0.0, + -1.0, + 0.0, + -0.13958552385568623, + -0.5809329965710639, + -0.1109548017382622, + -0.9983834087371826, + -0.6892688079476357, + -0.745897297000885, + 0.0007815830585608637, + -0.000049362923577427054, + -0.04081762075200677 + ], + "q99": [ + 6.3111974027633675, + -0.46855325088500965, + 0.7051179245471955, + 0.0, + 0.0, + 0.9996, + 0.9998, + 0.7251633455753326, + 0.5839828933954241, + 0.8639214414954184, + 0.9963686748266221, + 0.7791513680815696, + 0.726318956565857, + 0.6806885522310082, + 0.040635910079255703, + 0.00001817415971308578 + ] + }, + "actions": { + "mean": [ + 0.026344629004597664, + 0.0013832177501171827, + -0.0021026874892413616, + 0.0, + -0.8568487763404846, + -0.009222988970577717, + -0.0020607062615454197, + -0.051031820476055145, + 0.008847072720527649, + -0.018305813893675804, + -0.0024289467837661505, + -0.330047070980072 + ], + "std": [ + 0.16712690889835358, + 0.1392730474472046, + 0.12041067332029343, + 0.0, + 0.5155678391456604, + 0.4523046910762787, + 0.38781222701072693, + 0.4078434407711029, + 0.11646749824285507, + 0.13211272656917572, + 0.1214357316493988, + 0.9439644813537598 + ], + "q01": [ + -0.0716, + -0.5571999999999999, + -0.5371999999999999, + 0.0, + -1.0, + -1.0, + -1.0, + -1.0, + -0.346, + -0.45719999999999994, + -0.4032, + -1.0 + ], + "q99": [ + 0.9996, + 0.6356000000000002, + 0.4828000000000001, + 0.0, + 0.9996, + 0.9996, + 0.9996, + 0.9996, + 0.4456, + 0.3740000000000001, + 0.3884000000000001, + 0.9996 + ] + } + }, + "sample_ratio": 0.05, + "dataset_infos": { + "robotgeneralist/AdjustToasterOvenTemperature_pretrain": { + "commit_hash": "24364b1b9664fe7a58ecfce7e61cac66bcde1209", + "num_episodes": 107, + "num_frames": 21328 + }, + "robotgeneralist/AdjustWaterTemperature_pretrain": { + "commit_hash": "f1bcd907c0510f687521a44b98edaba314548a95", + "num_episodes": 106, + "num_frames": 20953 + }, + "robotgeneralist/CheesyBread_pretrain": { + "commit_hash": "d59106dabf8827be89ea3513399b6a48c8cbf814", + "num_episodes": 101, + "num_frames": 31141 + }, + "robotgeneralist/CloseBlenderLid_pretrain": { + "commit_hash": "68fb8f0d7495e9da1382c803467c488b97b55bf0", + "num_episodes": 106, + "num_frames": 36933 + }, + "robotgeneralist/CloseCabinet_pretrain": { + "commit_hash": "b58bca4c52b1673fb6a20fd0a15e9031f5b3eb72", + "num_episodes": 105, + "num_frames": 27754 + }, + "robotgeneralist/CloseDishwasher_pretrain": { + "commit_hash": "659c06a9486fb1a0eaf37c25b35af0b3e8ff2d4e", + "num_episodes": 111, + "num_frames": 15781 + }, + "robotgeneralist/CloseDrawer_pretrain": { + "commit_hash": "ab571fcde78de907f8154bcad20101423a8ee503", + "num_episodes": 110, + "num_frames": 15670 + }, + "robotgeneralist/CloseElectricKettleLid_pretrain": { + "commit_hash": "b4b4c5967fe877790ac86c5bf9e8a0216e229105", + "num_episodes": 102, + "num_frames": 7530 + }, + "robotgeneralist/CloseFridge_pretrain": { + "commit_hash": "fb04ab4ebc39a0a23dc27140f4f7701b91b7313b", + "num_episodes": 106, + "num_frames": 26888 + }, + "robotgeneralist/CloseFridgeDrawer_pretrain": { + "commit_hash": "ed9f1b467346f600ae6f755ebc7ad0e1dd841dda", + "num_episodes": 111, + "num_frames": 14946 + }, + "robotgeneralist/CloseMicrowave_pretrain": { + "commit_hash": "61e4ea6d29e504329df7147ca563d34324ae7aa5", + "num_episodes": 105, + "num_frames": 20075 + }, + "robotgeneralist/CloseOven_pretrain": { + "commit_hash": "61b80872b74919655d4fed2fc4e8667ce248fa48", + "num_episodes": 107, + "num_frames": 18230 + }, + "robotgeneralist/CloseStandMixerHead_pretrain": { + "commit_hash": "855c0a58a8396f8855eb8e213e6bed53dcde551a", + "num_episodes": 110, + "num_frames": 11593 + }, + "robotgeneralist/CloseToasterOvenDoor_pretrain": { + "commit_hash": "3c16da6f363ad8fd2bcd811efd4a87c6ed61a82c", + "num_episodes": 110, + "num_frames": 19815 + }, + "robotgeneralist/CoffeeServeMug_pretrain": { + "commit_hash": "23157f307108f6e352b98b536e941d1086c3d8af", + "num_episodes": 108, + "num_frames": 16921 + }, + "robotgeneralist/CoffeeSetupMug_pretrain": { + "commit_hash": "76e51e454262ddeb046b015486c1fe4602a53b49", + "num_episodes": 105, + "num_frames": 23636 + }, + "robotgeneralist/LowerHeat_pretrain": { + "commit_hash": "891f233633a61ce11f4fefe95a3b5081657c0227", + "num_episodes": 110, + "num_frames": 31174 + }, + "robotgeneralist/MakeIcedCoffee_pretrain": { + "commit_hash": "aee5552cade8433a5c9282851250ee0a7c6c4717", + "num_episodes": 102, + "num_frames": 29048 + }, + "robotgeneralist/NavigateKitchen_pretrain": { + "commit_hash": "9df39eed58424fbe7945717c54549e5308c9fcc6", + "num_episodes": 503, + "num_frames": 79550 + }, + "robotgeneralist/OpenBlenderLid_pretrain": { + "commit_hash": "302d625654bdb678e877dff9b6a0d9c9dcc3fc15", + "num_episodes": 106, + "num_frames": 20124 + }, + "robotgeneralist/OpenCabinet_pretrain": { + "commit_hash": "ee50aad81bc9a115a5f00885d45046d6539646cf", + "num_episodes": 107, + "num_frames": 37492 + }, + "robotgeneralist/OpenDishwasher_pretrain": { + "commit_hash": "7dd0903b5ddf482ddedbd0f013fe5223118d47b2", + "num_episodes": 111, + "num_frames": 18086 + }, + "robotgeneralist/OpenDrawer_pretrain": { + "commit_hash": "f2abb58f7fdf0157aeeaa645791a460de44be1d8", + "num_episodes": 102, + "num_frames": 20488 + }, + "robotgeneralist/OpenElectricKettleLid_pretrain": { + "commit_hash": "5416f530c440f61466d02f6ce0bf549973c0db95", + "num_episodes": 110, + "num_frames": 10928 + }, + "robotgeneralist/OpenFridge_pretrain": { + "commit_hash": "5869139ca48dbe66f7d8dabe51d4ecaf4fb71c9d", + "num_episodes": 105, + "num_frames": 33138 + }, + "robotgeneralist/OpenFridgeDrawer_pretrain": { + "commit_hash": "ceda3eb991f74682c8bc679c348402ea4587f5c4", + "num_episodes": 107, + "num_frames": 18517 + }, + "robotgeneralist/OpenMicrowave_pretrain": { + "commit_hash": "6222929e60bbee3d135cfdc27d6663e3fc535ead", + "num_episodes": 105, + "num_frames": 26017 + }, + "robotgeneralist/OpenOven_pretrain": { + "commit_hash": "b8e0e83300e3754fdd21de46680cb2852d8905c9", + "num_episodes": 108, + "num_frames": 15555 + }, + "robotgeneralist/OpenStandMixerHead_pretrain": { + "commit_hash": "43c2fe25f56fc1904f12db1b66855c1dd422463a", + "num_episodes": 109, + "num_frames": 13411 + }, + "robotgeneralist/OpenToasterOvenDoor_pretrain": { + "commit_hash": "ad2164f3590a05d06ba5cd2aea8216ba73619cb6", + "num_episodes": 105, + "num_frames": 15469 + }, + "robotgeneralist/PackDessert_pretrain": { + "commit_hash": "63d48fe23f8fb5d5a9f603b62c38f22ad0d5a418", + "num_episodes": 108, + "num_frames": 27994 + }, + "robotgeneralist/PickPlaceCabinetToCounter_pretrain": { + "commit_hash": "a338e9f408bad681248461fe42deffb77e0ab67a", + "num_episodes": 106, + "num_frames": 20201 + }, + "robotgeneralist/PickPlaceCounterToBlender_pretrain": { + "commit_hash": "6a2aba5d6a1fe63f65f7ad40d5098e8d68b08e72", + "num_episodes": 106, + "num_frames": 38892 + }, + "robotgeneralist/PickPlaceCounterToCabinet_pretrain": { + "commit_hash": "628a13d9c15c967e6133d394913b0dfe55e8fa8e", + "num_episodes": 108, + "num_frames": 24225 + }, + "robotgeneralist/PickPlaceCounterToDrawer_pretrain": { + "commit_hash": "35ac4c271761916cde68099be0fad4f59ad68dba", + "num_episodes": 107, + "num_frames": 28225 + }, + "robotgeneralist/PickPlaceCounterToMicrowave_pretrain": { + "commit_hash": "c9720edf6282bcf99f2a76505804f81d4ac97e92", + "num_episodes": 110, + "num_frames": 42012 + }, + "robotgeneralist/PickPlaceCounterToOven_pretrain": { + "commit_hash": "f1d9b1c899de4abec2f3a11b5abd0db519f555d7", + "num_episodes": 111, + "num_frames": 32014 + }, + "robotgeneralist/PickPlaceCounterToSink_pretrain": { + "commit_hash": "8642bb8c7db762fe4b82fd0b0e26847223f83de0", + "num_episodes": 108, + "num_frames": 22410 + }, + "robotgeneralist/PickPlaceCounterToStandMixer_pretrain": { + "commit_hash": "76ca30ee317ae85519afb52a0bec1ee8d116a732", + "num_episodes": 107, + "num_frames": 25467 + }, + "robotgeneralist/PickPlaceCounterToStove_pretrain": { + "commit_hash": "bf5a50869a7ecc1050f878001b39c6cb3bb0bfe5", + "num_episodes": 108, + "num_frames": 24039 + }, + "robotgeneralist/PickPlaceCounterToToasterOven_pretrain": { + "commit_hash": "515f871b5d014f070bb221d7cd9568508db890ef", + "num_episodes": 108, + "num_frames": 24313 + }, + "robotgeneralist/PickPlaceDrawerToCounter_pretrain": { + "commit_hash": "cf62a0e481b087d531fc8cc6b49f1d697ac82d9e", + "num_episodes": 103, + "num_frames": 31819 + }, + "robotgeneralist/PickPlaceFridgeDrawerToShelf_pretrain": { + "commit_hash": "b944b9dd65fb12c6253e3254f0acd7bfb7af730f", + "num_episodes": 106, + "num_frames": 26396 + }, + "robotgeneralist/PickPlaceFridgeShelfToDrawer_pretrain": { + "commit_hash": "5381c4c9ff0e1a09cf963d537836e1060ef75565", + "num_episodes": 111, + "num_frames": 27047 + }, + "robotgeneralist/PickPlaceMicrowaveToCounter_pretrain": { + "commit_hash": "de6cc72aa95974b46fd965d66ab60629c68503a3", + "num_episodes": 112, + "num_frames": 38729 + }, + "robotgeneralist/PickPlaceSinkToCounter_pretrain": { + "commit_hash": "854d27642b1810e7576cefe465c181bbee5dfcb3", + "num_episodes": 108, + "num_frames": 26397 + }, + "robotgeneralist/PickPlaceStoveToCounter_pretrain": { + "commit_hash": "fca1c62a4f719a24a0e8b587c0422e37c7ef5756", + "num_episodes": 109, + "num_frames": 23003 + }, + "robotgeneralist/PickPlaceToasterOvenToCounter_pretrain": { + "commit_hash": "18f6af5e40443639cb1af2e32cb991d8813e016e", + "num_episodes": 106, + "num_frames": 19323 + }, + "robotgeneralist/PickPlaceToasterToCounter_pretrain": { + "commit_hash": "f748ad7b5f13a1faec1fb21a8640903a344298ba", + "num_episodes": 105, + "num_frames": 26907 + }, + "robotgeneralist/PreheatOven_pretrain": { + "commit_hash": "fe6d8d7f32aabfb9605daec9d6dc9e5552222f43", + "num_episodes": 107, + "num_frames": 21102 + }, + "robotgeneralist/SlideDishwasherRack_pretrain": { + "commit_hash": "ee2be96cab8d45b0713a4d3f4c77b7d086b593e7", + "num_episodes": 100, + "num_frames": 19052 + }, + "robotgeneralist/SlideOvenRack_pretrain": { + "commit_hash": "ec897192c4630505dbf7ecafc87672d945b5806b", + "num_episodes": 113, + "num_frames": 23958 + }, + "robotgeneralist/SlideToasterOvenRack_pretrain": { + "commit_hash": "90c2db6d3c847d2a25ced10ba441cd79845218d3", + "num_episodes": 105, + "num_frames": 11496 + }, + "robotgeneralist/StartCoffeeMachine_pretrain": { + "commit_hash": "62b11dc4fa1459dffac7c20f596a282181f8ebba", + "num_episodes": 108, + "num_frames": 13722 + }, + "robotgeneralist/TurnOffMicrowave_pretrain": { + "commit_hash": "b40bb9ef7cdb8dd9c5da4487b3d4e538016d28ee", + "num_episodes": 108, + "num_frames": 15233 + }, + "robotgeneralist/TurnOffSinkFaucet_pretrain": { + "commit_hash": "4726894e94f3b4baf80a96cc9041eb95fe399fb1", + "num_episodes": 106, + "num_frames": 12309 + }, + "robotgeneralist/TurnOffStove_pretrain": { + "commit_hash": "c5029366ce8dd5becc73980076d4e6106d76dc76", + "num_episodes": 109, + "num_frames": 32741 + }, + "robotgeneralist/TurnOnBlender_pretrain": { + "commit_hash": "aaa1396e005c3d83f3fa320f3b66a8c240a1e86d", + "num_episodes": 107, + "num_frames": 11698 + }, + "robotgeneralist/TurnOnElectricKettle_pretrain": { + "commit_hash": "01adea2d774b56c2f269380b938dc822bd446552", + "num_episodes": 108, + "num_frames": 12460 + }, + "robotgeneralist/TurnOnMicrowave_pretrain": { + "commit_hash": "0fd65a1926ad490725cee73933917d7a9985cf57", + "num_episodes": 107, + "num_frames": 14010 + }, + "robotgeneralist/TurnOnSinkFaucet_pretrain": { + "commit_hash": "d4efe49ea58fde3cef15480faf663b17af6481f4", + "num_episodes": 107, + "num_frames": 23795 + }, + "robotgeneralist/TurnOnStove_pretrain": { + "commit_hash": "782bec492f509094e3dd61938b4d906db62a1321", + "num_episodes": 104, + "num_frames": 17910 + }, + "robotgeneralist/TurnOnToaster_pretrain": { + "commit_hash": "5b00e0998c7270bc7b1e106b6c0acd78ee15eac3", + "num_episodes": 108, + "num_frames": 10042 + }, + "robotgeneralist/TurnOnToasterOven_pretrain": { + "commit_hash": "f6808d51f8efc6191cf5d1d2d8e795d68657567a", + "num_episodes": 103, + "num_frames": 17051 + }, + "robotgeneralist/TurnSinkSpout_pretrain": { + "commit_hash": "504c7c807a1cf800867437f50b6fb08410121807", + "num_episodes": 109, + "num_frames": 11130 + } + } +} \ No newline at end of file diff --git a/robocasa_test_all_atomic/25000/params/_METADATA b/robocasa_test_all_atomic/25000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..9eb73624158f4514d49041e0806d2d702c62eee1 --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoder_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoder_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_0', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_0", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'LayerNorm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "LayerNorm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_0', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 4304]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MlpBlock_0', 'Dense_1', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MlpBlock_0", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 4304, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'key', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72, 1152]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'query', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "query", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 16, 72]}}, "('params', 'PaliGemma', 'img', 'Transformer', 'encoderblock', 'MultiHeadDotProductAttention_0', 'value', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "Transformer", "key_type": 2}, {"key": "encoderblock", "key_type": 2}, {"key": "MultiHeadDotProductAttention_0", "key_type": 2}, {"key": "value", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [27, 1152, 16, 72]}}, "('params', 'PaliGemma', 'img', 'embedding', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152]}}, "('params', 'PaliGemma', 'img', 'embedding', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [14, 14, 3, 1152]}}, "('params', 'PaliGemma', 'img', 'head', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'img', 'head', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "head", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1152, 2048]}}, "('params', 'PaliGemma', 'img', 'pos_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "img", "key_type": 2}, {"key": "pos_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 256, 1152]}}, "('params', 'PaliGemma', 'llm', 'embedder', 'input_embedding', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "input_embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [257152, 2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048]}}, "('params', 'PaliGemma', 'llm', 'final_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "final_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'attn_vec_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "attn_vec_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 256, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'kv_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "kv_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 2048, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'attn', 'q_einsum_1', 'w', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "attn", "key_type": 2}, {"key": "q_einsum_1", "key_type": 2}, {"key": "w", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 8, 1024, 256]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 2048, 16384]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 16384, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'gating_einsum', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "gating_einsum", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2, 1024, 4096]}}, "('params', 'PaliGemma', 'llm', 'layers', 'mlp_1', 'linear', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "mlp_1", "key_type": 2}, {"key": "linear", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 4096, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_attention_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_attention_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 2048]}}, "('params', 'PaliGemma', 'llm', 'layers', 'pre_ffw_norm_1', 'scale', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "PaliGemma", "key_type": 2}, {"key": "llm", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "pre_ffw_norm_1", "key_type": 2}, {"key": "scale", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [18, 1024]}}, "('params', 'action_in_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}, "('params', 'action_out_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', 'action_out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 32]}}, "('params', 'action_time_mlp_in', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_in', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_in", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2048, 1024]}}, "('params', 'action_time_mlp_out', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'action_time_mlp_out', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "action_time_mlp_out", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024, 1024]}}, "('params', 'state_proj', 'bias', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1024]}}, "('params', 'state_proj', 'kernel', 'value')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "state_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32, 1024]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/robocasa_test_all_atomic/25000/params/_sharding b/robocasa_test_all_atomic/25000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..924608abdcb99abb3c864bb9d1bfca6eb60f780e --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/_sharding @@ -0,0 +1 @@ +{"cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2Rlcl9ub3JtLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC52YWx1ZS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5rZXkua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk11bHRpSGVhZERvdFByb2R1Y3RBdHRlbnRpb25fMC5xdWVyeS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLk1scEJsb2NrXzAuRGVuc2VfMS5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8wLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuVHJhbnNmb3JtZXIuZW5jb2RlcmJsb2NrLkxheWVyTm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuZW1iZWRkaW5nLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcuaGVhZC5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5pbWcucG9zX2VtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZW1iZWRkZXIuaW5wdXRfZW1iZWRkaW5nLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0uZmluYWxfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5nYXRpbmdfZWluc3VtLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scC5saW5lYXIudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmdhdGluZ19laW5zdW0udmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLm1scF8xLmxpbmVhci52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4uYXR0bl92ZWNfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtLncudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ua3ZfZWluc3VtXzEudy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW0udy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLmF0dG4ucV9laW5zdW1fMS53LnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9hdHRlbnRpb25fbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybS5zY2FsZS52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLlBhbGlHZW1tYS5sbG0ubGF5ZXJzLnByZV9mZndfbm9ybV8xLnNjYWxlLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9pbi5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQuYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl90aW1lX21scF9vdXQua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmJpYXMudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9pbl9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLmFjdGlvbl9vdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}","cGFyYW1zLnN0YXRlX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1, 1], \"axis_names\": [\"batch\", \"fsdp\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [[{\"id\": 0}]]}}"} \ No newline at end of file diff --git a/robocasa_test_all_atomic/25000/params/array_metadatas/process_0 b/robocasa_test_all_atomic/25000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..6aa9fd1decf72607133eb54664d98802bf563b5c --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/array_metadatas/process_0 @@ -0,0 +1 @@ +{"array_metadatas": [{"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoder_norm.scale.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_0.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.LayerNorm_1.scale.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.bias.value", "write_shape": [27, 4304], "chunk_shape": [27, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_0.kernel.value", "write_shape": [27, 1152, 4304], "chunk_shape": [27, 1152, 4304], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MlpBlock_0.Dense_1.kernel.value", "write_shape": [27, 4304, 1152], "chunk_shape": [27, 4304, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.key.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.bias.value", "write_shape": [27, 1152], "chunk_shape": [27, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.out.kernel.value", "write_shape": [27, 16, 72, 1152], "chunk_shape": [27, 16, 72, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.query.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.bias.value", "write_shape": [27, 16, 72], "chunk_shape": [27, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.Transformer.encoderblock.MultiHeadDotProductAttention_0.value.kernel.value", "write_shape": [27, 1152, 16, 72], "chunk_shape": [27, 1152, 16, 72], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.bias.value", "write_shape": [1152], "chunk_shape": [1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.embedding.kernel.value", "write_shape": [14, 14, 3, 1152], "chunk_shape": [14, 14, 3, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.bias.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.head.kernel.value", "write_shape": [1152, 2048], "chunk_shape": [1152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.img.pos_embedding.value", "write_shape": [1, 256, 1152], "chunk_shape": [1, 256, 1152], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.embedder.input_embedding.value", "write_shape": [257152, 2048], "chunk_shape": [257152, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm.scale.value", "write_shape": [2048], "chunk_shape": [2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.final_norm_1.scale.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum.w.value", "write_shape": [18, 8, 256, 2048], "chunk_shape": [18, 8, 256, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.attn_vec_einsum_1.w.value", "write_shape": [18, 8, 256, 1024], "chunk_shape": [18, 8, 256, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum.w.value", "write_shape": [18, 2, 1, 2048, 256], "chunk_shape": [18, 2, 1, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.kv_einsum_1.w.value", "write_shape": [18, 2, 1, 1024, 256], "chunk_shape": [18, 2, 1, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum.w.value", "write_shape": [18, 8, 2048, 256], "chunk_shape": [18, 8, 2048, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.attn.q_einsum_1.w.value", "write_shape": [18, 8, 1024, 256], "chunk_shape": [18, 8, 1024, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.gating_einsum.value", "write_shape": [18, 2, 2048, 16384], "chunk_shape": [18, 2, 2048, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp.linear.value", "write_shape": [18, 16384, 2048], "chunk_shape": [18, 8192, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.gating_einsum.value", "write_shape": [18, 2, 1024, 4096], "chunk_shape": [18, 2, 1024, 4096], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.mlp_1.linear.value", "write_shape": [18, 4096, 1024], "chunk_shape": [18, 4096, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_attention_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm.scale.value", "write_shape": [18, 2048], "chunk_shape": [18, 2048], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.PaliGemma.llm.layers.pre_ffw_norm_1.scale.value", "write_shape": [18, 1024], "chunk_shape": [18, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_in_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.bias.value", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_out_proj.kernel.value", "write_shape": [1024, 32], "chunk_shape": [1024, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_in.kernel.value", "write_shape": [2048, 1024], "chunk_shape": [2048, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.action_time_mlp_out.kernel.value", "write_shape": [1024, 1024], "chunk_shape": [1024, 1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.bias.value", "write_shape": [1024], "chunk_shape": [1024], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.state_proj.kernel.value", "write_shape": [32, 1024], "chunk_shape": [32, 1024], "ext_metadata": null}}]} \ No newline at end of file diff --git a/robocasa_test_all_atomic/25000/params/d/6253c75e7be5d6ae0fe550c4784c3899 b/robocasa_test_all_atomic/25000/params/d/6253c75e7be5d6ae0fe550c4784c3899 new file mode 100644 index 0000000000000000000000000000000000000000..d7286a13619cab4f11a1054548b22994eb22adc4 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/d/6253c75e7be5d6ae0fe550c4784c3899 differ diff --git a/robocasa_test_all_atomic/25000/params/manifest.ocdbt b/robocasa_test_all_atomic/25000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..263d0bcbe75420d2f249c320dc495727a59f863a Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/04b02653e7076d54bfaf50dbefff2e66 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/04b02653e7076d54bfaf50dbefff2e66 new file mode 100644 index 0000000000000000000000000000000000000000..05925ae6be27469ae602fee21e6a33da2665746b Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/04b02653e7076d54bfaf50dbefff2e66 differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/09d47db0bcf3048d201066b586979a50 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/09d47db0bcf3048d201066b586979a50 new file mode 100644 index 0000000000000000000000000000000000000000..46cd46020c3fc37eed8126482d2295a61a10ce74 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/09d47db0bcf3048d201066b586979a50 differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/103260ea4f910af5238783686c1e8b0e b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/103260ea4f910af5238783686c1e8b0e new file mode 100644 index 0000000000000000000000000000000000000000..079cf3f8e63b0f603a36be879f71a11c96da3c7b --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/103260ea4f910af5238783686c1e8b0e @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8742cd1eb775d9e715cc625041817862501addc4409aacf7868995c6b78f105b +size 16187718 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/12300672920ad9e3cba201fdb87ec852 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/12300672920ad9e3cba201fdb87ec852 new file mode 100644 index 0000000000000000000000000000000000000000..ea062d811b428f9ea4bc724b26da47d406623d31 --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/12300672920ad9e3cba201fdb87ec852 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1ec9f1626c93836879bf7558d05f7e03b69e912fab92b2096ea0b9b527b57bc5 +size 132937328 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/1afbaa46434c0fd6a54f7832317a5f7c b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/1afbaa46434c0fd6a54f7832317a5f7c new file mode 100644 index 0000000000000000000000000000000000000000..dc7c8a75420df036bd7827b25aa5eceb939163fe --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/1afbaa46434c0fd6a54f7832317a5f7c @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1be92060c57f0faf3f787da6f8c71a74482749697b0b9e32994102a3a3058e17 +size 139970439 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/45cf9cd4afbc860169fd850ac59b9c45 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/45cf9cd4afbc860169fd850ac59b9c45 new file mode 100644 index 0000000000000000000000000000000000000000..67bf80df50a3de30677bb3835db52bac1361e49d --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/45cf9cd4afbc860169fd850ac59b9c45 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:29de94e43e4253f6106a65f6e72007cf4ff4f3b2299d216e7c634d0c0fe4781e +size 70081788 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/5a0478d092e8c38d6dc2286a32764170 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/5a0478d092e8c38d6dc2286a32764170 new file mode 100644 index 0000000000000000000000000000000000000000..500921cf5810610d276c118996caa283442a9ab1 --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/5a0478d092e8c38d6dc2286a32764170 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:da8e717bcb345086e7372cf2ad18692b10598b4013c79c6bcaf5ba9bce8c8215 +size 1957442766 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/7e1ff05a8276d7018f94ea807cf89dd6 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/7e1ff05a8276d7018f94ea807cf89dd6 new file mode 100644 index 0000000000000000000000000000000000000000..725a985f368c2c2c998aa9a439be370acf520b49 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/7e1ff05a8276d7018f94ea807cf89dd6 differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8096bcfab7faa6bfccb00ee709811345 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8096bcfab7faa6bfccb00ee709811345 new file mode 100644 index 0000000000000000000000000000000000000000..0aec8290e552137b670a7ef37b03a357bf27b910 --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8096bcfab7faa6bfccb00ee709811345 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e246aeab592471e1cedf190b2f85d38972e335b45dce60e9b8bc5d0f22803aef +size 36327490 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/873ddb891e2a9959613902029998f221 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/873ddb891e2a9959613902029998f221 new file mode 100644 index 0000000000000000000000000000000000000000..e41b7b404dc0c10db3ad694dd2d89fcbf74f0529 --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/873ddb891e2a9959613902029998f221 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1f2494f965cdf62fc5edb7a2255818c3f947864d9d41ea09ad9a3563227ab76b +size 8754113 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8907812120ec5be1d8db2b6f66657b61 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8907812120ec5be1d8db2b6f66657b61 new file mode 100644 index 0000000000000000000000000000000000000000..202e8b3cdac45e030f30cc9b3b82150e8f371269 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/8907812120ec5be1d8db2b6f66657b61 differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/941d813a2908dca895732ec191069501 b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/941d813a2908dca895732ec191069501 new file mode 100644 index 0000000000000000000000000000000000000000..ffc57c5cefe3160b4f5ec2c8030ab168d9324180 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/941d813a2908dca895732ec191069501 differ diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/a8a1680dfcbb3985189eeabd625e2cdf b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/a8a1680dfcbb3985189eeabd625e2cdf new file mode 100644 index 0000000000000000000000000000000000000000..d1350c5384aea457b8035eadf9fceeaab6bbe14b --- /dev/null +++ b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/d/a8a1680dfcbb3985189eeabd625e2cdf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:554670aa7ee3a009042f0e82f08597367bd1f8491bc0355acffcdfd9685c2d28 +size 132967102 diff --git a/robocasa_test_all_atomic/25000/params/ocdbt.process_0/manifest.ocdbt b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..7ef7ae689b4eb0d5a7db7d74283b6ab7f33c2fc9 Binary files /dev/null and b/robocasa_test_all_atomic/25000/params/ocdbt.process_0/manifest.ocdbt differ