Upload folder using huggingface_hub

wall-prob-tree-13x13/debug.log

@@ -0,0 +1,28 @@
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Configure stats pid to 3869025
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():830] calling init triggers
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 1.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 0.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():871] starting backend
+2025-08-07 18:02:11,650 INFO    MainThread:3869025 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 18:02:11,656 INFO    MainThread:3869025 [wandb_init.py:init():882] backend started and connected
+2025-08-07 18:02:11,660 INFO    MainThread:3869025 [wandb_init.py:init():953] updated telemetry
+2025-08-07 18:02:11,669 INFO    MainThread:3869025 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 18:02:12,236 INFO    MainThread:3869025 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 18:02:12,578 INFO    MainThread:3869025 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 18:58:49,629 INFO    MainThread:3869025 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0
+2025-08-07 18:58:49,630 INFO    MainThread:3869025 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2405] restore
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2411] restore done
+2025-08-07 18:58:52,061 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 18:58:52,076 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 18:58:52,083 INFO    MainThread:3869025 [wandb_run.py:_footer_sync_info():3864] logging synced files

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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": 1754579215036600070, "commit_timestamp_nsecs": 1754579215175892946, "custom_metadata": {}}

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/14065f5f7e608cd567d4b276f5b9cebf

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/aa9f6b4f8103559231db4a9d6e075637

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/c3fa4628eb352e5e16eb72654572e80f

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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": 1754579814437199379, "commit_timestamp_nsecs": 1754579814621894121, "custom_metadata": {}}

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9a4dffa36085bbf7e523a43f926ff88c

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/a6a3ae1d6fc2453df7f3c981b1dd58ce

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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": 1754579848832453239, "commit_timestamp_nsecs": 1754579848965540441, "custom_metadata": {}}

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19897cc70c3e579a4701bc6bedd3cd30

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/f73893b3800ac13b8528776462c2ae26

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

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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": 1754579883058944004, "commit_timestamp_nsecs": 1754579883230950856, "custom_metadata": {}}

wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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/run-20250807_150400-wp_0.0_alpha_1/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}}]}