Upload folder using huggingface_hub

antban2_al_0.6_g_0.98_id_63_seed_980663/action_logprobs.pth.gz

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

antban2_al_0.6_g_0.98_id_63_seed_980663/action_probs.tar.gz

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

antban2_al_0.6_g_0.98_id_63_seed_980663/checkpoints.tar.gz

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

antban2_al_0.6_g_0.98_id_63_seed_980663/config.cfg

@@ -0,0 +1,79 @@
+rl_action=train
+model_type=impala
+lr=5e-05
+discount_rate=0.98
+num_rollout_steps=64
+grad_acc_per_chunk=5
+num_rollout_chunks=1
+cheese_loc=any
+env_layout=open
+alpha=0.6
+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
+no_tqdm=True
+seed=980663
+mask_type=first_episode
+ckpt_dir=jaxgmg2_antennaban
+vis_average_state=False
+trim_episodes=False
+num_total_env_steps=9999974400
+eval_every=1
+eff_horizon=None
+optim=adam
+env_rule=ban:c[2,1], m[1,:]; c[1,2], m[:,1]
+env_rule_mixture=None
+hf_user=davidquarel
+hf_collection=davidquarel/jaxgmg
+use_hf=True
+num_hf_uploads=1
+use_log=True
+log_optimizer_state=True
+resume=None
+resume_id=None
+resume_optim=False
+checkpoint=antban2_al_0.6_g_0.98_id_63_seed_980663
+wandb_project=jaxgmg2_antennaban
+eval_schedule=0:1,250:2,500:5,2000:10
+render_sixel=False
+sixel_idx=60
+live_monitor=False
+run_id=63
+seed_formula=
+deterministic=True
+penalize_time=False
+f_str_ckpt=antban2_al_0.6_g_0.98_id_63_seed_980663
+duplication_factor=-1
+smoke=False
+ntfy=david_jaxgmg
+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=5
+chunk_size=9600
+env_steps_per_microbatch=122880
+ckpt_path=jaxgmg2_antennaban/antban2_al_0.6_g_0.98_id_63_seed_980663
+env_steps_per_loop=614400
+total_loops=16276

antban2_al_0.6_g_0.98_id_63_seed_980663/config.json

@@ -0,0 +1,81 @@
+{
+  "alpha": 0.6,
+  "checkpoint": "antban2_al_0.6_g_0.98_id_63_seed_980663",
+  "cheese_loc": "any",
+  "chunk_size": 9600,
+  "ckpt_dir": "jaxgmg2_antennaban",
+  "ckpt_path": "jaxgmg2_antennaban/antban2_al_0.6_g_0.98_id_63_seed_980663",
+  "compile": true,
+  "deterministic": true,
+  "discount_rate": 0.98,
+  "duplication_factor": -1,
+  "eff_acc_steps": 5,
+  "eff_horizon": null,
+  "env_layout": "open",
+  "env_rule": "ban:c[2,1], m[1,:]; c[1,2], m[:,1]",
+  "env_rule_mixture": null,
+  "env_size": 13,
+  "env_steps_per_loop": 614400,
+  "env_steps_per_microbatch": 122880,
+  "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": "antban2_al_0.6_g_0.98_id_63_seed_980663",
+  "force_re_download": false,
+  "grad_acc_per_chunk": 5,
+  "hf_collection": "davidquarel/jaxgmg",
+  "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_type": "impala",
+  "no_tqdm": true,
+  "ntfy": "david_jaxgmg",
+  "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": 9999974400,
+  "off_distribution_data": false,
+  "on_policy": true,
+  "optim": "adam",
+  "penalize_time": false,
+  "render_sixel": false,
+  "repo_id": "davidquarel/jaxgmg_ckpt_zip",
+  "resume": null,
+  "resume_id": null,
+  "resume_optim": false,
+  "rl_action": "train",
+  "rmsprop_burnin_steps": 20,
+  "run_id": 63,
+  "seed": 980663,
+  "seed_formula": "",
+  "sink": null,
+  "sixel_idx": 60,
+  "smoke": false,
+  "total_loops": 16276,
+  "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_antennaban",
+  "weight_restrictions": null,
+  "weight_restrictions_invert": false
+}

