Upload folder using huggingface_hub

al_0.47_g_0.99_102_bf16/action_logprobs.pth.gz

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

al_0.47_g_0.99_102_bf16/action_probs.tar.gz

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

al_0.47_g_0.99_102_bf16/checkpoints.tar.gz

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

al_0.47_g_0.99_102_bf16/config.cfg

@@ -0,0 +1,46 @@
+rl_action=train
+num_rollout_steps=64
+lr=5e-05
+discount_rate=0.99
+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.99_102_bf16
+render_sixel=True
+sixel_loc=(7, 7)
+seed=102
+mask_type=first_episode
+penalize_time=False
+optim=adam
+live_monitor=False
+use_bf16=True
+checkpoint_schedule=0:8
+grad_acc_per_chunk=16
+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}_bf16
+wandb_project=jaxgmg_3phase_bf16
+ckpt_dir=jaxgmg_3phase_bf16
+duplication_factor=-1
+smoke=False
+num_chains=6
+num_draws=3000
+on_policy=True
+nbeta=3000
+localization=10
+exact_solver_each_draw=False
+llc_optimizer=sgld
+iw_clip_eps=None
+rmsprop_burnin=20
+llc_data_file=llc_scan_open_reinforce.pkl
+llc_checkpoint_index=0
+repo_id=davidquarel/jaxgmg_ckpt_zip
+use_shuffled_checkpoints=0
+force_re_download=False

al_0.47_g_0.99_102_bf16/config.json

@@ -0,0 +1,51 @@
+{
+  "alpha": 0.47,
+  "checkpoint": "al_0.47_g_0.99_102_bf16",
+  "checkpoint_schedule": "0:8",
+  "cheese_loc": "any",
+  "ckpt_dir": "jaxgmg_3phase_bf16",
+  "discount_rate": 0.99,
+  "duplication_factor": -1,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_size": 13,
+  "eval_every": 1,
+  "exact_solver_each_draw": false,
+  "f_str_ckpt": "al_{alpha}_g_{discount_rate}_{seed}_bf16",
+  "force_re_download": false,
+  "grad_acc_per_chunk": 16,
+  "iw_clip_eps": null,
+  "live_monitor": false,
+  "llc_checkpoint_index": 0,
+  "llc_data_file": "llc_scan_open_reinforce.pkl",
+  "llc_optimizer": "sgld",
+  "localization": 10,
+  "lr": 5e-05,
+  "mask_type": "first_episode",
+  "nbeta": 3000,
+  "num_chains": 6,
+  "num_draws": 3000,
+  "num_levels": 9600,
+  "num_rollout_chunks": 1,
+  "num_rollout_steps": 64,
+  "num_total_env_steps": 5000000000,
+  "on_policy": true,
+  "optim": "adam",
+  "penalize_time": false,
+  "render_sixel": true,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "rl_action": "train",
+  "rmsprop_burnin": 20,
+  "seed": 102,
+  "sixel_loc": [
+    7,
+    7
+  ],
+  "smoke": false,
+  "use_bf16": true,
+  "use_hf": true,
+  "use_log": true,
+  "use_shuffled_checkpoints": 0,
+  "use_wandb": true,
+  "wandb_project": "jaxgmg_3phase_bf16"
+}

al_0.47_g_0.99_102_bf16/latest_eval.json

@@ -0,0 +1,9 @@
+{
+  "ts": "2025-10-20T19:07:25.183421Z",
+  "loop": 8136,
+  "env_steps": 4998758400,
+  "exact_value_avg": 0.9147202372550964,
+  "regret": 0.010819630697369576,
+  "action_probs_image_path": "jaxgmg_3phase_bf16/al_0.47_g_0.99_102_bf16/action_probs/img_8136.png",
+  "entropy": 0.3846891522407532
+}

al_0.47_g_0.99_102_bf16/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2025-10-20T19:07:25.781094Z",
+  "loop": 8137,
+  "env_steps": 4999987199,
+  "loss": 0.002683906117454171,
+  "avg_return": 0.9147175550460815,
+  "regret": 0.010819630697369576,
+  "entropy": 0.27988603711128235,
+  "lr": 5e-05,
+  "discount_rate": 0.99,
+  "num_rollout_steps": 64
+}

al_0.47_g_0.99_102_bf16/setup.json

@@ -0,0 +1,66 @@
+{
+  "config": {
+    "rl_action": "train",
+    "num_rollout_steps": 64,
+    "lr": 5e-05,
+    "discount_rate": 0.99,
+    "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.99_102_bf16",
+    "render_sixel": true,
+    "sixel_loc": [
+      7,
+      7
+    ],
+    "seed": 102,
+    "mask_type": "first_episode",
+    "penalize_time": false,
+    "optim": "adam",
+    "live_monitor": false,
+    "use_bf16": true,
+    "checkpoint_schedule": "0:8",
+    "grad_acc_per_chunk": 16,
+    "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}_bf16",
+    "wandb_project": "jaxgmg_3phase_bf16",
+    "ckpt_dir": "jaxgmg_3phase_bf16",
+    "duplication_factor": -1,
+    "smoke": false,
+    "num_chains": 6,
+    "num_draws": 3000,
+    "on_policy": true,
+    "nbeta": 3000,
+    "localization": 10,
+    "exact_solver_each_draw": false,
+    "llc_optimizer": "sgld",
+    "iw_clip_eps": null,
+    "rmsprop_burnin": 20,
+    "llc_data_file": "llc_scan_open_reinforce.pkl",
+    "llc_checkpoint_index": 0,
+    "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+    "use_shuffled_checkpoints": 0,
+    "force_re_download": false
+  },
+  "setup": {
+    "start_time_utc": "2025-10-20T17:47:36.496213Z",
+    "seed": 102,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.9.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "0a4b26e7f7a9",
+    "platform": "Linux-5.4.0-216-generic-x86_64-with-glibc2.35",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

al_0.47_g_0.99_102_bf16/setup.txt

@@ -0,0 +1,8 @@
+Start: 2025-10-20T17:47:36.496213Z
+Seed: 102
+Device: cuda
+Python: 3.11.11 | Torch: 2.9.0+cu128 | JAX: 0.6.2
+Host: 0a4b26e7f7a9
+Platform: Linux-5.4.0-216-generic-x86_64-with-glibc2.35
+Git: branch=None sha=None
+use_wandb: True