bmdanielsson / nnue-trainer

PyTorch trainer for NNUE style neural networks.
7 stars 1 forks source link

Overview

This repository contains the code used to train NNUE style networks for the Marvin chess engine.

Setup

python3 -m venv env
source env/bin/activate
pip install python-chess torch tensorboard

Building the data loader

This requires the Clang compiler.

make

Basic training steps

python train.py training.bin validation.bin

Logging

tensorboard --logdir=logs

Then, go to http://localhost:6006/

Acknowledgements