diff --git a/AlphaGo/model.py b/AlphaGo/model.py index c4338c8..764ba5f 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)} @@ -277,4 +277,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/")