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

modify the training config

Browse Source
This commit is contained in:
rtz19970824 2017-12-22 13:30:48 +08:00
parent 5c29dad263
commit 6b3efd7fca
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
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)}
@ -241,6 +241,7 @@ class ResNet(object):
self.saver.save(self.sess, self.checkpoint_path + save_path) self.saver.save(self.sess, self.checkpoint_path + save_path)
def _file_to_training_data(self, file_name): def _file_to_training_data(self, file_name):
print(file_name)
with open(file_name, 'r') as file: with open(file_name, 'r') as file:
data = cPickle.load(file) data = cPickle.load(file)
history = deque(maxlen=self.history_length) history = deque(maxlen=self.history_length)