A Multi-threaded Implementation of AlphaZero
A multi-threaded implementation of AlphaZero
Edit config.py
# Compile Python extension
mkdir build
cd build
cmake .. -DCMAKE_PREFIX_PATH=path/to/libtorch -DPYTHON_EXECUTABLE=path/to/python -DCMAKE_BUILD_TYPE=Release
make -j10
# Run
cd ../test
python learner_test.py train # train model
python learner_test.py play # play with human
Trained 2 days on GTX TITAN X (similar to GTX1070)
See GitHub Release: https://github.com/hijkzzz/alpha-zero-gomoku/releases