From d151f71ee3149eca4737a010f297222168fe1ca4 Mon Sep 17 00:00:00 2001 From: rtz19970824 Date: Fri, 22 Dec 2017 13:30:48 +0800 Subject: [PATCH] modify the training config --- AlphaGo/model.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/AlphaGo/model.py b/AlphaGo/model.py index 41f3a47..0d885ef 100644 --- a/AlphaGo/model.py +++ b/AlphaGo/model.py @@ -101,8 +101,8 @@ class ResNet(object): self._build_network(residual_block_num, self.checkpoint_path) # training hyper-parameters: - self.window_length = 1000 - self.save_freq = 1000 + self.window_length = 7000 + self.save_freq = 5000 self.training_data = {'states': deque(maxlen=self.window_length), 'probs': 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) def _file_to_training_data(self, file_name): + print(file_name) with open(file_name, 'r') as file: data = cPickle.load(file) history = deque(maxlen=self.history_length) @@ -267,4 +268,4 @@ class ResNet(object): if __name__=="__main__": model = ResNet(board_size=9, action_num=82) - model.train("file", data_path="./data/", batch_size=128, checkpoint_path="./checkpoint/") \ No newline at end of file + model.train("file", data_path="./data/", batch_size=128, checkpoint_path="./checkpoint/")