HRM-Sudoku

HRMACT Sudoku Solver

This repository hosts checkpoints for the Hierarchical Reasoning Model with Adaptive Computation Time (HRMACT), trained on Sudoku puzzles.

• Training steps: 3,000
• Batch size: 512
• Total training time: ~18 hours
• Checkpoint format: .safetensors

📌 Note: The original HRM paper recommends training for 10,000+ steps for best performance. These checkpoints are intended as a lightweight, educational reference.

---

🔗 Code & Usage

For the full implementation, training pipeline, and evaluation tools, see the main repo: 👉 ZoneTwelve/HRM-Sudoku

To run evaluation on a checkpoint:

python evaluate.py <checkpoint>

---

🔍 Evaluation

Evaluation is performed with evaluate.py.

Each checkpoint is evaluated 100 times per difficulty to estimate average performance.

Example format for results:

---

📖 Citation

If you use this model in academic work, please cite:

@misc{wang2025hierarchicalreasoningmodel,
      title={Hierarchical Reasoning Model},
      author={Guan Wang and Jin Li and Yuhao Sun and Xing Chen and Changling Liu and Yue Wu and Meng Lu and Sen Song and Yasin Abbasi Yadkori},
      year={2025},
      eprint={2506.21734},
      archivePrefix={arXiv},
      primaryClass={cs.AI},
      url={https://arxiv.org/abs/2506.21734},
}

---

📜 License

Apache License 2.0