From 725fc2c04eb7b98350684519dabbd7fdd48b32ea Mon Sep 17 00:00:00 2001 From: rtz19970824 Date: Tue, 26 Dec 2017 13:17:46 +0800 Subject: [PATCH] pass the checkpoint path to the model --- AlphaGo/game.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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)