Upload folder using huggingface_hub

README.md

@@ -0,0 +1,53 @@
+# JAXGMG checkpoints
+
+Goal Misgeneralisation Models trained 
+Trained with jaxgmg, see [jaxgmg](https://github.com/timaeus-research/jaxgmg), branch `david` for more details.
+
+
+## matt-ckpt
+
+Path pattern:
+```
+checkpoints/dr-*/7168
+```
+
+Trained on 15x15 grid
+File name include the alpha parameter e,g, dr-3eneg5 means alpha=3e-5
+Not much known, but useful for debugging.
+
+
+## alpha-blocks-13x13-sweep
+
+Path pattern:
+```
+checkpoints/run-alpha_${alpha}-steps_200M/files/checkpoints/${checkpoint_number}
+```
+
+All models trained with `blocks` environment generation, world size 13x13.
+
+```
+#!/bin/bash
+for alpha in 1e-0 1e-1 1e-2 1e-3 1e-4 3.3e-1 3.3e-2 3.3e-3 3.3e-4; do
+    python -m jaxgmg train corner --num-total-env-steps 200_000_000 --keep-all-checkpoints --num-cycles-per-checkpoint 64 --wandb-project jaxgmg2 --wandb-name alpha:${alpha}-steps:200M-theta:0 --prob-shift ${alpha} --env-size 13 --env-layout blocks
+done
+```
+
+Theta specifies the reward function:
+reward = proxy_goal * theta + true_goal * (1 - theta)
+
+All these models were trained with theta=0, i.e. the true goal of getting the cheese.
+
+The alpha parameter `prob-shift` controls the fraction of distinguishing v.s. undistinguishing environments.
+e.g. alpha=1 means the agetn always sees distinguishing environments (ones were the cheese is not in the corner)
+and alpha=0 means the agent always sees undistinguishing environments (ones were the cheese is always in the corner).
+
+Checkpoints are taken every 64 cycles (~512 env steps per cycle, ~32k env steps per checkpoint?).
+
+E.g. the path `run-alpha_1e1-steps_200M/files/checkpoints/128` corresponds to training with alpha=1e-1, after 128 cycles (the second checkpoint).
+
+
+
+
+
+
+

alpha-blocks-13x13-sweep/README.md

@@ -0,0 +1,30 @@
+Goal Misgeneralisation Models trained 
+
+All models trained with `blocks` environment generation, world size 13x13.
+Trained with jaxgmg, see [jaxgmg](https://github.com/timaeus-research/jaxgmg), branch `david` for more details.
+
+```
+#!/bin/bash
+for alpha in 1e-0 1e-1 1e-2 1e-3 1e-4 3.3e-1 3.3e-2 3.3e-3 3.3e-4; do
+    python -m jaxgmg train corner --num-total-env-steps 200_000_000 --keep-all-checkpoints --num-cycles-per-checkpoint 64 --wandb-project jaxgmg2 --wandb-name alpha:${alpha}-steps:200M-theta:0 --prob-shift ${alpha} --env-size 13 --env-layout blocks
+done
+```
+
+Theta specifies the reward function:
+reward = proxy_goal * theta + true_goal * (1 - theta)
+
+All these models were trained with theta=0, i.e. the true goal of getting the cheese.
+
+The alpha parameter `prob-shift` controls the fraction of distinguishing v.s. undistinguishing environments.
+e.g. alpha=1 means the agetn always sees distinguishing environments (ones were the cheese is not in the corner)
+and alpha=0 means the agent always sees undistinguishing environments (ones were the cheese is always in the corner).
+
+Checkpoints are taken every 64 cycles (~512 env steps per cycle, ~32k env steps per checkpoint?).
+
+E.g. the path `run-alpha_1e1-steps_200M/files/checkpoints/128` corresponds to training with alpha=1e-1, after 128 cycles (the second checkpoint).
+
+
+
+
+
+

alpha-blocks-13x13-sweep/debug.log

@@ -0,0 +1,28 @@
+2025-07-15 19:51:51,233 INFO    MainThread:39680 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_setup.py:_flush():80] Configure stats pid to 39680
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250715_195151-21e19fox/logs/debug.log
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250715_195151-21e19fox/logs/debug-internal.log
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_init.py:init():830] calling init triggers
+2025-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'blocks', '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.00033, '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': False, '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, '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-07-15 19:51:51,234 INFO    MainThread:39680 [wandb_init.py:init():871] starting backend
+2025-07-15 19:51:51,443 INFO    MainThread:39680 [wandb_init.py:init():874] sending inform_init request
+2025-07-15 19:51:51,453 INFO    MainThread:39680 [wandb_init.py:init():882] backend started and connected
+2025-07-15 19:51:51,457 INFO    MainThread:39680 [wandb_init.py:init():953] updated telemetry
+2025-07-15 19:51:51,473 INFO    MainThread:39680 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-07-15 19:51:52,131 INFO    MainThread:39680 [wandb_init.py:init():1029] starting run threads in backend
+2025-07-15 19:51:52,260 INFO    MainThread:39680 [wandb_run.py:_console_start():2458] atexit reg
+2025-07-15 19:51:52,260 INFO    MainThread:39680 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-07-15 19:51:52,260 INFO    MainThread:39680 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-07-15 19:51:52,260 INFO    MainThread:39680 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-07-15 19:51:52,264 INFO    MainThread:39680 [wandb_init.py:init():1075] run started, returning control to user process
+2025-07-15 20:20:43,140 INFO    MainThread:39680 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg2/21e19fox
+2025-07-15 20:20:43,140 INFO    MainThread:39680 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-07-15 20:20:43,141 INFO    MainThread:39680 [wandb_run.py:_restore():2405] restore
+2025-07-15 20:20:43,141 INFO    MainThread:39680 [wandb_run.py:_restore():2411] restore done
+2025-07-15 20:20:45,803 INFO    MainThread:39680 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-07-15 20:20:45,816 INFO    MainThread:39680 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-07-15 20:20:45,821 INFO    MainThread:39680 [wandb_run.py:_footer_sync_info():3864] logging synced files

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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": 1752602249166754025, "commit_timestamp_nsecs": 1752602249958711118, "custom_metadata": {}}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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\"}"}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}}]}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/files/checkpoints/0/default/ocdbt.process_0/d/2dc6347afc158bbef0f3a5a79c8b148c

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

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/files/checkpoints/0/default/ocdbt.process_0/d/53f2a653841e3647d8050830231e58d4

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

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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": 1752602538425884048, "commit_timestamp_nsecs": 1752602538522022592, "custom_metadata": {}}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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\"}"}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}}]}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/files/checkpoints/1024/default/ocdbt.process_0/d/fae7eb941e5ac10ed585d7eb5b89f2e2

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

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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": 1752602555749430941, "commit_timestamp_nsecs": 1752602555847060958, "custom_metadata": {}}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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\"}"}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}}]}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/files/checkpoints/1088/default/ocdbt.process_0/d/dfb2fffa636de518e3202f5eea917de5

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

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/files/checkpoints/1088/default/ocdbt.process_0/d/e94a05cc906ab31a40a4f5c91393f63a

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

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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": 1752602573029555435, "commit_timestamp_nsecs": 1752602573127106956, "custom_metadata": {}}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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\"}"}

alpha-blocks-13x13-sweep/run-alpha_1e-0-steps_200M/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}}]}