#neural-network #cuda #chess #trainer #bullet #nn #back-end

bullet_lib

双人对战游戏神经网络训练器

1 个稳定版本

1.0.0 2024年3月17日

#425机器学习

MIT 许可证

150KB
4.5K SLoC

Rust 4K SLoC // 0.0% comments CUDA 499 SLoC // 0.0% comments

bullet

A CUDA/CPU NN 训练器,用于训练用于 akimbo 的 NNUE 风格网络。

也被许多其他引擎使用,包括

目前支持的游戏

  • 国际象棋
  • Ataxx

提出支持新游戏的问题。

用法

使用以下命令导入包

bullet = { package = "bullet_lib", features = ["cuda"] }

查看 维基示例 了解如何使用此包。

实用工具

您可以使用以下命令构建 bullet-utils

  • 转换数据
  • 交错多个数据文件
  • 打乱数据文件
  • 验证数据文件

使用 ./target/release/bullet-utils[.exe] help 查看具体用法。

目前支持的后端

默认

CPU 后端 不建议用于严重训练。它适用于训练小型网络或各种实用程序,例如加载网络进行重新量化或将输出测试在特定位置上。

CUDA

“第一类”支持的后端。要编译到 CUDA 目标,您需要启用 cuda 功能,如 维基 中所示。

依赖项

~155KB