Potential error

[Fernadoo]

In game.py, line 62-72:

def current_state(self):
    ....
    square_state[0][move_curr // self.width, move_curr % self.height] = 1.0
    ....

I assume you may actually mean [move_curr // self.width, move_curr % self.width].
Not a fatal bug if you are testing with square boards.

[WangChuanYuan]

It should be [move_curr // self.height, move_curr % self.height]? As the shape of square_state is (4, self.width, self.height), so the third dimension should be height?
By the way, why reshape to (height, width) in

AlphaZero_Gomoku/train.py

Line 73 in a2555b2

mcts_porb.reshape(self.board_height, self.board_width)), i)

,
although they are the same.