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

Merge branch 'master' of https://github.com/sproblvem/tianshou

Browse Source
This commit is contained in:
rtz19970824 2017-12-22 13:47:38 +08:00
commit e72fd52913
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
AlphaGo/model.py
View File

@ -101,8 +101,8 @@ class ResNet(object):
self._build_network(residual_block_num, self.checkpoint_path) self._build_network(residual_block_num, self.checkpoint_path)
# training hyper-parameters: # training hyper-parameters:
self.window_length = 1000 self.window_length = 7000
self.save_freq = 1000 self.save_freq = 5000
self.training_data = {'states': deque(maxlen=self.window_length), 'probs': deque(maxlen=self.window_length), self.training_data = {'states': deque(maxlen=self.window_length), 'probs': deque(maxlen=self.window_length),
'winner': deque(maxlen=self.window_length)} 'winner': deque(maxlen=self.window_length)}
@ -277,4 +277,4 @@ class ResNet(object):
if __name__=="__main__": if __name__=="__main__":
model = ResNet(board_size=9, action_num=82) model = ResNet(board_size=9, action_num=82)
model.train("file", data_path="./data/", batch_size=128, checkpoint_path="./checkpoint/") model.train("file", data_path="./data/", batch_size=128, checkpoint_path="./checkpoint/")