AlphaZero implemented Chinese chess. AlphaGo Zero / AlphaZero实践项目，实现中国象棋。
Author chengstone
e-Mail [email protected]
代码详解请参见文内jupyter notebook和↓↓↓
知乎专栏：https://zhuanlan.zhihu.com/p/34433581
博客：http://blog.csdn.net/chengcheng1394/article/details/79526474
欢迎转发扩散 ^_^
这是一个AlphaZero的实践项目，实现了一个中国象棋程序，使用TensorFlow1.0和Python 3.5开发，还要安装uvloop。
因为我的模型训练的不充分，只训练了不到4K次，模型刚刚学会用象和士防守，总之仍然下棋很烂。
如果您有条件可以再多训练试试，我自从收到信用卡扣款400美元通知以后就把aws下线了：D 贫穷限制了我的想象力O(∩_∩)O
我训练的模型文件下载地址：https://pan.baidu.com/s/1dLvxFFpeWZK-aZ2Koewrvg
解压后放到项目根目录下即可，文件夹名叫做gpu_models
现在介绍下命令如何使用：
命令分为两类，一类是训练，一类是下棋。
训练专用：
- --mode 指定是训练（train）还是下棋（play），默认是训练
- --train_playout 指定MCTS的模拟次数，论文中是1600，我做训练时使用1200
- --batch_size 指定训练数据达到多少时开始训练，默认512
- --search_threads 指定执行MCTS时的线程个数，默认16
- --processor 指定是使用cpu还是gpu，默认是cpu
- --num_gpus 指定gpu的个数，默认是1
- --res_block_nums 指定残差块的层数，论文中是19或39层，我默认是7
下棋专用：
- --ai_count 指定ai的个数，1是人机对战，2是看两个ai下棋
- --ai_function 指定ai的下棋方法，是思考（mcts，会慢），还是直觉（net，下棋快）
- --play_playout 指定ai进行MCTS的模拟次数
- --delay和--end_delay默认就好，两个ai下棋太快，就不知道俩ai怎么下的了：）
- --human_color 指定人类棋手的颜色，w是先手，b是后手
训练命令举例：
python main.py --mode train --train_playout 1200 --batch_size 512 --search_threads 16 --processor gpu --num_gpus 2 --res_block_nums 7
下棋命令举例：
python main.py --mode play --ai_count 1 --ai_function mcts --play_playout 1200 --human_color w
Licensed under the MIT License with the 996ICU License.