Upload folder using huggingface_hub

al_0.47_g_0.975_105/action_logprobs.pth.gz

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

al_0.47_g_0.975_105/action_probs.tar.gz

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

al_0.47_g_0.975_105/checkpoints.tar.gz

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

al_0.47_g_0.975_105/config.cfg

@@ -0,0 +1,57 @@
+rl_action=train
+num_rollout_steps=64
+lr=5e-05
+discount_rate=0.975
+eff_horizon=None
+eval_every=1
+use_wandb=True
+use_hf=True
+use_log=True
+num_total_env_steps=5000000000
+checkpoint=al_0.47_g_0.975_105
+render_sixel=False
+sixel_idx=60
+seed=105
+mask_type=first_episode
+penalize_time=False
+optim=adam
+live_monitor=False
+use_bf16=False
+deterministic=True
+eval_schedule=0:1,250:2,500:5,1000:10,2000:20
+grad_acc_per_chunk=5
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=0.47
+env_size=13
+num_levels=9600
+f_str_ckpt=al_{alpha}_g_{discount_rate}_{seed}
+wandb_project=jaxgmg2_3phase_seed
+ckpt_dir=jaxgmg2_3phase_seed
+duplication_factor=-1
+smoke=False
+compile=True
+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
+weight_restrictions=None
+weight_restrictions_invert=False
+evaluate_every_position=False
+use_prev_action=False
+num_prev_actions=1

al_0.47_g_0.975_105/config.json

@@ -0,0 +1,59 @@
+{
+  "alpha": 0.47,
+  "checkpoint": "al_0.47_g_0.975_105",
+  "cheese_loc": "any",
+  "ckpt_dir": "jaxgmg2_3phase_seed",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.975,
+  "duplication_factor": -1,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_size": 13,
+  "eval_every": 1,
+  "eval_schedule": "0:1,250:2,500:5,1000:10,2000:20",
+  "evaluate_every_position": false,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": "al_{alpha}_g_{discount_rate}_{seed}",
+  "force_re_download": false,
+  "grad_acc_per_chunk": 5,
+  "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,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_levels": 9600,
+  "num_prev_actions": 1,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_steps_bw_draws": 1,
+  "num_total_env_steps": 5000000000,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "seed": 105,
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "use_bf16": false,
+  "use_hf": true,
+  "use_log": true,
+  "use_prev_action": false,
+  "use_shuffled_checkpoints": false,
+  "use_wandb": true,
+  "wandb_project": "jaxgmg2_3phase_seed",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}

al_0.47_g_0.975_105/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2026-01-24T10:34:16.910677Z",
+  "loop": 8137,
+  "env_steps": 4999987199,
+  "loss": 0.002025768393650651,
+  "avg_return": 0.5442076325416565,
+  "regret": 0.7529319524765015,
+  "entropy": 0.04175533726811409,
+  "lr": 5e-05,
+  "discount_rate": 0.975,
+  "num_rollout_steps": 64
+}

al_0.47_g_0.975_105/setup.json

