From 66589dfde32891e3412c9e2beef36e859e192ea4 Mon Sep 17 00:00:00 2001 From: Alexander Munch-Hansen Date: Thu, 19 Apr 2018 16:01:19 +0200 Subject: [PATCH] fixed global step, now using exp decay --- network.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/network.py b/network.py index 4486790..3449703 100644 --- a/network.py +++ b/network.py @@ -38,10 +38,10 @@ class Network: # Can't remember the best learning_rate, look this up self.max_learning_rate = 0.1 self.min_learning_rate = 0.001 - self.learning_rate = 0.01 + # self.learning_rate = 0.01 self.global_step = tf.Variable(0, trainable=False, name="global_step") - # self.learning_rate = tf.maximum(self.min_learning_rate, tf.train.exponential_decay(self.max_learning_rate, self.global_step, 50000, 0.96, staircase=True), name="learning_rate") + self.learning_rate = tf.maximum(self.min_learning_rate, tf.train.exponential_decay(self.max_learning_rate, self.global_step, 50000, 0.96, staircase=True), name="learning_rate") @@ -164,11 +164,10 @@ class Network: # Restore trained episode count for model episode_count_path = os.path.join(self.checkpoint_path, "episodes_trained") - if os.path.isfile(episode_count_path): +p if os.path.isfile(episode_count_path): with open(episode_count_path, 'r') as f: self.config['start_episode'] = int(f.read()) - else: - assert False + def make_move(self, sess, board, roll, player): # print(Board.pretty(board))