怎么训练其它棋盘大小的模型

[Startshark]

如题，可以请问一下各位大佬，如果要训练不同大小的正方形棋盘，我应该怎么修改哪些部分的代码？

[IcyFeather233]

在 train.py 里面改这个部分：

class TrainPipeline():
    def __init__(self, init_model=None):
        # params of the board and the game
        self.board_width = 6
        self.board_height = 6
        self.n_in_row = 4