Aneesh19's picture
Upload base planner + gating checkpoints
353da6d verified
|
Raw
History Blame Contribute Delete
1.68 kB

Checkpoint manifest

One base planner + one gating policy checkpoint per environment, about 200 MB total, stored with git lfs. A clone (with git lfs installed) populates this directory as follows.

Committed-action (Jumanji)

Path Phase Notes
committed_action/pacman/base/training_state_best.pkl AlphaZero PacManKT, action delay K=1
committed_action/pacman/gating/gating_state_best.pkl PPO gating sims = {32,64,96,128}
committed_action/tetris_rt/base/training_state_best.pkl AlphaZero TetrisRTKT, K=1
committed_action/tetris_rt/gating/gating_state_best.pkl PPO gating sims = {32,64,96,128}
committed_action/snake/base/training_state_best.pkl AlphaZero SnakeKT, K=3
committed_action/snake/gating/gating_state_best.pkl PPO gating sims = {32,64,96,128}

Clock (pgx)

Path Phase Notes
clock/hex/base/base_planner.ckpt AlphaZero 11x11, nsim=32
clock/hex/gating/gate.pkl PPO gating (GRU) options = {2,8,32,128}
clock/hex/gating_timeout/gate.pkl PPO gating (GRU, strict-timeout) options = {2,8,32,128}; pairs with clock/hex/base; backs Appendix H
clock/go/base/base_planner.ckpt AlphaZero 9x9, nsim=16
clock/go/gating/gate.pkl PPO gating (GRU) options = {16,32,64,96}

Notes

  • The launcher scripts default to these exact paths. Override via AZ_CKPT, GATING_CKPT, CKPT_ROOT, GATE_ROOT environment variables.
  • Other seeds and intermediate epoch checkpoints used in the paper's variance and ablation studies are not included to keep the bundle small; they can be regenerated from the included training scripts.