Skip to content

syys96/libtorch_learn

Repository files navigation

Learning to play nogo from scratch with neural network and MCTS

This project is based on several excellent works:
Requirements
  • Libtorch, CUDA and CUDNN, with matched version. 10.2 recommended.

  • c++ compiler supporting c++14

  • cmake 3.0+

Build
mkdir build && cd build

cmake .. -DCMAKE_BUILD_TYPE=Release

make
How to train a neural net

​ build the runtool target with make and run it.

​ runtool will detect the model file automatically and start loop of selfplay, training and evaluation.

TODO
  • 把submit改成长时运行
  • 去除不必要的验证代码,加速
  • 网络加大一点,block和filter都搞起来,现在网络才1m,能不弱吗