Upload folder using huggingface_hub

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/action_logprobs.pth.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/action_probs.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/checkpoints.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/config.cfg

@@ -0,0 +1,84 @@
+rl_action=train
+model_type=impala
+lr=5e-05
+discount_rate=0.98
+num_rollout_steps=64
+grad_acc_per_chunk=4
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=1.0
+env_size=13
+num_levels=9600
+compile=True
+use_prev_action=False
+weight_restrictions=None
+weight_restrictions_invert=False
+use_bf16=False
+use_wandb=True
+seed=42
+mask_type=first_episode
+ckpt_dir=jaxgmg2_3phase_optim_state_patt
+vis_average_state=False
+trim_episodes=False
+num_total_env_steps=7372800000
+eval_every=1
+eff_horizon=None
+optim=adam
+env_rule=None
+hf_user=davidquarel
+hf_collection=None
+use_hf=True
+num_hf_uploads=1
+use_log=True
+log_optimizer_state=True
+resume=jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+resume_id=1330
+resume_optim=True
+checkpoint=al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000
+wandb_project=jaxgmg2_patt
+eval_schedule=0:1,250:2,500:5,2000:10
+render_sixel=False
+sixel_idx=60
+live_monitor=False
+run_id=0
+seed_formula=None
+deterministic=True
+penalize_time=False
+f_str_ckpt=None
+duplication_factor=-1
+smoke=False
+ntfy=None
+num_chains=6
+num_draws=3000
+num_steps_bw_draws=1
+on_policy=True
+llc_nbeta=3000
+localization=10
+exact_solver_each_draw=False
+llc_optimizer=sgld
+iw_clip_eps=None
+rmsprop_burnin_steps=20
+llc_data_file=llc_scan_open_reinforce.pkl
+llc_checkpoint_index=None
+llc_checkpoint_number=None
+sink=None
+repo_id=davidquarel/jaxgmg_ckpt_zip
+use_shuffled_checkpoints=False
+force_re_download=False
+off_distribution_data=False
+evaluate_every_position=False
+num_prev_actions=1
+eff_acc_steps=4
+chunk_size=9600
+env_steps_per_microbatch=153600
+ckpt_path=jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000
+env_steps_per_loop=614400
+total_loops=12000
+patt_mode=mp-inv
+patt_h=0.0
+patt_use_hmax=False
+patt_t=0
+patt_cluster=east
+suscept_id=2000
+model_id=al_1.0_g_0.98_id_19_seed_981019

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/config.json

