1 个稳定版本
1.0.0 | 2024年3月17日 |
---|
#425 在 机器学习 中
150KB
4.5K SLoC
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