antban2_al_0.6_g_0.98_id_63_seed_980663/latest_train.json

@@ -0,0 +1,12 @@
+{
+  "ts": "2026-03-20T16:06:02.057610Z",
+  "loop": 16275,
+  "env_steps": 9999974399,
+  "loss": 0.01642804592847824,
+  "avg_return": 0.861775815486908,
+  "regret": 0.00034624370164237916,
+  "entropy": 0.0775086060166359,
+  "lr": 5e-05,
+  "discount_rate": 0.98,
+  "num_rollout_steps": 64
+}

antban2_al_0.6_g_0.98_id_63_seed_980663/lvl_rngs.npy

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

antban2_al_0.6_g_0.98_id_63_seed_980663/optimizer_state.tar.gz

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

antban2_al_0.6_g_0.98_id_63_seed_980663/rng_keys.tar.gz

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

antban2_al_0.6_g_0.98_id_63_seed_980663/setup.json

@@ -0,0 +1,96 @@
+{
+  "config": {
+    "rl_action": "train",
+    "model_type": "impala",
+    "lr": 5e-05,
+    "discount_rate": 0.98,
+    "num_rollout_steps": 64,
+    "grad_acc_per_chunk": 5,
+    "num_rollout_chunks": 1,
+    "cheese_loc": "any",
+    "env_layout": "open",
+    "alpha": 0.6,
+    "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,
+    "no_tqdm": true,
+    "seed": 980663,
+    "mask_type": "first_episode",
+    "ckpt_dir": "jaxgmg2_antennaban",
+    "vis_average_state": false,
+    "trim_episodes": false,
+    "num_total_env_steps": 9999974400,
+    "eval_every": 1,
+    "eff_horizon": null,
+    "optim": "adam",
+    "env_rule": "ban:c[2,1], m[1,:]; c[1,2], m[:,1]",
+    "env_rule_mixture": null,
+    "hf_user": "davidquarel",
+    "hf_collection": "davidquarel/jaxgmg",
+    "use_hf": true,
+    "num_hf_uploads": 1,
+    "use_log": true,
+    "log_optimizer_state": true,
+    "resume": null,
+    "resume_id": null,
+    "resume_optim": false,
+    "checkpoint": "antban2_al_0.6_g_0.98_id_63_seed_980663",
+    "wandb_project": "jaxgmg2_antennaban",
+    "eval_schedule": "0:1,250:2,500:5,2000:10",
+    "render_sixel": false,
+    "sixel_idx": 60,
+    "live_monitor": false,
+    "run_id": 63,
+    "seed_formula": "",
+    "deterministic": true,
+    "penalize_time": false,
+    "f_str_ckpt": "antban2_al_0.6_g_0.98_id_63_seed_980663",
+    "duplication_factor": -1,
+    "smoke": false,
+    "ntfy": "david_jaxgmg",
+    "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": 5,
+    "chunk_size": 9600,
+    "env_steps_per_microbatch": 122880,
+    "ckpt_path": "jaxgmg2_antennaban/antban2_al_0.6_g_0.98_id_63_seed_980663",
+    "env_steps_per_loop": 614400,
+    "total_loops": 16276
+  },
+  "setup": {
+    "start_time_utc": "2026-03-20T11:59:00.657828Z",
+    "seed": 980663,
+    "device": "cuda",
+    "python_version": "3.11.11",
+    "torch_version": "2.10.0+cu128",
+    "jax_version": "0.6.2",
+    "hostname": "74c96e82cb27",
+    "platform": "Linux-5.15.0-171-generic-x86_64-with-glibc2.39",
+    "git_sha": null,
+    "git_branch": null,
+    "use_wandb": true
+  }
+}

antban2_al_0.6_g_0.98_id_63_seed_980663/setup.txt

@@ -0,0 +1,8 @@
+Start: 2026-03-20T11:59:00.657828Z
+Seed: 980663
+Device: cuda
+Python: 3.11.11 | Torch: 2.10.0+cu128 | JAX: 0.6.2
+Host: 74c96e82cb27
+Platform: Linux-5.15.0-171-generic-x86_64-with-glibc2.39
+Git: branch=None sha=None
+use_wandb: True

antban2_al_0.6_g_0.98_id_63_seed_980663/state_counts.npy

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