Upload folder using huggingface_hub

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754482481556950692, "commit_timestamp_nsecs": 1754482482299600869, "custom_metadata": {}}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/1939475727c6a178d094dff3166fcfaa

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/5e0fec74064308f464da38c98d7d4032

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/fd3ecfae94ae5fda919443607480304e

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754482745656180074, "commit_timestamp_nsecs": 1754482745724203672, "custom_metadata": {}}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1024/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1024/default/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/065857f3e80e65ea12c22fb44945c21d

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754482761612591674, "commit_timestamp_nsecs": 1754482761680843156, "custom_metadata": {}}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1088/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1088/default/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/cf27ab277ba2274fc5ad8c4bfc1ea6dc

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754482777568479437, "commit_timestamp_nsecs": 1754482777635080057, "custom_metadata": {}}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1152/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1152/default/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/9fad392636294d1a8763ec592a8dc2c4

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

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754482793532463650, "commit_timestamp_nsecs": 1754482793598547472, "custom_metadata": {}}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1216/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1216/default/_sharding

@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

wall-prob-tree-13x13/wp_0.1_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}