@@ -0,0 +1,86 @@
+{
+  "alpha": 1.0,
+  "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000",
+  "cheese_loc": "any",
+  "chunk_size": 9600,
+  "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+  "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.98,
+  "duplication_factor": -1,
+  "eff_acc_steps": 4,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_rule": null,
+  "env_size": 13,
+  "env_steps_per_loop": 614400,
+  "env_steps_per_microbatch": 153600,
+  "eval_every": 1,
+  "eval_schedule": "0:1,250:2,500:5,2000:10",
+  "evaluate_every_position": false,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": null,
+  "force_re_download": false,
+  "grad_acc_per_chunk": 4,
+  "hf_collection": null,
+  "hf_user": "davidquarel",
+  "iw_clip_eps": null,
+  "live_monitor": false,
+  "llc_checkpoint_index": null,
+  "llc_checkpoint_number": null,
+  "llc_data_file": "llc_scan_open_reinforce.pkl",
+  "llc_nbeta": 3000,
+  "llc_optimizer": "sgld",
+  "localization": 10,
+  "log_optimizer_state": true,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "model_id": "al_1.0_g_0.98_id_19_seed_981019",
+  "model_type": "impala",
+  "ntfy": null,
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_hf_uploads": 1,
+  "num_levels": 9600,
+  "num_prev_actions": 1,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_steps_bw_draws": 1,
+  "num_total_env_steps": 7372800000,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "patt_cluster": "east",
+  "patt_h": 0.0,
+  "patt_mode": "mp-inv",
+  "patt_t": 0,
+  "patt_use_hmax": false,
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+  "resume_id": 1330,
+  "resume_optim": true,
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "run_id": 0,
+  "seed": 42,
+  "seed_formula": null,
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "suscept_id": 2000,
+  "total_loops": 12000,
+  "trim_episodes": false,
+  "use_bf16": false,
+  "use_hf": true,
+  "use_log": true,
+  "use_prev_action": false,
+  "use_shuffled_checkpoints": false,
+  "use_wandb": true,
+  "vis_average_state": false,
+  "wandb_project": "jaxgmg2_patt",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2026-03-05T19:56:01.816596Z",
+  "loop": 11999,
+  "env_steps": 7372799999,
+  "loss": 0.027530066668987274,
+  "avg_return": 0.8819411396980286,
+  "regret": 0.00022316505783237517,
+  "entropy": 0.10536854714155197,
+  "lr": 5e-05,
+  "discount_rate": 0.98,
+  "num_rollout_steps": 64
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/optimizer_state.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/patterning.log

@@ -0,0 +1,15 @@
+============================================================
+Patterning config summary
+============================================================
+  patt_mode:      mp-inv
+  patt_h:         0.0
+  patt_cluster:   east
+  alpha:          1.0
+  model_id:     al_1.0_g_0.98_id_19_seed_981019
+  resume:       jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+  resume_id:    1330
+  resume_optim: True
+  suscept_id:   2000
+  ckpt_path:    jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000
+  smoke:        False
+============================================================

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/setup.json

@@ -0,0 +1,101 @@
+{
+  "config": {
+    "rl_action": "train",
+    "model_type": "impala",
+    "lr": 5e-05,
+    "discount_rate": 0.98,
+    "num_rollout_steps": 64,
+    "grad_acc_per_chunk": 4,
+    "num_rollout_chunks": 1,
+    "cheese_loc": "any",
+    "env_layout": "open",
+    "alpha": 1.0,
+    "env_size": 13,
+    "num_levels": 9600,
+    "compile": true,
+    "use_prev_action": false,
+    "weight_restrictions": null,
+    "weight_restrictions_invert": false,
+    "use_bf16": false,
+    "use_wandb": true,
+    "seed": 42,
+    "mask_type": "first_episode",
+    "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+    "vis_average_state": false,
+    "trim_episodes": false,
+    "num_total_env_steps": 7372800000,
+    "eval_every": 1,
+    "eff_horizon": null,
+    "optim": "adam",
+    "env_rule": null,
+    "hf_user": "davidquarel",
+    "hf_collection": null,
+    "use_hf": true,
+    "num_hf_uploads": 1,
+    "use_log": true,
+    "log_optimizer_state": true,
+    "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+    "resume_id": 1330,
+    "resume_optim": true,
+    "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000",
+    "wandb_project": "jaxgmg2_patt",
+    "eval_schedule": "0:1,250:2,500:5,2000:10",
+    "render_sixel": false,
+    "sixel_idx": 60,
+    "live_monitor": false,
+    "run_id": 0,
+    "seed_formula": null,
+    "deterministic": true,
+    "penalize_time": false,
+    "f_str_ckpt": null,
+    "duplication_factor": -1,
+    "smoke": false,
+    "ntfy": null,
+    "num_chains": 6,
+    "num_draws": 3000,
+    "num_steps_bw_draws": 1,
+    "on_policy": true,
+    "llc_nbeta": 3000,
+    "localization": 10,
+    "exact_solver_each_draw": false,
+    "llc_optimizer": "sgld",
+    "iw_clip_eps": null,
+    "rmsprop_burnin_steps": 20,
+    "llc_data_file": "llc_scan_open_reinforce.pkl",
+    "llc_checkpoint_index": null,
+    "llc_checkpoint_number": null,
+    "sink": null,
+    "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+    "use_shuffled_checkpoints": false,
+    "force_re_download": false,
+    "off_distribution_data": false,
+    "evaluate_every_position": false,
+    "num_prev_actions": 1,
+    "eff_acc_steps": 4,
+    "chunk_size": 9600,
+    "env_steps_per_microbatch": 153600,
+    "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000",
+    "env_steps_per_loop": 614400,
+    "total_loops": 12000,
+    "patt_mode": "mp-inv",
+    "patt_h": 0.0,
+    "patt_use_hmax": false,
+    "patt_t": 0,
+    "patt_cluster": "east",
+    "suscept_id": 2000,
+    "model_id": "al_1.0_g_0.98_id_19_seed_981019"
+  },
+  "setup": {
+    "start_time_utc": "2026-03-05T17:08:14.446346Z",
+    "seed": 42,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.10.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "e5babfa66034",
+    "platform": "Linux-6.8.0-60-generic-x86_64-with-glibc2.39",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/setup.txt

@@ -0,0 +1,8 @@
+Start: 2026-03-05T17:08:14.446346Z
+Seed: 42
+Device: cuda
+Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
+Host: e5babfa66034
+Platform: Linux-6.8.0-60-generic-x86_64-with-glibc2.39
+Git: branch=None sha=None
+use_wandb: True

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_1330_sus_2000_seed_42/state_counts.npy

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/action_logprobs.pth.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/action_probs.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/checkpoints.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/config.cfg

@@ -0,0 +1,84 @@
+rl_action=train
+model_type=impala
+lr=5e-05
+discount_rate=0.98
+num_rollout_steps=64
+grad_acc_per_chunk=4
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=1.0
+env_size=13
+num_levels=9600
+compile=True
+use_prev_action=False
+weight_restrictions=None
+weight_restrictions_invert=False
+use_bf16=False
+use_wandb=True
+seed=42
+mask_type=first_episode
+ckpt_dir=jaxgmg2_3phase_optim_state_patt
+vis_average_state=False
+trim_episodes=False
+num_total_env_steps=7372800000
+eval_every=1
+eff_horizon=None
+optim=adam
+env_rule=None
+hf_user=davidquarel
+hf_collection=None
+use_hf=True
+num_hf_uploads=1
+use_log=True
+log_optimizer_state=True
+resume=jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+resume_id=175
+resume_optim=True
+checkpoint=al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000
+wandb_project=jaxgmg2_patt
+eval_schedule=0:1,250:2,500:5,2000:10
+render_sixel=False
+sixel_idx=60
+live_monitor=False
+run_id=0
+seed_formula=None
+deterministic=True
+penalize_time=False
+f_str_ckpt=None
+duplication_factor=-1
+smoke=False
+ntfy=None
+num_chains=6
+num_draws=3000
+num_steps_bw_draws=1
+on_policy=True
+llc_nbeta=3000
+localization=10
+exact_solver_each_draw=False
+llc_optimizer=sgld
+iw_clip_eps=None
+rmsprop_burnin_steps=20
+llc_data_file=llc_scan_open_reinforce.pkl
+llc_checkpoint_index=None
+llc_checkpoint_number=None
+sink=None
+repo_id=davidquarel/jaxgmg_ckpt_zip
+use_shuffled_checkpoints=False
+force_re_download=False
+off_distribution_data=False
+evaluate_every_position=False
+num_prev_actions=1
+eff_acc_steps=4
+chunk_size=9600
+env_steps_per_microbatch=153600
+ckpt_path=jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000
+env_steps_per_loop=614400
+total_loops=12000
+patt_mode=mp-inv
+patt_h=0.0
+patt_use_hmax=False
+patt_t=0
+patt_cluster=east
+suscept_id=2000
+model_id=al_1.0_g_0.98_id_19_seed_981019

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/config.json

@@ -0,0 +1,86 @@
+{
+  "alpha": 1.0,
+  "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000",
+  "cheese_loc": "any",
+  "chunk_size": 9600,
+  "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+  "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.98,
+  "duplication_factor": -1,
+  "eff_acc_steps": 4,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_rule": null,
+  "env_size": 13,
+  "env_steps_per_loop": 614400,
+  "env_steps_per_microbatch": 153600,
+  "eval_every": 1,
+  "eval_schedule": "0:1,250:2,500:5,2000:10",
+  "evaluate_every_position": false,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": null,
+  "force_re_download": false,
+  "grad_acc_per_chunk": 4,
+  "hf_collection": null,
+  "hf_user": "davidquarel",
+  "iw_clip_eps": null,
+  "live_monitor": false,
+  "llc_checkpoint_index": null,
+  "llc_checkpoint_number": null,
+  "llc_data_file": "llc_scan_open_reinforce.pkl",
+  "llc_nbeta": 3000,
+  "llc_optimizer": "sgld",
+  "localization": 10,
+  "log_optimizer_state": true,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "model_id": "al_1.0_g_0.98_id_19_seed_981019",
+  "model_type": "impala",
+  "ntfy": null,
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_hf_uploads": 1,
+  "num_levels": 9600,
+  "num_prev_actions": 1,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_steps_bw_draws": 1,
+  "num_total_env_steps": 7372800000,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "patt_cluster": "east",
+  "patt_h": 0.0,
+  "patt_mode": "mp-inv",
+  "patt_t": 0,
+  "patt_use_hmax": false,
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+  "resume_id": 175,
+  "resume_optim": true,
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "run_id": 0,
+  "seed": 42,
+  "seed_formula": null,
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "suscept_id": 2000,
+  "total_loops": 12000,
+  "trim_episodes": false,
+  "use_bf16": false,
+  "use_hf": true,
+  "use_log": true,
+  "use_prev_action": false,
+  "use_shuffled_checkpoints": false,
+  "use_wandb": true,
+  "vis_average_state": false,
+  "wandb_project": "jaxgmg2_patt",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2026-03-05T19:55:05.283675Z",
+  "loop": 11999,
+  "env_steps": 7372799999,
+  "loss": 0.026208939030766487,
+  "avg_return": 0.8814659118652344,
+  "regret": 0.00043080400791950524,
+  "entropy": 0.09881289303302765,
+  "lr": 5e-05,
+  "discount_rate": 0.98,
+  "num_rollout_steps": 64
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/optimizer_state.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/patterning.log

@@ -0,0 +1,15 @@
+============================================================
+Patterning config summary
+============================================================
+  patt_mode:      mp-inv
+  patt_h:         0.0
+  patt_cluster:   east
+  alpha:          1.0
+  model_id:     al_1.0_g_0.98_id_19_seed_981019
+  resume:       jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+  resume_id:    175
+  resume_optim: True
+  suscept_id:   2000
+  ckpt_path:    jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000
+  smoke:        False
+============================================================

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/setup.json

@@ -0,0 +1,101 @@
+{
+  "config": {
+    "rl_action": "train",
+    "model_type": "impala",
+    "lr": 5e-05,
+    "discount_rate": 0.98,
+    "num_rollout_steps": 64,
+    "grad_acc_per_chunk": 4,
+    "num_rollout_chunks": 1,
+    "cheese_loc": "any",
+    "env_layout": "open",
+    "alpha": 1.0,
+    "env_size": 13,
+    "num_levels": 9600,
+    "compile": true,
+    "use_prev_action": false,
+    "weight_restrictions": null,
+    "weight_restrictions_invert": false,
+    "use_bf16": false,
+    "use_wandb": true,
+    "seed": 42,
+    "mask_type": "first_episode",
+    "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+    "vis_average_state": false,
+    "trim_episodes": false,
+    "num_total_env_steps": 7372800000,
+    "eval_every": 1,
+    "eff_horizon": null,
+    "optim": "adam",
+    "env_rule": null,
+    "hf_user": "davidquarel",
+    "hf_collection": null,
+    "use_hf": true,
+    "num_hf_uploads": 1,
+    "use_log": true,
+    "log_optimizer_state": true,
+    "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+    "resume_id": 175,
+    "resume_optim": true,
+    "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000",
+    "wandb_project": "jaxgmg2_patt",
+    "eval_schedule": "0:1,250:2,500:5,2000:10",
+    "render_sixel": false,
+    "sixel_idx": 60,
+    "live_monitor": false,
+    "run_id": 0,
+    "seed_formula": null,
+    "deterministic": true,
+    "penalize_time": false,
+    "f_str_ckpt": null,
+    "duplication_factor": -1,
+    "smoke": false,
+    "ntfy": null,
+    "num_chains": 6,
+    "num_draws": 3000,
+    "num_steps_bw_draws": 1,
+    "on_policy": true,
+    "llc_nbeta": 3000,
+    "localization": 10,
+    "exact_solver_each_draw": false,
+    "llc_optimizer": "sgld",
+    "iw_clip_eps": null,
+    "rmsprop_burnin_steps": 20,
+    "llc_data_file": "llc_scan_open_reinforce.pkl",
+    "llc_checkpoint_index": null,
+    "llc_checkpoint_number": null,
+    "sink": null,
+    "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+    "use_shuffled_checkpoints": false,
+    "force_re_download": false,
+    "off_distribution_data": false,
+    "evaluate_every_position": false,
+    "num_prev_actions": 1,
+    "eff_acc_steps": 4,
+    "chunk_size": 9600,
+    "env_steps_per_microbatch": 153600,
+    "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000",
+    "env_steps_per_loop": 614400,
+    "total_loops": 12000,
+    "patt_mode": "mp-inv",
+    "patt_h": 0.0,
+    "patt_use_hmax": false,
+    "patt_t": 0,
+    "patt_cluster": "east",
+    "suscept_id": 2000,
+    "model_id": "al_1.0_g_0.98_id_19_seed_981019"
+  },
+  "setup": {
+    "start_time_utc": "2026-03-05T17:07:16.119445Z",
+    "seed": 42,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.10.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "8bab973d7ae6",
+    "platform": "Linux-6.8.0-58-lowlatency-x86_64-with-glibc2.39",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/setup.txt

@@ -0,0 +1,8 @@
+Start: 2026-03-05T17:07:16.119445Z
+Seed: 42
+Device: cuda
+Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
+Host: 8bab973d7ae6
+Platform: Linux-6.8.0-58-lowlatency-x86_64-with-glibc2.39
+Git: branch=None sha=None
+use_wandb: True

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_175_sus_2000_seed_42/state_counts.npy

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/action_logprobs.pth.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/action_probs.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/checkpoints.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/config.cfg

@@ -0,0 +1,84 @@
+rl_action=train
+model_type=impala
+lr=5e-05
+discount_rate=0.98
+num_rollout_steps=64
+grad_acc_per_chunk=4
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=1.0
+env_size=13
+num_levels=9600
+compile=True
+use_prev_action=False
+weight_restrictions=None
+weight_restrictions_invert=False
+use_bf16=False
+use_wandb=True
+seed=42
+mask_type=first_episode
+ckpt_dir=jaxgmg2_3phase_optim_state_patt
+vis_average_state=False
+trim_episodes=False
+num_total_env_steps=7372800000
+eval_every=1
+eff_horizon=None
+optim=adam
+env_rule=None
+hf_user=davidquarel
+hf_collection=None
+use_hf=True
+num_hf_uploads=1
+use_log=True
+log_optimizer_state=True
+resume=jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+resume_id=264
+resume_optim=True
+checkpoint=al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000
+wandb_project=jaxgmg2_patt
+eval_schedule=0:1,250:2,500:5,2000:10
+render_sixel=False
+sixel_idx=60
+live_monitor=False
+run_id=0
+seed_formula=None
+deterministic=True
+penalize_time=False
+f_str_ckpt=None
+duplication_factor=-1
+smoke=False
+ntfy=None
+num_chains=6
+num_draws=3000
+num_steps_bw_draws=1
+on_policy=True
+llc_nbeta=3000
+localization=10
+exact_solver_each_draw=False
+llc_optimizer=sgld
+iw_clip_eps=None
+rmsprop_burnin_steps=20
+llc_data_file=llc_scan_open_reinforce.pkl
+llc_checkpoint_index=None
+llc_checkpoint_number=None
+sink=None
+repo_id=davidquarel/jaxgmg_ckpt_zip
+use_shuffled_checkpoints=False
+force_re_download=False
+off_distribution_data=False
+evaluate_every_position=False
+num_prev_actions=1
+eff_acc_steps=4
+chunk_size=9600
+env_steps_per_microbatch=153600
+ckpt_path=jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000
+env_steps_per_loop=614400
+total_loops=12000
+patt_mode=mp-inv
+patt_h=0.0
+patt_use_hmax=False
+patt_t=0
+patt_cluster=east
+suscept_id=2000
+model_id=al_1.0_g_0.98_id_19_seed_981019

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/config.json

@@ -0,0 +1,86 @@
+{
+  "alpha": 1.0,
+  "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000",
+  "cheese_loc": "any",
+  "chunk_size": 9600,
+  "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+  "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.98,
+  "duplication_factor": -1,
+  "eff_acc_steps": 4,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_rule": null,
+  "env_size": 13,
+  "env_steps_per_loop": 614400,
+  "env_steps_per_microbatch": 153600,
+  "eval_every": 1,
+  "eval_schedule": "0:1,250:2,500:5,2000:10",
+  "evaluate_every_position": false,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": null,
+  "force_re_download": false,
+  "grad_acc_per_chunk": 4,
+  "hf_collection": null,
+  "hf_user": "davidquarel",
+  "iw_clip_eps": null,
+  "live_monitor": false,
+  "llc_checkpoint_index": null,
+  "llc_checkpoint_number": null,
+  "llc_data_file": "llc_scan_open_reinforce.pkl",
+  "llc_nbeta": 3000,
+  "llc_optimizer": "sgld",
+  "localization": 10,
+  "log_optimizer_state": true,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "model_id": "al_1.0_g_0.98_id_19_seed_981019",
+  "model_type": "impala",
+  "ntfy": null,
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_hf_uploads": 1,
+  "num_levels": 9600,
+  "num_prev_actions": 1,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_steps_bw_draws": 1,
+  "num_total_env_steps": 7372800000,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "patt_cluster": "east",
+  "patt_h": 0.0,
+  "patt_mode": "mp-inv",
+  "patt_t": 0,
+  "patt_use_hmax": false,
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+  "resume_id": 264,
+  "resume_optim": true,
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "run_id": 0,
+  "seed": 42,
+  "seed_formula": null,
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "suscept_id": 2000,
+  "total_loops": 12000,
+  "trim_episodes": false,
+  "use_bf16": false,
+  "use_hf": true,
+  "use_log": true,
+  "use_prev_action": false,
+  "use_shuffled_checkpoints": false,
+  "use_wandb": true,
+  "vis_average_state": false,
+  "wandb_project": "jaxgmg2_patt",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2026-03-05T19:58:40.844448Z",
+  "loop": 11999,
+  "env_steps": 7372799999,
+  "loss": 0.026006359606981277,
+  "avg_return": 0.8824062347412109,
+  "regret": 0.0001295174442930147,
+  "entropy": 0.0964682549238205,
+  "lr": 5e-05,
+  "discount_rate": 0.98,
+  "num_rollout_steps": 64
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/optimizer_state.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/patterning.log

@@ -0,0 +1,15 @@
+============================================================
+Patterning config summary
+============================================================
+  patt_mode:      mp-inv
+  patt_h:         0.0
+  patt_cluster:   east
+  alpha:          1.0
+  model_id:     al_1.0_g_0.98_id_19_seed_981019
+  resume:       jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+  resume_id:    264
+  resume_optim: True
+  suscept_id:   2000
+  ckpt_path:    jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000
+  smoke:        False
+============================================================

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/setup.json

@@ -0,0 +1,101 @@
+{
+  "config": {
+    "rl_action": "train",
+    "model_type": "impala",
+    "lr": 5e-05,
+    "discount_rate": 0.98,
+    "num_rollout_steps": 64,
+    "grad_acc_per_chunk": 4,
+    "num_rollout_chunks": 1,
+    "cheese_loc": "any",
+    "env_layout": "open",
+    "alpha": 1.0,
+    "env_size": 13,
+    "num_levels": 9600,
+    "compile": true,
+    "use_prev_action": false,
+    "weight_restrictions": null,
+    "weight_restrictions_invert": false,
+    "use_bf16": false,
+    "use_wandb": true,
+    "seed": 42,
+    "mask_type": "first_episode",
+    "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+    "vis_average_state": false,
+    "trim_episodes": false,
+    "num_total_env_steps": 7372800000,
+    "eval_every": 1,
+    "eff_horizon": null,
+    "optim": "adam",
+    "env_rule": null,
+    "hf_user": "davidquarel",
+    "hf_collection": null,
+    "use_hf": true,
+    "num_hf_uploads": 1,
+    "use_log": true,
+    "log_optimizer_state": true,
+    "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+    "resume_id": 264,
+    "resume_optim": true,
+    "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000",
+    "wandb_project": "jaxgmg2_patt",
+    "eval_schedule": "0:1,250:2,500:5,2000:10",
+    "render_sixel": false,
+    "sixel_idx": 60,
+    "live_monitor": false,
+    "run_id": 0,
+    "seed_formula": null,
+    "deterministic": true,
+    "penalize_time": false,
+    "f_str_ckpt": null,
+    "duplication_factor": -1,
+    "smoke": false,
+    "ntfy": null,
+    "num_chains": 6,
+    "num_draws": 3000,
+    "num_steps_bw_draws": 1,
+    "on_policy": true,
+    "llc_nbeta": 3000,
+    "localization": 10,
+    "exact_solver_each_draw": false,
+    "llc_optimizer": "sgld",
+    "iw_clip_eps": null,
+    "rmsprop_burnin_steps": 20,
+    "llc_data_file": "llc_scan_open_reinforce.pkl",
+    "llc_checkpoint_index": null,
+    "llc_checkpoint_number": null,
+    "sink": null,
+    "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+    "use_shuffled_checkpoints": false,
+    "force_re_download": false,
+    "off_distribution_data": false,
+    "evaluate_every_position": false,
+    "num_prev_actions": 1,
+    "eff_acc_steps": 4,
+    "chunk_size": 9600,
+    "env_steps_per_microbatch": 153600,
+    "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000",
+    "env_steps_per_loop": 614400,
+    "total_loops": 12000,
+    "patt_mode": "mp-inv",
+    "patt_h": 0.0,
+    "patt_use_hmax": false,
+    "patt_t": 0,
+    "patt_cluster": "east",
+    "suscept_id": 2000,
+    "model_id": "al_1.0_g_0.98_id_19_seed_981019"
+  },
+  "setup": {
+    "start_time_utc": "2026-03-05T17:07:47.193146Z",
+    "seed": 42,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.10.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "7ebec1ae3866",
+    "platform": "Linux-5.15.0-52-generic-x86_64-with-glibc2.39",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/setup.txt

@@ -0,0 +1,8 @@
+Start: 2026-03-05T17:07:47.193146Z
+Seed: 42
+Device: cuda
+Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
+Host: 7ebec1ae3866
+Platform: Linux-5.15.0-52-generic-x86_64-with-glibc2.39
+Git: branch=None sha=None
+use_wandb: True

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.00_ld-opt_1_ckpt_264_sus_2000_seed_42/state_counts.npy

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42/action_logprobs.pth.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42/action_probs.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42/checkpoints.tar.gz

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

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42/config.cfg

@@ -0,0 +1,83 @@
+rl_action=train
+model_type=impala
+lr=5e-05
+discount_rate=0.98
+num_rollout_steps=64
+grad_acc_per_chunk=4
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=1.0
+env_size=13
+num_levels=9600
+compile=True
+use_prev_action=False
+weight_restrictions=None
+weight_restrictions_invert=False
+use_bf16=False
+use_wandb=True
+seed=42
+mask_type=first_episode
+ckpt_dir=jaxgmg2_3phase_optim_state_patt
+vis_average_state=False
+trim_episodes=False
+num_total_env_steps=7372800000
+eval_every=1
+eff_horizon=None
+optim=adam
+env_rule=None
+hf_user=davidquarel
+hf_collection=None
+use_hf=True
+num_hf_uploads=1
+use_log=True
+log_optimizer_state=True
+resume=jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019
+resume_id=175
+resume_optim=True
+checkpoint=al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42
+wandb_project=jaxgmg2_patt
+eval_schedule=0:1,250:2,500:5,2000:10
+render_sixel=False
+sixel_idx=60
+live_monitor=False
+run_id=0
+seed_formula=None
+deterministic=True
+penalize_time=False
+f_str_ckpt=None
+duplication_factor=-1
+smoke=False
+ntfy=None
+num_chains=6
+num_draws=3000
+num_steps_bw_draws=1
+on_policy=True
+llc_nbeta=3000
+localization=10
+exact_solver_each_draw=False
+llc_optimizer=sgld
+iw_clip_eps=None
+rmsprop_burnin_steps=20
+llc_data_file=llc_scan_open_reinforce.pkl
+llc_checkpoint_index=None
+llc_checkpoint_number=None
+sink=None
+repo_id=davidquarel/jaxgmg_ckpt_zip
+use_shuffled_checkpoints=False
+force_re_download=False
+off_distribution_data=False
+evaluate_every_position=False
+num_prev_actions=1
+eff_acc_steps=4
+chunk_size=9600
+env_steps_per_microbatch=153600
+ckpt_path=jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42
+env_steps_per_loop=614400
+total_loops=12000
+patt_mode=mp-inv
+patt_h=0.0
+patt_use_hmax=False
+patt_t=0.05
+suscept_id=2000
+model_id=al_1.0_g_0.98_id_19_seed_981019

al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42/config.json

@@ -0,0 +1,85 @@
+{
+  "alpha": 1.0,
+  "checkpoint": "al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42",
+  "cheese_loc": "any",
+  "chunk_size": 9600,
+  "ckpt_dir": "jaxgmg2_3phase_optim_state_patt",
+  "ckpt_path": "jaxgmg2_3phase_optim_state_patt/al_1.0_g_0.98_id_19_seed_981019/patt_mp-inv_t_0.05_ld-opt_1_ckpt_175_sus_2000_seed_42",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.98,
+  "duplication_factor": -1,
+  "eff_acc_steps": 4,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_rule": null,
+  "env_size": 13,
+  "env_steps_per_loop": 614400,
+  "env_steps_per_microbatch": 153600,
+  "eval_every": 1,
+  "eval_schedule": "0:1,250:2,500:5,2000:10",
+  "evaluate_every_position": false,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": null,
+  "force_re_download": false,
+  "grad_acc_per_chunk": 4,
+  "hf_collection": null,
+  "hf_user": "davidquarel",
+  "iw_clip_eps": null,
+  "live_monitor": false,
+  "llc_checkpoint_index": null,
+  "llc_checkpoint_number": null,
+  "llc_data_file": "llc_scan_open_reinforce.pkl",
+  "llc_nbeta": 3000,
+  "llc_optimizer": "sgld",
+  "localization": 10,
+  "log_optimizer_state": true,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "model_id": "al_1.0_g_0.98_id_19_seed_981019",
+  "model_type": "impala",
+  "ntfy": null,
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_hf_uploads": 1,
+  "num_levels": 9600,
+  "num_prev_actions": 1,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_steps_bw_draws": 1,
+  "num_total_env_steps": 7372800000,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "patt_h": 0.0,
+  "patt_mode": "mp-inv",
+  "patt_t": 0.05,
+  "patt_use_hmax": false,
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "resume": "jaxgmg2_3phase_optim_state/al_1.0_g_0.98_id_19_seed_981019",
+  "resume_id": 175,
+  "resume_optim": true,
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "run_id": 0,
+  "seed": 42,
+  "seed_formula": null,
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "suscept_id": 2000,
+  "total_loops": 12000,
+  "trim_episodes": false,
+  "use_bf16": false,
+  "use_hf": true,
+  "use_log": true,
+  "use_prev_action": false,
+  "use_shuffled_checkpoints": false,
+  "use_wandb": true,
+  "vis_average_state": false,
+  "wandb_project": "jaxgmg2_patt",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}