hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pass the checkpoint path to the model

Browse Source
This commit is contained in:
rtz19970824 2017-12-26 13:17:46 +08:00
parent 76f641a0f1
commit 725fc2c04e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
AlphaGo/game.py
View File

@ -46,7 +46,7 @@ class Game:
else: else:
raise ValueError(name + " is an unknown game...") 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) self.latest_boards = deque(maxlen=self.history_length)
for _ in range(self.history_length): for _ in range(self.history_length):
self.latest_boards.append(self.board) self.latest_boards.append(self.board)