{
"checkpoint_directory": "50000",
"checkpoint_step": 50000,
"files": [
{
"byte_identical_to_source": true,
"path": "50000/_CHECKPOINT_METADATA",
"sha256": "e3026544b4e02facf6f7d852fe57b7c241cf3bef263067541a2bb44ad27bc674",
"size_bytes": 426,
"source_sha256": "e3026544b4e02facf6f7d852fe57b7c241cf3bef263067541a2bb44ad27bc674"
},
{
"byte_identical_to_source": true,
"path": "50000/assets/pi05_libero/norm_stats.json",
"sha256": "0c0d329a3345d2ea2e1348dac0edf4ae175085fac200cca1cab967aae1ae1767",
"size_bytes": 1874,
"source_sha256": "0c0d329a3345d2ea2e1348dac0edf4ae175085fac200cca1cab967aae1ae1767"
},
{
"byte_identical_to_source": false,
"path": "50000/assets/robomem_memory.json",
"sha256": "58b9f9b99bcec5098d8a9987cba2d0c241ecdfd9308388db14c3be11ecf7e5ce",
"size_bytes": 2090,
"source_sha256": "bf3acddb4e3d497dc34e3c0b086faa5554baec3f40f8c87780a2acd1fc5a6a06"
},
{
"byte_identical_to_source": true,
"path": "50000/params/_METADATA",
"sha256": "374a1306796f1f69a7fe7996dd61e7459a80e0646370d2228ad294860ecb0841",
"size_bytes": 77885,
"source_sha256": "374a1306796f1f69a7fe7996dd61e7459a80e0646370d2228ad294860ecb0841"
},
{
"byte_identical_to_source": true,
"path": "50000/params/_sharding",
"sha256": "847e82744fdeb7ad43ae3940f0ab3dbb336b7c16f2b1e24e3f4b8a4b24106bdc",
"size_bytes": 58527,
"source_sha256": "847e82744fdeb7ad43ae3940f0ab3dbb336b7c16f2b1e24e3f4b8a4b24106bdc"
},
{
"byte_identical_to_source": true,
"path": "50000/params/array_metadatas/process_0",
"sha256": "9a0a569010277416cdbeee1b7bcff8d5eda73cdb02c85dfdb246841ea14c8de9",
"size_bytes": 28305,
"source_sha256": "9a0a569010277416cdbeee1b7bcff8d5eda73cdb02c85dfdb246841ea14c8de9"
},
{
"byte_identical_to_source": true,
"path": "50000/params/d/7d9023ea1a42ac128d7a54bd3a95d5fa",
"sha256": "bfc579f0bc71333aafb9b0b67f050ff228525b0dccb8179d997e23b3abd35a9a",
"size_bytes": 268748,
"source_sha256": "bfc579f0bc71333aafb9b0b67f050ff228525b0dccb8179d997e23b3abd35a9a"
},
{
"byte_identical_to_source": true,
"path": "50000/params/manifest.ocdbt",
"sha256": "ba4ec2909f05dc3c399f56eb0fafd4f30b758b6d7a575d1766d387eb4f21afc5",
"size_bytes": 120,
"source_sha256": "ba4ec2909f05dc3c399f56eb0fafd4f30b758b6d7a575d1766d387eb4f21afc5"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/1edd4e8d94f7efddfdbb208ddedc388a",
"sha256": "35c662bc04f0bb3328674f84e993871bb52d0bbdff9fca7a6b7e314cf5fa9ad1",
"size_bytes": 204,
"source_sha256": "35c662bc04f0bb3328674f84e993871bb52d0bbdff9fca7a6b7e314cf5fa9ad1"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/227ff146edc9a927477ff7097b73bcd1",
"sha256": "a8f2f6271f68121a740f4e87d84fca725405091678ee84cb3c7442ed1be9c30e",
"size_bytes": 443,
"source_sha256": "a8f2f6271f68121a740f4e87d84fca725405091678ee84cb3c7442ed1be9c30e"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/23acae17f3af0149f4523126d559cf0b",
"sha256": "863cc9feae1f1f19282ad574474bc493a7f4c1a66ee9ffd00d3387995097f2cc",
"size_bytes": 2620,
"source_sha256": "863cc9feae1f1f19282ad574474bc493a7f4c1a66ee9ffd00d3387995097f2cc"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/25b7e6f31e300c3b2d4a9e2b4c538f9d",
"sha256": "48f59b2ad2551c134d710b0d0c2254b084994c62410232897e70403315bc07ef",
"size_bytes": 1680537192,
"source_sha256": "48f59b2ad2551c134d710b0d0c2254b084994c62410232897e70403315bc07ef"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/5d863ece6d46de5cbc0930f1a3997cd8",
"sha256": "ed9f6b4a4e4d8b214abeaf22c77fe8b4da91a3416d7ebde27ce672eac9381093",
"size_bytes": 2414062880,
"source_sha256": "ed9f6b4a4e4d8b214abeaf22c77fe8b4da91a3416d7ebde27ce672eac9381093"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/64f6f44ef8a8abd11864d5d395c45ff8",
"sha256": "6d8f8d4bbbb3d44922ca3f8831068489d478ed53289f2965f247947f3ea64e22",
"size_bytes": 2240348440,
"source_sha256": "6d8f8d4bbbb3d44922ca3f8831068489d478ed53289f2965f247947f3ea64e22"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/77abcea3bdcba70ca53f0b22bfa47b0d",
"sha256": "61001355f9aee385ad9326bb10bf6f6fc09ee39cd2be6aede69092184c1132f2",
"size_bytes": 951775140,
"source_sha256": "61001355f9aee385ad9326bb10bf6f6fc09ee39cd2be6aede69092184c1132f2"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/853f72a28504554bab43017754533dcc",
"sha256": "abfb391246a534434e577497e60799e3ae1ad9060f822a100d94983c2a4d6473",
"size_bytes": 12620693,
"source_sha256": "abfb391246a534434e577497e60799e3ae1ad9060f822a100d94983c2a4d6473"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/9c0484f0dd420f287201d23da040c669",
"sha256": "0a529931ecdff929a542cdae30cd624be1aa34c82a05882c4487d9403faf7256",
"size_bytes": 1498,
"source_sha256": "0a529931ecdff929a542cdae30cd624be1aa34c82a05882c4487d9403faf7256"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/ab379bd37f74c0e87ddba02d44100a8c",
"sha256": "591913bf353721fc771065c42b91698f75fda8b94a887a8990a2b2724e5aebb0",
"size_bytes": 2343563491,
"source_sha256": "591913bf353721fc771065c42b91698f75fda8b94a887a8990a2b2724e5aebb0"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/e4739f3902de861621dbe39cfc153205",
"sha256": "4250661ef4ca347fac3445ca31125e5e79782b17a141ee29b347d2c4a5f38dba",
"size_bytes": 2006112315,
"source_sha256": "4250661ef4ca347fac3445ca31125e5e79782b17a141ee29b347d2c4a5f38dba"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/e50c941de03f7c07acbdb2c0ca78e91d",
"sha256": "f3bc1a1aa69428a6bd3436e716b4303e542ba3f86ebc5f5d7de9bc36af299e84",
"size_bytes": 436,
"source_sha256": "f3bc1a1aa69428a6bd3436e716b4303e542ba3f86ebc5f5d7de9bc36af299e84"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/e8478528f279a7195ee86bbbe181aea6",
"sha256": "9a69508273c31cbd9b63c31b030544590669500f302cdf28533e6ed9f3ae4c68",
"size_bytes": 7274188,
"source_sha256": "9a69508273c31cbd9b63c31b030544590669500f302cdf28533e6ed9f3ae4c68"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/manifest.ocdbt",
"sha256": "1909df2edbcbdd1d8a7ee393e9c3ae1ef1d5196918a8dad9db6f4b4f94c7a757",
"size_bytes": 527,
"source_sha256": "1909df2edbcbdd1d8a7ee393e9c3ae1ef1d5196918a8dad9db6f4b4f94c7a757"
},
{
"byte_identical_to_source": false,
"path": "recipe.json",
"sha256": "e5f5e60dd6d6512b33054d5ae1ae23ddac1923ce1d719455cbe78898d2f0ecef",
"size_bytes": 14500,
"source_sha256": "d6c660e62a96bf60459c5cb661964f0ead989a498ce5e294155e016483271f89"
},
{
"byte_identical_to_source": true,
"path": "tokenizer/paligemma_tokenizer.model",
"sha256": "8986bb4f423f07f8c7f70d0dbe3526fb2316056c17bae71b1ea975e77a168fc6",
"size_bytes": 4264023,
"source_sha256": "8986bb4f423f07f8c7f70d0dbe3526fb2316056c17bae71b1ea975e77a168fc6"
}
],
"format": "Orbax PyTreeCheckpointer OCDBT",
"metadata_note": "Root _CHECKPOINT_METADATA is preserved byte-for-byte and retains the original train_state handler description; train_state is intentionally absent. Load params with the policy loader; this is not a training-resume checkpoint.",
"metadata_provenance": {
"50000/assets/robomem_memory.json": {
"change": "Replace training-host absolute data/checkpoint/cache paths with relative identifiers; retain all other fields and values.",
"release_sha256": "58b9f9b99bcec5098d8a9987cba2d0c241ecdfd9308388db14c3be11ecf7e5ce",
"source_sha256": "bf3acddb4e3d497dc34e3c0b086faa5554baec3f40f8c87780a2acd1fc5a6a06"
},
"recipe.json": {
"change": "Replace training-host absolute data/checkpoint/cache paths with relative identifiers; retain all other fields and values.",
"release_sha256": "e5f5e60dd6d6512b33054d5ae1ae23ddac1923ce1d719455cbe78898d2f0ecef",
"source_sha256": "d6c660e62a96bf60459c5cb661964f0ead989a498ce5e294155e016483271f89"
}
},
"model_id": "taskanchor-robomemarena-pi05-50k",
"package_type": "inference checkpoint; complete params and normalization assets; optimizer/train_state excluded",
"parameter_tree": [
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/0/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/0/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/0/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/0/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/0/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/0/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/0/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/0/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/0/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/1/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/1/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/1/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/1/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/1/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/1/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/1/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/1/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/1/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/2/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/2/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/2/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/2/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/2/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/2/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/2/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/2/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/2/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/3/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/3/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/3/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/3/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/3/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/3/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/3/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/3/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/3/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/4/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/4/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/4/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/4/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/4/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/4/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/4/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/4/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/4/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/5/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/5/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/5/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/5/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/5/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/5/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/5/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/5/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/5/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/6/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/6/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/6/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/6/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/6/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/6/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/6/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/6/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/6/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "adapter/blocks/7/fc1/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/7/fc1/kernel/value",
"shape": [
512,
2048
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/fc2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "adapter/blocks/7/fc2/kernel/value",
"shape": [
2048,
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/norm1/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/norm1/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/norm2/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/norm2/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/blocks/7/ttt/proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 262144,
"path": "adapter/blocks/7/ttt/proj/kernel/value",
"shape": [
512,
512
]
},
{
"dtype": "float32",
"elements": 1536,
"path": "adapter/blocks/7/ttt/qkv/bias/value",
"shape": [
1536
]
},
{
"dtype": "float32",
"elements": 786432,
"path": "adapter/blocks/7/ttt/qkv/kernel/value",
"shape": [
512,
1536
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/7/ttt/w1/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "adapter/blocks/7/ttt/w2/value",
"shape": [
1,
8,
64,
64
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/current_norm/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/current_norm/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 1152,
"path": "adapter/current_out_proj/bias/value",
"shape": [
1152
]
},
{
"dtype": "float32",
"elements": 589824,
"path": "adapter/current_out_proj/kernel/value",
"shape": [
512,
1152
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/input_proj/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 589824,
"path": "adapter/input_proj/kernel/value",
"shape": [
1152,
512
]
},
{
"dtype": "float32",
"elements": 1152,
"path": "adapter/output_norm/bias/value",
"shape": [
1152
]
},
{
"dtype": "float32",
"elements": 1152,
"path": "adapter/output_norm/scale/value",
"shape": [
1152
]
},
{
"dtype": "float32",
"elements": 1,
"path": "adapter/progress_head/bias/value",
"shape": [
1
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/progress_head/kernel/value",
"shape": [
512,
1
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/progress_norm/bias/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/progress_norm/scale/value",
"shape": [
512
]
},
{
"dtype": "float32",
"elements": 512,
"path": "adapter/progress_token/value",
"shape": [
1,
1,
512
]
},
{
"dtype": "bfloat16",
"elements": 1152,
"path": "backbone/PaliGemma/img/Transformer/encoder_norm/bias/value",
"shape": [
1152
]
},
{
"dtype": "bfloat16",
"elements": 1152,
"path": "backbone/PaliGemma/img/Transformer/encoder_norm/scale/value",
"shape": [
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/LayerNorm_0/bias/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/LayerNorm_0/scale/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/LayerNorm_1/bias/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/LayerNorm_1/scale/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 116208,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MlpBlock_0/Dense_0/bias/value",
"shape": [
27,
4304
]
},
{
"dtype": "bfloat16",
"elements": 133871616,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MlpBlock_0/Dense_0/kernel/value",
"shape": [
27,
1152,
4304
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MlpBlock_0/Dense_1/bias/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 133871616,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MlpBlock_0/Dense_1/kernel/value",
"shape": [
27,
4304,
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/key/bias/value",
"shape": [
27,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 35831808,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/key/kernel/value",
"shape": [
27,
1152,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/out/bias/value",
"shape": [
27,
1152
]
},
{
"dtype": "bfloat16",
"elements": 35831808,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/out/kernel/value",
"shape": [
27,
16,
72,
1152
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/query/bias/value",
"shape": [
27,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 35831808,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/query/kernel/value",
"shape": [
27,
1152,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 31104,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/value/bias/value",
"shape": [
27,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 35831808,
"path": "backbone/PaliGemma/img/Transformer/encoderblock/MultiHeadDotProductAttention_0/value/kernel/value",
"shape": [
27,
1152,
16,
72
]
},
{
"dtype": "bfloat16",
"elements": 1152,
"path": "backbone/PaliGemma/img/embedding/bias/value",
"shape": [
1152
]
},
{
"dtype": "bfloat16",
"elements": 677376,
"path": "backbone/PaliGemma/img/embedding/kernel/value",
"shape": [
14,
14,
3,
1152
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "backbone/PaliGemma/img/head/bias/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 2359296,
"path": "backbone/PaliGemma/img/head/kernel/value",
"shape": [
1152,
2048
]
},
{
"dtype": "bfloat16",
"elements": 294912,
"path": "backbone/PaliGemma/img/pos_embedding/value",
"shape": [
1,
256,
1152
]
},
{
"dtype": "float32",
"elements": 526647296,
"path": "backbone/PaliGemma/llm/embedder/input_embedding/value",
"shape": [
257152,
2048
]
},
{
"dtype": "float32",
"elements": 2048,
"path": "backbone/PaliGemma/llm/final_norm/scale/value",
"shape": [
2048
]
},
{
"dtype": "float32",
"elements": 3072,
"path": "backbone/PaliGemma/llm/final_norm_1/Dense_0/bias/value",
"shape": [
3072
]
},
{
"dtype": "float32",
"elements": 3145728,
"path": "backbone/PaliGemma/llm/final_norm_1/Dense_0/kernel/value",
"shape": [
1024,
3072
]
},
{
"dtype": "float32",
"elements": 75497472,
"path": "backbone/PaliGemma/llm/layers/attn/attn_vec_einsum/w/value",
"shape": [
18,
8,
256,
2048
]
},
{
"dtype": "float32",
"elements": 37748736,
"path": "backbone/PaliGemma/llm/layers/attn/attn_vec_einsum_1/w/value",
"shape": [
18,
8,
256,
1024
]
},
{
"dtype": "float32",
"elements": 18874368,
"path": "backbone/PaliGemma/llm/layers/attn/kv_einsum/w/value",
"shape": [
18,
2,
1,
2048,
256
]
},
{
"dtype": "float32",
"elements": 9437184,
"path": "backbone/PaliGemma/llm/layers/attn/kv_einsum_1/w/value",
"shape": [
18,
2,
1,
1024,
256
]
},
{
"dtype": "float32",
"elements": 75497472,
"path": "backbone/PaliGemma/llm/layers/attn/q_einsum/w/value",
"shape": [
18,
8,
2048,
256
]
},
{
"dtype": "float32",
"elements": 37748736,
"path": "backbone/PaliGemma/llm/layers/attn/q_einsum_1/w/value",
"shape": [
18,
8,
1024,
256
]
},
{
"dtype": "float32",
"elements": 1207959552,
"path": "backbone/PaliGemma/llm/layers/mlp/gating_einsum/value",
"shape": [
18,
2,
2048,
16384
]
},
{
"dtype": "float32",
"elements": 603979776,
"path": "backbone/PaliGemma/llm/layers/mlp/linear/value",
"shape": [
18,
16384,
2048
]
},
{
"dtype": "float32",
"elements": 150994944,
"path": "backbone/PaliGemma/llm/layers/mlp_1/gating_einsum/value",
"shape": [
18,
2,
1024,
4096
]
},
{
"dtype": "float32",
"elements": 75497472,
"path": "backbone/PaliGemma/llm/layers/mlp_1/linear/value",
"shape": [
18,
4096,
1024
]
},
{
"dtype": "float32",
"elements": 36864,
"path": "backbone/PaliGemma/llm/layers/pre_attention_norm/scale/value",
"shape": [
18,
2048
]
},
{
"dtype": "float32",
"elements": 55296,
"path": "backbone/PaliGemma/llm/layers/pre_attention_norm_1/Dense_0/bias/value",
"shape": [
18,
3072
]
},
{
"dtype": "float32",
"elements": 56623104,
"path": "backbone/PaliGemma/llm/layers/pre_attention_norm_1/Dense_0/kernel/value",
"shape": [
18,
1024,
3072
]
},
{
"dtype": "float32",
"elements": 36864,
"path": "backbone/PaliGemma/llm/layers/pre_ffw_norm/scale/value",
"shape": [
18,
2048
]
},
{
"dtype": "float32",
"elements": 55296,
"path": "backbone/PaliGemma/llm/layers/pre_ffw_norm_1/Dense_0/bias/value",
"shape": [
18,
3072
]
},
{
"dtype": "float32",
"elements": 56623104,
"path": "backbone/PaliGemma/llm/layers/pre_ffw_norm_1/Dense_0/kernel/value",
"shape": [
18,
1024,
3072
]
},
{
"dtype": "float32",
"elements": 1024,
"path": "backbone/action_in_proj/bias/value",
"shape": [
1024
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "backbone/action_in_proj/kernel/value",
"shape": [
32,
1024
]
},
{
"dtype": "float32",
"elements": 32,
"path": "backbone/action_out_proj/bias/value",
"shape": [
32
]
},
{
"dtype": "float32",
"elements": 32768,
"path": "backbone/action_out_proj/kernel/value",
"shape": [
1024,
32
]
},
{
"dtype": "float32",
"elements": 1024,
"path": "backbone/time_mlp_in/bias/value",
"shape": [
1024
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "backbone/time_mlp_in/kernel/value",
"shape": [
1024,
1024
]
},
{
"dtype": "float32",
"elements": 1024,
"path": "backbone/time_mlp_out/bias/value",
"shape": [
1024
]
},
{
"dtype": "float32",
"elements": 1048576,
"path": "backbone/time_mlp_out/kernel/value",
"shape": [
1024,
1024
]
}
],
"payload_file_count": 24,
"schema_version": 1,
"source_run": "main_v5_spatial4x4_h191_a20_exec10_bs256_5k50k",
"tokenizer": {
"path": "tokenizer/paligemma_tokenizer.model",
"required": true,
"upstream_uri": "gs://big_vision/paligemma_tokenizer.model"
},
"total_payload_bytes": 11661016565,
"validation": {
"all_parameters_finite": true,
"full_model_forward_test": "not run",
"independent_destination_inodes": true,
"jax_devices": [
"TFRT_CPU_0"
],
"jax_version": "0.5.3",
"orbax_restore": "passed",
"orbax_version": "0.11.13",
"parameter_array_count": 176,
"parameter_count": 3380363921,
"parameter_elements_by_dtype": {
"bfloat16": 412442352,
"float32": 2967921569
},
"parameter_elements_by_root": {
"adapter": 26930049,
"backbone": 3353433872
},
"restore_and_finiteness_elapsed_seconds": 12.69,
"restored_on": "CPU only",
"simulator_rollout_test": "not run",
"source_file_hash_comparison": "passed for every unmodified copied file",
"uncompressed_parameter_bytes": 12696570980
}
}
{
"checkpoint_directory": "50000",
"checkpoint_step": 50000,
"files": [
{
"byte_identical_to_source": true,
"path": "50000/_CHECKPOINT_METADATA",
"sha256": "e3026544b4e02facf6f7d852fe57b7c241cf3bef263067541a2bb44ad27bc674",
"size_bytes": 426,
"source_sha256": "e3026544b4e02facf6f7d852fe57b7c241cf3bef263067541a2bb44ad27bc674"
},
{
"byte_identical_to_source": true,
"path": "50000/assets/pi05_libero/norm_stats.json",
"sha256": "0c0d329a3345d2ea2e1348dac0edf4ae175085fac200cca1cab967aae1ae1767",
"size_bytes": 1874,
"source_sha256": "0c0d329a3345d2ea2e1348dac0edf4ae175085fac200cca1cab967aae1ae1767"
},
{
"byte_identical_to_source": false,
"path": "50000/assets/robomem_memory.json",
"sha256": "58b9f9b99bcec5098d8a9987cba2d0c241ecdfd9308388db14c3be11ecf7e5ce",
"size_bytes": 2090,
"source_sha256": "bf3acddb4e3d497dc34e3c0b086faa5554baec3f40f8c87780a2acd1fc5a6a06"
},
{
"byte_identical_to_source": true,
"path": "50000/params/_METADATA",
"sha256": "374a1306796f1f69a7fe7996dd61e7459a80e0646370d2228ad294860ecb0841",
"size_bytes": 77885,
"source_sha256": "374a1306796f1f69a7fe7996dd61e7459a80e0646370d2228ad294860ecb0841"
},
{
"byte_identical_to_source": true,
"path": "50000/params/_sharding",
"sha256": "847e82744fdeb7ad43ae3940f0ab3dbb336b7c16f2b1e24e3f4b8a4b24106bdc",
"size_bytes": 58527,
"source_sha256": "847e82744fdeb7ad43ae3940f0ab3dbb336b7c16f2b1e24e3f4b8a4b24106bdc"
},
{
"byte_identical_to_source": true,
"path": "50000/params/array_metadatas/process_0",
"sha256": "9a0a569010277416cdbeee1b7bcff8d5eda73cdb02c85dfdb246841ea14c8de9",
"size_bytes": 28305,
"source_sha256": "9a0a569010277416cdbeee1b7bcff8d5eda73cdb02c85dfdb246841ea14c8de9"
},
{
"byte_identical_to_source": true,
"path": "50000/params/d/7d9023ea1a42ac128d7a54bd3a95d5fa",
"sha256": "bfc579f0bc71333aafb9b0b67f050ff228525b0dccb8179d997e23b3abd35a9a",
"size_bytes": 268748,
"source_sha256": "bfc579f0bc71333aafb9b0b67f050ff228525b0dccb8179d997e23b3abd35a9a"
},
{
"byte_identical_to_source": true,
"path": "50000/params/manifest.ocdbt",
"sha256": "ba4ec2909f05dc3c399f56eb0fafd4f30b758b6d7a575d1766d387eb4f21afc5",
"size_bytes": 120,
"source_sha256": "ba4ec2909f05dc3c399f56eb0fafd4f30b758b6d7a575d1766d387eb4f21afc5"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/1edd4e8d94f7efddfdbb208ddedc388a",
"sha256": "35c662bc04f0bb3328674f84e993871bb52d0bbdff9fca7a6b7e314cf5fa9ad1",
"size_bytes": 204,
"source_sha256": "35c662bc04f0bb3328674f84e993871bb52d0bbdff9fca7a6b7e314cf5fa9ad1"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/227ff146edc9a927477ff7097b73bcd1",
"sha256": "a8f2f6271f68121a740f4e87d84fca725405091678ee84cb3c7442ed1be9c30e",
"size_bytes": 443,
"source_sha256": "a8f2f6271f68121a740f4e87d84fca725405091678ee84cb3c7442ed1be9c30e"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/23acae17f3af0149f4523126d559cf0b",
"sha256": "863cc9feae1f1f19282ad574474bc493a7f4c1a66ee9ffd00d3387995097f2cc",
"size_bytes": 2620,
"source_sha256": "863cc9feae1f1f19282ad574474bc493a7f4c1a66ee9ffd00d3387995097f2cc"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/25b7e6f31e300c3b2d4a9e2b4c538f9d",
"sha256": "48f59b2ad2551c134d710b0d0c2254b084994c62410232897e70403315bc07ef",
"size_bytes": 1680537192,
"source_sha256": "48f59b2ad2551c134d710b0d0c2254b084994c62410232897e70403315bc07ef"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/5d863ece6d46de5cbc0930f1a3997cd8",
"sha256": "ed9f6b4a4e4d8b214abeaf22c77fe8b4da91a3416d7ebde27ce672eac9381093",
"size_bytes": 2414062880,
"source_sha256": "ed9f6b4a4e4d8b214abeaf22c77fe8b4da91a3416d7ebde27ce672eac9381093"
},
{
"byte_identical_to_source": true,
"path": "50000/params/ocdbt.process_0/d/64f6f44ef8a8abd11864d5d395c45ff8",
"sha256": "6d8f8d4bbbb3d44922ca3f8831068489d478ed53289f2965f247947f3ea64e22",
"size_bytes": 2240348440,