official-stockfish/nnue-pytorch

解决的问题

该项目提供了一个基于 PyTorch 的框架，用于训练 Stockfish 国际象棋引擎中使用的 NNUE（高效可更新神经网络）。它允许开发者训练高性能的评估网络，并将其集成到引擎中，以提高其国际象棋对弈水平。

工作原理

该项目使用 PyTorch 实现训练流程。它支持多种硬件加速，包括 NVIDIA GPU（CUDA）、AMD GPU（ROCm）和 Apple Silicon（MPS），以及仅 CPU 的测试。它利用快速稀疏数据加载器来处理训练这些网络所需的大型数据集。

适用人群

对为 Stockfish 训练自定义评估网络感兴趣的国际象棋引擎开发者和研究人员。

亮点

• 支持多种 GPU 后端（NVIDIA、AMD、Apple Silicon）。
• 基于 Docker 的设置，简化环境配置和 C++ 编译过程。
• 集成快速稀疏数据加载器，实现高效训练。
• 支持可复现的训练和引擎测试工作流程。