diff --git a/AlphaGo/game.py b/AlphaGo/game.py index a962f5c..72ae2e0 100644 --- a/AlphaGo/game.py +++ b/AlphaGo/game.py @@ -22,8 +22,8 @@ import time class Game: ''' Load the real game and trained weights. - - TODO : Maybe merge with the engine class in future, + + TODO : Maybe merge with the engine class in future, currently leave it untouched for interacting with Go UI. ''' def __init__(self, name=None, role=None, debug=False, checkpoint_path=None): @@ -46,7 +46,7 @@ class Game: else: raise ValueError(name + " is an unknown game...") - self.evaluator = model.ResNet(self.size, self.size ** 2 + 1, history_length=self.history_length) + self.evaluator = model.ResNet(self.size, self.size ** 2 + 1, history_length=self.history_length, checkpoint_path=checkpoint_path) self.latest_boards = deque(maxlen=self.history_length) for _ in range(self.history_length): self.latest_boards.append(self.board)