diff --git a/.gitattributes b/.gitattributes index bb895bc0f5add17e2d4f3c99f71c88422f418593..d5adc31cc36a7b0ccd52b8369385d9ab0ade1e62 100644 --- a/.gitattributes +++ b/.gitattributes @@ -97,3 +97,30 @@ robocasa_test_all_atomic/10000/params/ocdbt.process_0/d/2e8652c2871cac327c798416 robocasa_test_all_atomic/10000/params/ocdbt.process_0/d/e0399ea5fb953f49be4693dc020e57ec filter=lfs diff=lfs merge=lfs -text robocasa_test_all_atomic/10000/params/ocdbt.process_0/d/e23cfd91fcf1cdb857e6d4f2f5523be9 filter=lfs diff=lfs merge=lfs -text robocasa_test_all_atomic/10000/params/ocdbt.process_0/d/fa34ae6352738b0498066afa7280cc43 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/ecc5647631ee9d5120212c025910ae4e filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a4e70ae5855d738f5986a53b5f598954 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/5858dcef70dab03c80c42498da149742 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/1d593c9d62349b6cbaf408ffca3427b9 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/df70b8ed24ad4f39fdf4ea69efd2c52d filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/786c87c704c84afd277204b0536280cb filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/2eb18c4bb4e92b99c18988caad47e6cc filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ab643f64ff47d16928047782f50742a8 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/127ef3507813b0971202deb3373b7c09 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ffcb915dab3ba236d24c08cb9c088d6a filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/4c4df646095fe62c2a02d86ed37a01e8 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/03985f56db209ed2f954d087cd975570 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/74a77a5d3ba3b3df71170d0899fb4f24 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/cdce9ad7d2d6345dd8c163be67f4d297 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/96ce1e9519714acfdae49abe9c4805bf filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/d5dc2666d2648f11f5682cf8870910cb filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/9755e2ad714b5b5d0731af7033e82110 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/de453a99166108df1b4aef4d6fd5d785 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/6a926ccfca27ec26a7f457c23e6eed97 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/80414f95b12bb7cc6d6ad42c8d496cba filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/88f55de2a4872e8248b0d51a66b73b3e filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a98ecbfbd8d7648df2b2e18ae39d491c filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/b6a6b696cf78acc99881e05dae0be139 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/46f57c7bf2e8d3a2de0f1529bd48d657 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11edc3848fbb3f286f64ed77820e94d8 filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/04c4c100d0b538cf6f3fba1412a065da filter=lfs diff=lfs merge=lfs -text +robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/4f3754514ff44d4f2d30f5fa78479992 filter=lfs diff=lfs merge=lfs -text diff --git a/robocasa_test_all_atomic/15000/_CHECKPOINT_METADATA b/robocasa_test_all_atomic/15000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e447e47498bb33d5f9bb2fcaf87a3e8da38397e5 --- /dev/null +++ b/robocasa_test_all_atomic/15000/_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": 1775505214766818521, "commit_timestamp_nsecs": 1775505357251576053, "custom_metadata": {}} \ No newline at end of file diff --git a/robocasa_test_all_atomic/15000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json b/robocasa_test_all_atomic/15000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..a61f23a33c416ca9fb46fde8f50af856841be99a --- /dev/null +++ b/robocasa_test_all_atomic/15000/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/15000/params/_METADATA b/robocasa_test_all_atomic/15000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..9eb73624158f4514d49041e0806d2d702c62eee1 --- /dev/null +++ b/robocasa_test_all_atomic/15000/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/15000/params/_sharding b/robocasa_test_all_atomic/15000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..924608abdcb99abb3c864bb9d1bfca6eb60f780e --- /dev/null +++ b/robocasa_test_all_atomic/15000/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/15000/params/array_metadatas/process_0 b/robocasa_test_all_atomic/15000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..6aa9fd1decf72607133eb54664d98802bf563b5c --- /dev/null +++ b/robocasa_test_all_atomic/15000/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/15000/params/d/42e0f7b39b7c3a7b601d4ae907b39bff b/robocasa_test_all_atomic/15000/params/d/42e0f7b39b7c3a7b601d4ae907b39bff new file mode 100644 index 0000000000000000000000000000000000000000..8bad292882b9219979eaffd42d4bf879149a1b99 Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/d/42e0f7b39b7c3a7b601d4ae907b39bff differ diff --git a/robocasa_test_all_atomic/15000/params/manifest.ocdbt b/robocasa_test_all_atomic/15000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..75bd6d4999221331dfbda800645ac5089a9a680c Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/03985f56db209ed2f954d087cd975570 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/03985f56db209ed2f954d087cd975570 new file mode 100644 index 0000000000000000000000000000000000000000..1c7c0310dd2562e85e08d4b56ff38c4c96602c86 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/03985f56db209ed2f954d087cd975570 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9c294c0145e92b39daf0665afddf5d12d84abf9c6e57b9ce0f8aab1c85ef90c8 +size 280039152 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/127ef3507813b0971202deb3373b7c09 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/127ef3507813b0971202deb3373b7c09 new file mode 100644 index 0000000000000000000000000000000000000000..a7319c9712111fbade38def155e1b0874741a88a --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/127ef3507813b0971202deb3373b7c09 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:baea28386807c40c27e48cb0c63de47cc8c542d6a4dae2a2872f6e8c54d8179b +size 272882864 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/1d593c9d62349b6cbaf408ffca3427b9 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/1d593c9d62349b6cbaf408ffca3427b9 new file mode 100644 index 0000000000000000000000000000000000000000..451b0f339fc3473aa1808f513a6559a194670c05 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/1d593c9d62349b6cbaf408ffca3427b9 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1019f5ca92a8dd0041cf3a22740a9e3bd5d060ca6ff77590f1cdcf73ec0933d8 +size 34975319 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/2eb18c4bb4e92b99c18988caad47e6cc b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/2eb18c4bb4e92b99c18988caad47e6cc new file mode 100644 index 0000000000000000000000000000000000000000..3edf9a872eebe27ddba8a70a564ab0240be02107 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/2eb18c4bb4e92b99c18988caad47e6cc @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:88a3beb783ff1476395cc4e8a337b6928e3189ab90193c73c92a513abf0c7994 +size 132938410 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/490ed1095a0b8e9d63e7c541a2c73ce4 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/490ed1095a0b8e9d63e7c541a2c73ce4 new file mode 100644 index 0000000000000000000000000000000000000000..46cd46020c3fc37eed8126482d2295a61a10ce74 Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/490ed1095a0b8e9d63e7c541a2c73ce4 differ diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/4c4df646095fe62c2a02d86ed37a01e8 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/4c4df646095fe62c2a02d86ed37a01e8 new file mode 100644 index 0000000000000000000000000000000000000000..5f7ed5d9fd98562f4adf1e93e66dfbacf614945e --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/4c4df646095fe62c2a02d86ed37a01e8 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a8e3f0f36752b5bf16bd86d5ff571264b21f30c08fc995a9ce5bf0e6bdbd0acc +size 279246184 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/5858dcef70dab03c80c42498da149742 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/5858dcef70dab03c80c42498da149742 new file mode 100644 index 0000000000000000000000000000000000000000..29ed4d473ee2ed91f8fa6964ade4439f0654cbf4 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/5858dcef70dab03c80c42498da149742 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:09f417e1e0080c1b0435da5801b8cbd9ba50c87e7e800cdcf9aa580a1d12ebad +size 8754280 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/6a926ccfca27ec26a7f457c23e6eed97 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/6a926ccfca27ec26a7f457c23e6eed97 new file mode 100644 index 0000000000000000000000000000000000000000..0bb4eb08a985f58ebad5de0366df1a8ea18d48b8 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/6a926ccfca27ec26a7f457c23e6eed97 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:32398eda6653c4140c5047d79a78f578ee2f83babab766aed6beb6fc5ef847ce +size 1120115426 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/74a77a5d3ba3b3df71170d0899fb4f24 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/74a77a5d3ba3b3df71170d0899fb4f24 new file mode 100644 index 0000000000000000000000000000000000000000..0a60a5edf6dbd9a7345ef734951a14e7c9651ad6 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/74a77a5d3ba3b3df71170d0899fb4f24 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:2ad0b5f3506c4286bdd4a32578e94eec68449df2c43c7272dc21c7cb0d1a5621 +size 280187679 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/786c87c704c84afd277204b0536280cb b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/786c87c704c84afd277204b0536280cb new file mode 100644 index 0000000000000000000000000000000000000000..5e76788c88a6ff30b5800d9b3ae399708f1cc01a --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/786c87c704c84afd277204b0536280cb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8ab4a13b29b7d5cddc7d9e35ea9edd97860e80a5555a78d4d56044bb17ce1db2 +size 70082014 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/80414f95b12bb7cc6d6ad42c8d496cba b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/80414f95b12bb7cc6d6ad42c8d496cba new file mode 100644 index 0000000000000000000000000000000000000000..97f9b6c44da14b4194e7b9c9a778a380bbe965a0 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/80414f95b12bb7cc6d6ad42c8d496cba @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7f75862c68a9cdc566d602bdd217f178a14fa4322c01bde5de8aa404e4ca3506 +size 1120129558 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/88f55de2a4872e8248b0d51a66b73b3e b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/88f55de2a4872e8248b0d51a66b73b3e new file mode 100644 index 0000000000000000000000000000000000000000..a8819a39444de52ce6dcf3b40a422ab499a5eb58 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/88f55de2a4872e8248b0d51a66b73b3e @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:7ff4427762d84b570ddb19a2078aa7e87769a5e2b893211c428b2a397c9bcbb3 +size 1120117660 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/96ce1e9519714acfdae49abe9c4805bf b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/96ce1e9519714acfdae49abe9c4805bf new file mode 100644 index 0000000000000000000000000000000000000000..e6f8d16c7460074c8272b0882cc025c694245507 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/96ce1e9519714acfdae49abe9c4805bf @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4ebde10e7d3e2d673fce6456652d8f7286c5c0c22315e498a280c77be393a749 +size 496757957 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/9755e2ad714b5b5d0731af7033e82110 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/9755e2ad714b5b5d0731af7033e82110 new file mode 100644 index 0000000000000000000000000000000000000000..45ead22c58b526e74135b8f59cb47a0aaec42fbd --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/9755e2ad714b5b5d0731af7033e82110 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:999f5378cccaf6cdc7b58249889992a1a9cd470d31b75f7965bed20d545855ac +size 1120110240 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a492bd9a0adb7cf8fc5671d8c971ef70 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a492bd9a0adb7cf8fc5671d8c971ef70 new file mode 100644 index 0000000000000000000000000000000000000000..ffc57c5cefe3160b4f5ec2c8030ab168d9324180 Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a492bd9a0adb7cf8fc5671d8c971ef70 differ diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a4e70ae5855d738f5986a53b5f598954 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a4e70ae5855d738f5986a53b5f598954 new file mode 100644 index 0000000000000000000000000000000000000000..ae440c83d19bbba34bcf45d75c4392d0176a8118 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a4e70ae5855d738f5986a53b5f598954 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:65ab3e42244d0bf47feee3c861f7574878d377d5254b66669422d6e12e8431b5 +size 17544129 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a98ecbfbd8d7648df2b2e18ae39d491c b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a98ecbfbd8d7648df2b2e18ae39d491c new file mode 100644 index 0000000000000000000000000000000000000000..d7478db9bc9e212850c6f242dbb6d24da2d11b5b --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/a98ecbfbd8d7648df2b2e18ae39d491c @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f76a4edf021a88fa2f8bfa6026f3af632a569d4f743c096068901bc51fab1dfe +size 1120121083 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ab643f64ff47d16928047782f50742a8 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ab643f64ff47d16928047782f50742a8 new file mode 100644 index 0000000000000000000000000000000000000000..9859d7f854b6e83fb54aa5a57521992030b9a56e --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ab643f64ff47d16928047782f50742a8 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:908d1c27060fc9830f47a1bf3179214f1651fca05a0eb724ca2b228d6fcb5aae +size 132966422 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/af4edee3e7bf303c015284a2eedc89b5 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/af4edee3e7bf303c015284a2eedc89b5 new file mode 100644 index 0000000000000000000000000000000000000000..cc601269f5b410b258895a699376676e8490d712 Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/af4edee3e7bf303c015284a2eedc89b5 differ diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/b6a6b696cf78acc99881e05dae0be139 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/b6a6b696cf78acc99881e05dae0be139 new file mode 100644 index 0000000000000000000000000000000000000000..2c9af19975de333089e7a83323f3c1d0e047927e --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/b6a6b696cf78acc99881e05dae0be139 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b372db0e6b036ca6faf8fa1bdf413588d59ed14b6e54d20dc67c53b819efff72 +size 1957442678 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/cdce9ad7d2d6345dd8c163be67f4d297 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/cdce9ad7d2d6345dd8c163be67f4d297 new file mode 100644 index 0000000000000000000000000000000000000000..b39589a1708e92d223f44380e060084693870d40 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/cdce9ad7d2d6345dd8c163be67f4d297 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:46eb46ef5f7832035526b78a4dea2244f8e601230914df6ba0516be601ac22c9 +size 497032065 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/d5dc2666d2648f11f5682cf8870910cb b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/d5dc2666d2648f11f5682cf8870910cb new file mode 100644 index 0000000000000000000000000000000000000000..a3678d1e9157740c97da6d73a5026fd6102d06a8 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/d5dc2666d2648f11f5682cf8870910cb @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4f1f3518bfc0232e0aa2f893bd230d2367b0b06cda5e2cd89194dfffff49c774 +size 559728631 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/de453a99166108df1b4aef4d6fd5d785 b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/de453a99166108df1b4aef4d6fd5d785 new file mode 100644 index 0000000000000000000000000000000000000000..70f39e5297746dca2eb4c1db6829945e571a21c6 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/de453a99166108df1b4aef4d6fd5d785 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:55767ca978ecbd8f582cd9bf9a36844947f179e4aaad24c301331687dd0571f2 +size 1120137330 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/df70b8ed24ad4f39fdf4ea69efd2c52d b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/df70b8ed24ad4f39fdf4ea69efd2c52d new file mode 100644 index 0000000000000000000000000000000000000000..c662afc7492332f307623744d5cdc939a4d3479b --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/df70b8ed24ad4f39fdf4ea69efd2c52d @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:980230edb3aa834f71c0c8ae92aa6e5caf98be3a5f5f504d82a9632d3c4aa763 +size 132844982 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ffcb915dab3ba236d24c08cb9c088d6a b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ffcb915dab3ba236d24c08cb9c088d6a new file mode 100644 index 0000000000000000000000000000000000000000..02e0d7d3767e6d2e4ecd9457bf873c22ce884756 --- /dev/null +++ b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/d/ffcb915dab3ba236d24c08cb9c088d6a @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:094a67ad5423b37d10c9a54e4405ddf565dd42107024b729b2aa6cce7c7e23f1 +size 139970301 diff --git a/robocasa_test_all_atomic/15000/params/ocdbt.process_0/manifest.ocdbt b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..5098dff1223d271df62602a9b16ba7f6b7b61fbd Binary files /dev/null and b/robocasa_test_all_atomic/15000/params/ocdbt.process_0/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/17000/_CHECKPOINT_METADATA b/robocasa_test_all_atomic/17000/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..1588526075cd461a0280d9236cc244855ea68109 --- /dev/null +++ b/robocasa_test_all_atomic/17000/_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": 1775508214516380151, "commit_timestamp_nsecs": 1775508618632824527, "custom_metadata": {}} \ No newline at end of file diff --git a/robocasa_test_all_atomic/17000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json b/robocasa_test_all_atomic/17000/assets/robotgeneralist/AdjustToasterOvenTemperature_pretrain/norm_stats.json new file mode 100644 index 0000000000000000000000000000000000000000..a61f23a33c416ca9fb46fde8f50af856841be99a --- /dev/null +++ b/robocasa_test_all_atomic/17000/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/17000/params/_METADATA b/robocasa_test_all_atomic/17000/params/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..9eb73624158f4514d49041e0806d2d702c62eee1 --- /dev/null +++ b/robocasa_test_all_atomic/17000/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/17000/params/_sharding b/robocasa_test_all_atomic/17000/params/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..924608abdcb99abb3c864bb9d1bfca6eb60f780e --- /dev/null +++ b/robocasa_test_all_atomic/17000/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/17000/params/array_metadatas/process_0 b/robocasa_test_all_atomic/17000/params/array_metadatas/process_0 new file mode 100644 index 0000000000000000000000000000000000000000..6aa9fd1decf72607133eb54664d98802bf563b5c --- /dev/null +++ b/robocasa_test_all_atomic/17000/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/17000/params/d/6690cdeda98ab52dc8855c5f76097050 b/robocasa_test_all_atomic/17000/params/d/6690cdeda98ab52dc8855c5f76097050 new file mode 100644 index 0000000000000000000000000000000000000000..a75761302334152f2f15fa9817fce307553839df Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/d/6690cdeda98ab52dc8855c5f76097050 differ diff --git a/robocasa_test_all_atomic/17000/params/manifest.ocdbt b/robocasa_test_all_atomic/17000/params/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..642e9035ea2b6c198b1eb39c16c04fcd0ee38b49 Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/manifest.ocdbt differ diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/04c4c100d0b538cf6f3fba1412a065da b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/04c4c100d0b538cf6f3fba1412a065da new file mode 100644 index 0000000000000000000000000000000000000000..de090a30806650a4bbf84045c8a26ac36ee6f073 --- /dev/null +++ b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/04c4c100d0b538cf6f3fba1412a065da @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:54b70aed66caa3328fb93074cb033a2677daeecdae38eee580642c304a9cf0d5 +size 363163 diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11c320a276b53a3e6119a092fc1c8243 b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11c320a276b53a3e6119a092fc1c8243 new file mode 100644 index 0000000000000000000000000000000000000000..3ea8e46eaf6d2ae3c3cb73f80a88c16ec8637f28 Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11c320a276b53a3e6119a092fc1c8243 differ diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11edc3848fbb3f286f64ed77820e94d8 b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11edc3848fbb3f286f64ed77820e94d8 new file mode 100644 index 0000000000000000000000000000000000000000..da01bae0e8b84cd3d79148ead11e7a7e1d63aaee --- /dev/null +++ b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/11edc3848fbb3f286f64ed77820e94d8 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e58f26fd37949a903ee5b9cc4a1d209ab631d5e064bc7afb880eb4bb8cc5d312 +size 18048420 diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/46f57c7bf2e8d3a2de0f1529bd48d657 b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/46f57c7bf2e8d3a2de0f1529bd48d657 new file mode 100644 index 0000000000000000000000000000000000000000..8a8c58635aacbfc9124b03b3cee5f3af57c28f1a --- /dev/null +++ b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/46f57c7bf2e8d3a2de0f1529bd48d657 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:4cd81bf37b20610e6cc15930dc77eb9827dcdbc7c797d45302f482f6f4327e3a +size 122065 diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/4f3754514ff44d4f2d30f5fa78479992 b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/4f3754514ff44d4f2d30f5fa78479992 new file mode 100644 index 0000000000000000000000000000000000000000..da54da867c40d6cfdad3253f08a7056f2e864f76 --- /dev/null +++ b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/4f3754514ff44d4f2d30f5fa78479992 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bb59f8a6f23e973367928be7144ae423e1d2f12da105e56ff64881c0ae381647 +size 7767402 diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/a29261ee4500ac34c5f5ed465b5b38aa b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/a29261ee4500ac34c5f5ed465b5b38aa new file mode 100644 index 0000000000000000000000000000000000000000..46cd46020c3fc37eed8126482d2295a61a10ce74 Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/a29261ee4500ac34c5f5ed465b5b38aa differ diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/d2c589091a0ec5b4564a22ac6d86c66f b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/d2c589091a0ec5b4564a22ac6d86c66f new file mode 100644 index 0000000000000000000000000000000000000000..284fd3eef2439e04935ef930b4b361e42a78302e Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/d2c589091a0ec5b4564a22ac6d86c66f differ diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/db6996dc1b5bfef1d7fc99ea83de35fb b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/db6996dc1b5bfef1d7fc99ea83de35fb new file mode 100644 index 0000000000000000000000000000000000000000..42acb067b5ffa995fd6b2ec83f993b6d550f0084 Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/db6996dc1b5bfef1d7fc99ea83de35fb differ diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/ecc5647631ee9d5120212c025910ae4e b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/ecc5647631ee9d5120212c025910ae4e new file mode 100644 index 0000000000000000000000000000000000000000..e74d13cea629a449f18fc22b90ff5a32b49fbded --- /dev/null +++ b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/d/ecc5647631ee9d5120212c025910ae4e @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1535b3d06c89f026440189eb8f393d3e860e3e54832a6d7dec2885cf15db73e3 +size 1957442778 diff --git a/robocasa_test_all_atomic/17000/params/ocdbt.process_0/manifest.ocdbt b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..f913ebcf3f6dc150858367495d43cf2156a57a19 Binary files /dev/null and b/robocasa_test_all_atomic/17000/params/ocdbt.process_0/manifest.ocdbt differ