@@ -0,0 +1,74 @@
+{
+  "config": {
+    "rl_action": "train",
+    "num_rollout_steps": 64,
+    "lr": 5e-05,
+    "discount_rate": 0.975,
+    "eff_horizon": null,
+    "eval_every": 1,
+    "use_wandb": true,
+    "use_hf": true,
+    "use_log": true,
+    "num_total_env_steps": 5000000000,
+    "checkpoint": "al_0.47_g_0.975_105",
+    "render_sixel": false,
+    "sixel_idx": 60,
+    "seed": 105,
+    "mask_type": "first_episode",
+    "penalize_time": false,
+    "optim": "adam",
+    "live_monitor": false,
+    "use_bf16": false,
+    "deterministic": true,
+    "eval_schedule": "0:1,250:2,500:5,1000:10,2000:20",
+    "grad_acc_per_chunk": 5,
+    "num_rollout_chunks": 1,
+    "cheese_loc": "any",
+    "env_layout": "open",
+    "alpha": 0.47,
+    "env_size": 13,
+    "num_levels": 9600,
+    "f_str_ckpt": "al_{alpha}_g_{discount_rate}_{seed}",
+    "wandb_project": "jaxgmg2_3phase_seed",
+    "ckpt_dir": "jaxgmg2_3phase_seed",
+    "duplication_factor": -1,
+    "smoke": false,
+    "compile": true,
+    "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,
+    "weight_restrictions": null,
+    "weight_restrictions_invert": false,
+    "evaluate_every_position": false,
+    "use_prev_action": false,
+    "num_prev_actions": 1
+  },
+  "setup": {
+    "start_time_utc": "2026-01-24T08:37:56.174273Z",
+    "seed": 105,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.10.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "5a72857b6734",
+    "platform": "Linux-5.15.0-164-generic-x86_64-with-glibc2.35",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

al_0.47_g_0.975_105/setup.txt

@@ -0,0 +1,8 @@
+Start: 2026-01-24T08:37:56.174273Z
+Seed: 105
+Device: cuda
+Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
+Host: 5a72857b6734
+Platform: Linux-5.15.0-164-generic-x86_64-with-glibc2.35
+Git: branch=None sha=None
+use_wandb: True

al_0.47_g_0.975_seed_105_pa_1/action_logprobs.pth.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:49fc947737a051023851180b0035090ac1d8a1019c20365859910c861d1a6b4a
-size 335754488
+oid sha256:c12bf031269eccf077d2a7fada5a08b538abca881cbe3e1785b49f84d88a747c
+size 66773090

al_0.47_g_0.975_seed_105_pa_1/action_probs.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:52d67640773b3ee3f8df13359536216188aa13cf55aef6bc7f981b1081346df7
-size 25373535
+oid sha256:f73fb2d9efd72751a655ac27712e76c009cb8419a3d22836f021f32d77037a84
+size 26112336

al_0.47_g_0.975_seed_105_pa_1/checkpoints.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4d109d01c379b95235df86967f8817c07f8ce2199af2f231c135612a2c6869a9
-size 648887790
+oid sha256:805fb7c15ee8c42c3e649f0a0ef52af4f0c09565e6c4dd9d8fca04d141fbca27
+size 644916792

al_0.47_g_0.975_seed_105_pa_1/config.cfg

@@ -53,6 +53,6 @@ off_distribution_data=False
 weight_restrictions=None
 weight_restrictions_invert=False
 evaluate_every_position=False
-use_prev_action=True
+use_prev_action=False
 num_prev_actions=1
 ntfy=david_jaxgmg

al_0.47_g_0.975_seed_105_pa_1/config.json

@@ -51,7 +51,7 @@
   "use_bf16": false,
   "use_hf": true,
   "use_log": true,
-  "use_prev_action": true,
+  "use_prev_action": false,
   "use_shuffled_checkpoints": false,
   "use_wandb": true,
   "wandb_project": "jaxgmg2_3phase_seed",

al_0.47_g_0.975_seed_105_pa_1/latest_train.json

@@ -1,11 +1,11 @@
 {
-  "ts": "2026-01-26T21:29:55.074009Z",
+  "ts": "2026-01-25T10:23:43.652843Z",
   "loop": 8137,
   "env_steps": 4999987199,
-  "loss": 0.0031139643397182226,
-  "avg_return": 0.54128497838974,
-  "regret": 0.2840884029865265,
-  "entropy": 0.05126394331455231,
+  "loss": 0.002025768393650651,
+  "avg_return": 0.5442076325416565,
+  "regret": 0.7529319524765015,
+  "entropy": 0.04175533726811409,
   "lr": 5e-05,
   "discount_rate": 0.975,
   "num_rollout_steps": 64

al_0.47_g_0.975_seed_105_pa_1/setup.json

@@ -55,19 +55,19 @@
     "weight_restrictions": null,
     "weight_restrictions_invert": false,
     "evaluate_every_position": false,
-    "use_prev_action": true,
+    "use_prev_action": false,
     "num_prev_actions": 1,
     "ntfy": "david_jaxgmg"
   },
   "setup": {
-    "start_time_utc": "2026-01-26T19:35:14.504074Z",
+    "start_time_utc": "2026-01-25T08:26:35.661617Z",
     "seed": 105,
     "device": "cuda",
     "python_version": "3.11.11",
     "torch_version": "2.10.0+cu128",
     "jax_version": "0.6.2",
-    "hostname": "627114e368a8",
-    "platform": "Linux-6.8.0-90-generic-x86_64-with-glibc2.35",
+    "hostname": "5a72857b6734",
+    "platform": "Linux-5.15.0-164-generic-x86_64-with-glibc2.35",
     "git_sha": null,
     "git_branch": null,
     "use_wandb": true

al_0.47_g_0.975_seed_105_pa_1/setup.txt

@@ -1,8 +1,8 @@
-Start: 2026-01-26T19:35:14.504074Z
+Start: 2026-01-25T08:26:35.661617Z
 Seed: 105
 Device: cuda
 Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
-Host: 627114e368a8
-Platform: Linux-6.8.0-90-generic-x86_64-with-glibc2.35
+Host: 5a72857b6734
+Platform: Linux-5.15.0-164-generic-x86_64-with-glibc2.35
 Git: branch=None sha=None
 use_wandb: True

al_0.47_g_0.975_seed_106_pa_1/action_probs.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:903fb755f885fc61a87012e5c21f7ee89f8531bf64f8b434b401bf9a9a9febbd
-size 27173623
+oid sha256:fa2ae6e5c80257c15769f34429161c88655aad5bf652c4956f3c2566b0e845d5
+size 27173631

al_0.47_g_0.975_seed_106_pa_1/checkpoints.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3bcd419b715476e8cf3fe7f8aa7959c7910d153ddd4bbbec5acf486b27f00b6e
-size 644696638
+oid sha256:14bb57262110aa0cbe3b6349f3608087068083723ab51e60909313663c69c5b4
+size 644696492

al_0.47_g_0.975_seed_106_pa_1/latest_train.json

@@ -1,10 +1,10 @@
 {
-  "ts": "2026-01-26T19:56:59.871677Z",
+  "ts": "2026-01-25T12:21:46.846635Z",
   "loop": 8137,
   "env_steps": 4999987199,
   "loss": 0.00011000235099345446,
   "avg_return": 0.5428342819213867,
-  "regret": 0.2835543751716614,
+  "regret": 0.7053660154342651,
   "entropy": 0.009456370957195759,
   "lr": 5e-05,
   "discount_rate": 0.975,

al_0.47_g_0.975_seed_106_pa_1/setup.json

@@ -60,14 +60,14 @@
     "ntfy": "david_jaxgmg"
   },
   "setup": {
-    "start_time_utc": "2026-01-26T17:59:42.860772Z",
+    "start_time_utc": "2026-01-25T10:24:47.727928Z",
     "seed": 106,
     "device": "cuda",
     "python_version": "3.11.11",
     "torch_version": "2.10.0+cu128",
     "jax_version": "0.6.2",
-    "hostname": "954412276ea4",
-    "platform": "Linux-6.8.0-90-generic-x86_64-with-glibc2.35",
+    "hostname": "5a72857b6734",
+    "platform": "Linux-5.15.0-164-generic-x86_64-with-glibc2.35",
     "git_sha": null,
     "git_branch": null,
     "use_wandb": true

al_0.47_g_0.975_seed_106_pa_1/setup.txt

@@ -1,8 +1,8 @@
-Start: 2026-01-26T17:59:42.860772Z
+Start: 2026-01-25T10:24:47.727928Z
 Seed: 106
 Device: cuda
 Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
-Host: 954412276ea4
-Platform: Linux-6.8.0-90-generic-x86_64-with-glibc2.35
+Host: 5a72857b6734
+Platform: Linux-5.15.0-164-generic-x86_64-with-glibc2.35
 Git: branch=None sha=None
 use_wandb: True

al_0.47_g_0.975_seed_107_pa_1/action_probs.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:76211ade074b8baf3f0503674c383eecc33e666a1ff73ca1cc1517a37d8f8d92
-size 22889738
+oid sha256:21d7aca17d31b2951bf49fa48d5e9a07eaa244fd39faf51b68553560fb1fbf36
+size 22889683

al_0.47_g_0.975_seed_107_pa_1/checkpoints.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7121241482ed6d9ccfc310d7a2a198144063e76d9fa0c7348b13776e4bdfad3e
-size 644704193
+oid sha256:b912ccedc6dcc47c155f694305fc9f042f28fd830d6c7b660d07df3afcd1dc64
+size 644704197

al_0.47_g_0.975_seed_107_pa_1/latest_train.json

@@ -1,10 +1,10 @@
 {
-  "ts": "2026-01-26T20:03:51.407321Z",
+  "ts": "2026-01-25T14:19:46.782534Z",
   "loop": 8137,
-  "env_steps": 4999988410,
+  "env_steps": 4999987199,
   "loss": 0.0006763355922885239,
   "avg_return": 0.5420774221420288,
-  "regret": 0.283868670463562,
+  "regret": 0.706595778465271,
   "entropy": 0.039437636733055115,
   "lr": 5e-05,
   "discount_rate": 0.975,

al_0.47_g_0.975_seed_107_pa_1/setup.json

@@ -60,14 +60,14 @@
     "ntfy": "david_jaxgmg"
   },
   "setup": {
-    "start_time_utc": "2026-01-26T18:01:10.851356Z",
+    "start_time_utc": "2026-01-25T12:22:45.119462Z",
     "seed": 107,
     "device": "cuda",
     "python_version": "3.11.11",
     "torch_version": "2.10.0+cu128",
     "jax_version": "0.6.2",
-    "hostname": "86ec2907f6ae",
-    "platform": "Linux-5.15.0-157-generic-x86_64-with-glibc2.35",
+    "hostname": "5a72857b6734",
+    "platform": "Linux-5.15.0-164-generic-x86_64-with-glibc2.35",
     "git_sha": null,
     "git_branch": null,
     "use_wandb": true

al_0.47_g_0.975_seed_107_pa_1/setup.txt

@@ -1,8 +1,8 @@
-Start: 2026-01-26T18:01:10.851356Z
+Start: 2026-01-25T12:22:45.119462Z
 Seed: 107
 Device: cuda
 Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
-Host: 86ec2907f6ae
-Platform: Linux-5.15.0-157-generic-x86_64-with-glibc2.35
+Host: 5a72857b6734
+Platform: Linux-5.15.0-164-generic-x86_64-with-glibc2.35
 Git: branch=None sha=None
 use_wandb: True

al_0.47_g_0.975_seed_108_pa_1/action_probs.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ae45905c0fdd9b9461efed413b998c445a1c67ecf0b0fc53b0423bee42980215
-size 23538540
+oid sha256:070ace0b70e1cf353488122d379bcd8c529e88297720b20c436716e32664a75e
+size 23538427

al_0.47_g_0.975_seed_108_pa_1/checkpoints.tar.gz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ef353f7154d0bbef5192f8cfdac43a7b0f475ee99596a0b437f331ab086ef0e
-size 644837825
+oid sha256:86aab2294e20e77a430dc954deb5a2334531cc4140c465b969523c669f445391
+size 644838058

al_0.47_g_0.975_seed_108_pa_1/latest_train.json

@@ -1,10 +1,10 @@
 {
-  "ts": "2026-01-26T20:55:09.124574Z",
+  "ts": "2026-01-25T16:19:56.697944Z",
   "loop": 8137,
   "env_steps": 4999987199,
   "loss": 0.001199210062623024,
   "avg_return": 0.5378735661506653,
-  "regret": 0.2837236225605011,
+  "regret": 0.7057788372039795,
   "entropy": 0.031318098306655884,
   "lr": 5e-05,
   "discount_rate": 0.975,

al_0.47_g_0.975_seed_108_pa_1/setup.json

@@ -60,14 +60,14 @@
     "ntfy": "david_jaxgmg"
   },
   "setup": {
-    "start_time_utc": "2026-01-26T18:30:09.252786Z",
+    "start_time_utc": "2026-01-25T14:21:38.612709Z",
     "seed": 108,
     "device": "cuda",
     "python_version": "3.11.11",
     "torch_version": "2.10.0+cu128",
     "jax_version": "0.6.2",
-    "hostname": "bed1bffe896b",
-    "platform": "Linux-6.8.0-58-lowlatency-x86_64-with-glibc2.35",
+    "hostname": "5a72857b6734",
+    "platform": "Linux-5.15.0-164-generic-x86_64-with-glibc2.35",
     "git_sha": null,
     "git_branch": null,
     "use_wandb": true

al_0.47_g_0.975_seed_108_pa_1/setup.txt

@@ -1,8 +1,8 @@
-Start: 2026-01-26T18:30:09.252786Z
+Start: 2026-01-25T14:21:38.612709Z
 Seed: 108
 Device: cuda
 Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
-Host: bed1bffe896b
-Platform: Linux-6.8.0-58-lowlatency-x86_64-with-glibc2.35
+Host: 5a72857b6734
+Platform: Linux-5.15.0-164-generic-x86_64-with-glibc2.35
 Git: branch=None sha=None
 use_wandb: True