diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/manifest.ocdbt b/logs/G1Cat_lowcorner/checkpoints/000403046400/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..23a4bd7058bcef596c468dcd90061629a6b35775 Binary files /dev/null and b/logs/G1Cat_lowcorner/checkpoints/000403046400/manifest.ocdbt differ diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/323acee8e6a3f3ca0d3a5cb8f22e43f4 b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/323acee8e6a3f3ca0d3a5cb8f22e43f4 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/323acee8e6a3f3ca0d3a5cb8f22e43f4 differ diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/7a7d38bac1b40a94b067d8c2237b533d b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/7a7d38bac1b40a94b067d8c2237b533d new file mode 100644 index 0000000000000000000000000000000000000000..8c57b857662e0e792d220b650b6f54deda54512a --- /dev/null +++ b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/7a7d38bac1b40a94b067d8c2237b533d @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:8e4aba044ed8255d9af2b5eb5ed40af04831f2474b099e1994e87659962c35ee +size 1289717 diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/a1869db8b4c59eac311be780d59d1d28 b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/a1869db8b4c59eac311be780d59d1d28 new file mode 100644 index 0000000000000000000000000000000000000000..019f558dad68b4731a9596110ade2c191143dddc --- /dev/null +++ b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/a1869db8b4c59eac311be780d59d1d28 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:9ec4f021d2f2a78f8efffecaaf72563639c8beb3dc3e2e4ae086b50fc9c1bb9d +size 123143 diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/b35034fdc0024029c043319863129723 b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/b35034fdc0024029c043319863129723 new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/d/b35034fdc0024029c043319863129723 differ diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..5be8d277b92dd28fea5d6c412b46cad0fd44533e Binary files /dev/null and b/logs/G1Cat_lowcorner/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/onnx_convert.png b/logs/G1Cat_lowcorner/checkpoints/000403046400/onnx_convert.png new file mode 100644 index 0000000000000000000000000000000000000000..07de3c48155601a10760fe8b8f4cb4f632e9c69d Binary files /dev/null and b/logs/G1Cat_lowcorner/checkpoints/000403046400/onnx_convert.png differ diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/policy.onnx b/logs/G1Cat_lowcorner/checkpoints/000403046400/policy.onnx new file mode 100644 index 0000000000000000000000000000000000000000..c71e48dac958c8c85e6fffb2d918621f1d316e40 --- /dev/null +++ b/logs/G1Cat_lowcorner/checkpoints/000403046400/policy.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:238191cdc34cfc478c479d95a4691bf3ff686bb2b42ca6d3f17ae9a69f1fb8bf +size 339955 diff --git a/logs/G1Cat_lowcorner/checkpoints/000403046400/ppo_network_config.json b/logs/G1Cat_lowcorner/checkpoints/000403046400/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_lowcorner/checkpoints/000403046400/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/_CHECKPOINT_METADATA b/logs/G1Cat_narrow0/checkpoints/000080609280/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..87c1f3b37efdcd9eb65cadfcd2ae4ccba3af3e3b --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000080609280/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765785045005387910, "commit_timestamp_nsecs": 1765785045211318781, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/_METADATA b/logs/G1Cat_narrow0/checkpoints/000080609280/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000080609280/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/_sharding b/logs/G1Cat_narrow0/checkpoints/000080609280/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000080609280/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/d/d187877b49516297c074290a9746e09b b/logs/G1Cat_narrow0/checkpoints/000080609280/d/d187877b49516297c074290a9746e09b new file mode 100644 index 0000000000000000000000000000000000000000..287157c2064bf7bff72e66bc082af014ff208a5f Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/d/d187877b49516297c074290a9746e09b differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000080609280/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..a2dd68dcaaee74bd541c33be66a9054bad30db81 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/66eb95289240fb64aee29be84778ee49 b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/66eb95289240fb64aee29be84778ee49 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/66eb95289240fb64aee29be84778ee49 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/68f0838f7a4c71bd9f328af4c5819bbd b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/68f0838f7a4c71bd9f328af4c5819bbd new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/68f0838f7a4c71bd9f328af4c5819bbd differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a6fd456569e5a1af65ccbea3c054c750 b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a6fd456569e5a1af65ccbea3c054c750 new file mode 100644 index 0000000000000000000000000000000000000000..5a59acf1f97e1622462e9c6b09fecdb803c70840 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a6fd456569e5a1af65ccbea3c054c750 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b6115c9981f68796854613aa4141ef38428963e19a19c1caa79f427e8beca921 +size 1402772 diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a7a964716dfee960e0ccb98acbe0b34b b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a7a964716dfee960e0ccb98acbe0b34b new file mode 100644 index 0000000000000000000000000000000000000000..c1f445805c1da0dc3f02885377fe8ba8d727b6c8 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/d/a7a964716dfee960e0ccb98acbe0b34b differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..c6d2858ad1d6ad6c4539e289493c635442fabae2 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000080609280/ppo_network_config.json b/logs/G1Cat_narrow0/checkpoints/000080609280/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000080609280/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/_CHECKPOINT_METADATA b/logs/G1Cat_narrow0/checkpoints/000161218560/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..f6f76431fecec17280d405c4766c8b3399c66db1 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000161218560/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765786328050838021, "commit_timestamp_nsecs": 1765786328253030829, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/_METADATA b/logs/G1Cat_narrow0/checkpoints/000161218560/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000161218560/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/_sharding b/logs/G1Cat_narrow0/checkpoints/000161218560/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000161218560/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/d/21ba79bea1abab52ba47f016fdd62640 b/logs/G1Cat_narrow0/checkpoints/000161218560/d/21ba79bea1abab52ba47f016fdd62640 new file mode 100644 index 0000000000000000000000000000000000000000..958384ad06104fface65a0e7ea7e6afa8f1cc08d Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/d/21ba79bea1abab52ba47f016fdd62640 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000161218560/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..78cf150a5294e17321fcdc78c2f298c4166cb75a Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/4c7f4f64fd52886c410238be6763f817 b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/4c7f4f64fd52886c410238be6763f817 new file mode 100644 index 0000000000000000000000000000000000000000..3f58bd1cea1a27bc9e1a03487ac5dccbfc2f5e27 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/4c7f4f64fd52886c410238be6763f817 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b01a4af70abc90a274d8dd3c8c6dabdc30bdf96edeb95227679305319adc158e +size 1405796 diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/67c23fe60900968449e87c0359b6adb7 b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/67c23fe60900968449e87c0359b6adb7 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/67c23fe60900968449e87c0359b6adb7 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/7ec8ec0922c7b64cfa88820b8fc53b36 b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/7ec8ec0922c7b64cfa88820b8fc53b36 new file mode 100644 index 0000000000000000000000000000000000000000..a93c5fe587f8c7b996c2cd4b8c8b1f892174a24d Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/7ec8ec0922c7b64cfa88820b8fc53b36 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/dae246874686f2f95705c56286dfe5b6 b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/dae246874686f2f95705c56286dfe5b6 new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/d/dae246874686f2f95705c56286dfe5b6 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..f959a1be8c0971e2e305697a5a29ba51e3e77245 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000161218560/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000161218560/ppo_network_config.json b/logs/G1Cat_narrow0/checkpoints/000161218560/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000161218560/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/_CHECKPOINT_METADATA b/logs/G1Cat_narrow0/checkpoints/000241827840/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e52a224eedd55bf63ab9920719b2f9e164882fc7 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000241827840/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765787612541178721, "commit_timestamp_nsecs": 1765787612747394041, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/_METADATA b/logs/G1Cat_narrow0/checkpoints/000241827840/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000241827840/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/_sharding b/logs/G1Cat_narrow0/checkpoints/000241827840/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000241827840/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/d/e52e15a61123c445cf75bb73bbaa19a2 b/logs/G1Cat_narrow0/checkpoints/000241827840/d/e52e15a61123c445cf75bb73bbaa19a2 new file mode 100644 index 0000000000000000000000000000000000000000..ebfeeadcd566c01ad2805e3874ef6e0c95a26df4 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/d/e52e15a61123c445cf75bb73bbaa19a2 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000241827840/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..2ac10807bbe2f70b4a54d8d3375e1e711b34fddf Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/6afb96f538796d1c113c448cb7c87c49 b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/6afb96f538796d1c113c448cb7c87c49 new file mode 100644 index 0000000000000000000000000000000000000000..f8f440459a87a759c4632019f0494955f0a10227 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/6afb96f538796d1c113c448cb7c87c49 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:837a5759771aee3a168698d4b364fdf05f97b66076a487db40b1c8ffcdadff4d +size 1408078 diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/849518442d7fcbb386f072409a2fe3aa b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/849518442d7fcbb386f072409a2fe3aa new file mode 100644 index 0000000000000000000000000000000000000000..08b8a326ef93c39403f6a4ca0be16c2ac28956d0 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/849518442d7fcbb386f072409a2fe3aa differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/8aa74ee6bd073031ffec3912f47d825b b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/8aa74ee6bd073031ffec3912f47d825b new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/8aa74ee6bd073031ffec3912f47d825b differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/c96fa1fac7a3e760c241d599a51fe612 b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/c96fa1fac7a3e760c241d599a51fe612 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/d/c96fa1fac7a3e760c241d599a51fe612 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..09e4e3e58bb0479f6d257c0ea70a86243f0c398d Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000241827840/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000241827840/ppo_network_config.json b/logs/G1Cat_narrow0/checkpoints/000241827840/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000241827840/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/_CHECKPOINT_METADATA b/logs/G1Cat_narrow0/checkpoints/000322437120/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..a3bad80b3256481759b24712ba734bf5906997b7 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000322437120/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765788896092264687, "commit_timestamp_nsecs": 1765788896293402840, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/_METADATA b/logs/G1Cat_narrow0/checkpoints/000322437120/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000322437120/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/_sharding b/logs/G1Cat_narrow0/checkpoints/000322437120/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000322437120/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/d/ec33c4f11719b1c8aa238ce8304dd13c b/logs/G1Cat_narrow0/checkpoints/000322437120/d/ec33c4f11719b1c8aa238ce8304dd13c new file mode 100644 index 0000000000000000000000000000000000000000..ea978f111ab1ca879f6317389693f234e83db1a9 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/d/ec33c4f11719b1c8aa238ce8304dd13c differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000322437120/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..f90af27c6ac52084ad98a00d165dd2b13002ed3a Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/5b60da211331ade627c1a85c3b24910c b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/5b60da211331ade627c1a85c3b24910c new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/5b60da211331ade627c1a85c3b24910c differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/69eeec01480a4bad797f0e38f1d47487 b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/69eeec01480a4bad797f0e38f1d47487 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/69eeec01480a4bad797f0e38f1d47487 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/6a8bc9544e74ce0eb8a7fca702dc36b9 b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/6a8bc9544e74ce0eb8a7fca702dc36b9 new file mode 100644 index 0000000000000000000000000000000000000000..0215261f795b6c928650c5558e1414e7d32871fd --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/6a8bc9544e74ce0eb8a7fca702dc36b9 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:eec52f47dc0abb5baf2ee903638853c89644449d5c9845b95584ca244ded06fb +size 1409347 diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/9762ad0c495e2adeb39bfd1e4d161660 b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/9762ad0c495e2adeb39bfd1e4d161660 new file mode 100644 index 0000000000000000000000000000000000000000..5e9edf488966010cf1964d3f23a4edbe0c31419e Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/d/9762ad0c495e2adeb39bfd1e4d161660 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..0d1da83c820a1c93b9ccb9965e38ff7bc17e468a Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000322437120/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000322437120/ppo_network_config.json b/logs/G1Cat_narrow0/checkpoints/000322437120/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000322437120/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/_CHECKPOINT_METADATA b/logs/G1Cat_narrow0/checkpoints/000403046400/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..0b82e3bf092cabe1b86f8893685a5af299bf5114 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765790181753527674, "commit_timestamp_nsecs": 1765790181963166551, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/_METADATA b/logs/G1Cat_narrow0/checkpoints/000403046400/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/_sharding b/logs/G1Cat_narrow0/checkpoints/000403046400/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/d/f498a2fbf8d44da2150ecb28708bc3bf b/logs/G1Cat_narrow0/checkpoints/000403046400/d/f498a2fbf8d44da2150ecb28708bc3bf new file mode 100644 index 0000000000000000000000000000000000000000..5e1c96611bb18e59efd2bb1b083bd8fa24a1039e Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/d/f498a2fbf8d44da2150ecb28708bc3bf differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000403046400/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..3290fa8562ffee1eb522aec91bd67287818640b9 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/2e46450abd6b44233efab4bb3974cbc6 b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/2e46450abd6b44233efab4bb3974cbc6 new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/2e46450abd6b44233efab4bb3974cbc6 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/834c58ca1beec00de517d8288b6dd51e b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/834c58ca1beec00de517d8288b6dd51e new file mode 100644 index 0000000000000000000000000000000000000000..a038786df954132e66dd0e3cde123c4d6141ddbf Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/834c58ca1beec00de517d8288b6dd51e differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/eaddf003d9ffd28ec9c62c53090c1280 b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/eaddf003d9ffd28ec9c62c53090c1280 new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/eaddf003d9ffd28ec9c62c53090c1280 differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/fb4941ede5442107f069f3096058d78c b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/fb4941ede5442107f069f3096058d78c new file mode 100644 index 0000000000000000000000000000000000000000..a1dc3365434a1ad41d1ebfb94271be2b87f59fc9 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/d/fb4941ede5442107f069f3096058d78c @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c917ae577544e38e7e5618f9afcd3aaa4682797f4168ed1f800eb34042a5849e +size 1410077 diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..81679b1f9fb2b750d3da480300de77eb6f73200c Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/onnx_convert.png b/logs/G1Cat_narrow0/checkpoints/000403046400/onnx_convert.png new file mode 100644 index 0000000000000000000000000000000000000000..5c55585ba4c93ece61e877d74bd059d5186a697e Binary files /dev/null and b/logs/G1Cat_narrow0/checkpoints/000403046400/onnx_convert.png differ diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/policy.onnx b/logs/G1Cat_narrow0/checkpoints/000403046400/policy.onnx new file mode 100644 index 0000000000000000000000000000000000000000..a1f6e7be3bf7d50c587466072843518815a21198 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/policy.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:231904ce3abfe14b23d81b86b38a0d38a13e9930c311981e1db345e4814f5909 +size 339955 diff --git a/logs/G1Cat_narrow0/checkpoints/000403046400/ppo_network_config.json b/logs/G1Cat_narrow0/checkpoints/000403046400/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow0/checkpoints/000403046400/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/_CHECKPOINT_METADATA b/logs/G1Cat_narrow1/checkpoints/000080609280/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..542b91f0886a7a3a003fbcbae480fa76f5b3eb96 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000080609280/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765785062520065258, "commit_timestamp_nsecs": 1765785062728072909, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/_METADATA b/logs/G1Cat_narrow1/checkpoints/000080609280/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000080609280/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/_sharding b/logs/G1Cat_narrow1/checkpoints/000080609280/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000080609280/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/d/88bc83f978268c22a1c54f55592f5b89 b/logs/G1Cat_narrow1/checkpoints/000080609280/d/88bc83f978268c22a1c54f55592f5b89 new file mode 100644 index 0000000000000000000000000000000000000000..61d928b4945463a95b8582b645129b81c9cda07d Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/d/88bc83f978268c22a1c54f55592f5b89 differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/manifest.ocdbt b/logs/G1Cat_narrow1/checkpoints/000080609280/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..26651c59f43b5cfb169044ddd4fba0f9b4e90858 Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/301be85581daa706ab0ced904b8fa8d9 b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/301be85581daa706ab0ced904b8fa8d9 new file mode 100644 index 0000000000000000000000000000000000000000..aac03eb0cf3ffb7c3d17be2a3344bc2212be2eb1 Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/301be85581daa706ab0ced904b8fa8d9 differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/41dba4b18fa700de1fc1cb9a07c19daf b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/41dba4b18fa700de1fc1cb9a07c19daf new file mode 100644 index 0000000000000000000000000000000000000000..c3fee691f10e263b6bfa3a06ed5c0bf90512fc06 Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/41dba4b18fa700de1fc1cb9a07c19daf differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/55ad0521b630c56140a71c3d9d504d7c b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/55ad0521b630c56140a71c3d9d504d7c new file mode 100644 index 0000000000000000000000000000000000000000..341e2759e3ca9d67aa4e7b61180a45392af7ddf7 Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/55ad0521b630c56140a71c3d9d504d7c differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/da8d841ae52b349e0c9e5031d7af3023 b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/da8d841ae52b349e0c9e5031d7af3023 new file mode 100644 index 0000000000000000000000000000000000000000..e32753817e8adb4dc5be9c2010168496faa48785 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/d/da8d841ae52b349e0c9e5031d7af3023 @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f49e2552b440b9896c444bb92b755b74b9db431de5cf95825f59a72512d14daa +size 1403618 diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt new file mode 100644 index 0000000000000000000000000000000000000000..3ea739432dc3b0c2eb278f846006bd257e79d1c2 Binary files /dev/null and b/logs/G1Cat_narrow1/checkpoints/000080609280/ocdbt.process_0/manifest.ocdbt differ diff --git a/logs/G1Cat_narrow1/checkpoints/000080609280/ppo_network_config.json b/logs/G1Cat_narrow1/checkpoints/000080609280/ppo_network_config.json new file mode 100644 index 0000000000000000000000000000000000000000..61353f67b8549927153fe2414414b050b804bca2 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000080609280/ppo_network_config.json @@ -0,0 +1 @@ +{"network_factory_kwargs": {"policy_hidden_layer_sizes": [256, 128, 64], "value_hidden_layer_sizes": [512, 256, 128], "policy_obs_key": "state", "value_obs_key": "privileged_state"}, "normalize_observations": false, "observation_size": {"privileged_state": [250], "state": [162]}, "action_size": 12} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000161218560/_CHECKPOINT_METADATA b/logs/G1Cat_narrow1/checkpoints/000161218560/_CHECKPOINT_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..7915d90fa807db0c6a6252d6b18a2a2d5b3f0e15 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000161218560/_CHECKPOINT_METADATA @@ -0,0 +1 @@ +{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1765786359538787635, "commit_timestamp_nsecs": 1765786359742891585, "custom": {}} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000161218560/_METADATA b/logs/G1Cat_narrow1/checkpoints/000161218560/_METADATA new file mode 100644 index 0000000000000000000000000000000000000000..e876cfc20d891d6a26877d967ba6c8e3f6250cc2 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000161218560/_METADATA @@ -0,0 +1 @@ +{"tree_metadata": {"('0', 'mean', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'mean', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mean", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'std', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "std", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'privileged_state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "privileged_state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('0', 'summed_variance', 'state')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "summed_variance", "key_type": 2}, {"key": "state", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('1', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_0', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_1', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_2', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'bias')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('2', 'params', 'hidden_3', 'kernel')": {"key_metadata": [{"key": "2", "key_type": 1}, {"key": "params", "key_type": 2}, {"key": "hidden_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null} \ No newline at end of file diff --git a/logs/G1Cat_narrow1/checkpoints/000161218560/_sharding b/logs/G1Cat_narrow1/checkpoints/000161218560/_sharding new file mode 100644 index 0000000000000000000000000000000000000000..3ef68a54ca384161d8f160ec2522e945adb6fb86 --- /dev/null +++ b/logs/G1Cat_narrow1/checkpoints/000161218560/_sharding @@ -0,0 +1 @@ +{"MC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnByaXZpbGVnZWRfc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5tZWFuLnN0YXRl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdGQucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2Uuc3RhdGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MC5zdW1tZWRfdmFyaWFuY2UucHJpdmlsZWdlZF9zdGF0ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","MS5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Mi5wYXJhbXMuaGlkZGVuXzMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"} \ No newline at end of file