请问想增大网络input的维度应该在哪修改呢？

junxiaosong / AlphaZero_Gomoku

An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)

MIT License

3.25k stars 965 forks source link

Closed survivingME closed 5 years ago

survivingME commented 5 years ago

我想使网络的input变成11widthheight即5步双方的棋子和一个表示棋子颜色的形式，这方面的代码在什么位置呢？谢谢！！！

junxiaosong commented 5 years ago

局面描述在game.py中Board类的current_state()方法中返回，可以在那里改。当然要返回5步的情况的话，你需要用一个变量存储最近几步的落子情况，现在的代码里只记录了最近一步，即self.last_move

survivingME commented 5 years ago

非常感谢！！