merge

AlphaGo/model.py

@@ -133,7 +133,7 @@ class ResNet(object):
         with tf.control_dependencies(self.update_ops):
             self.train_op = tf.train.AdamOptimizer(1e-4).minimize(self.total_loss)
         self.var_list = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES)
-        self.saver = tf.train.Saver(var_list=self.var_list)
+        self.saver = tf.train.Saver(max_to_keep=0, var_list=self.var_list)
         self.sess = multi_gpu.create_session()
         self.sess.run(tf.global_variables_initializer())
         if checkpoint_path is not None:
@@ -203,6 +203,7 @@ class ResNet(object):
         iters = 0
         while True:
             new_file_list = list(set(os.listdir(data_path)).difference(all_file_list))
+            if new_file_list:
                 all_file_list = os.listdir(data_path)
             new_file_list.sort(
                 key=lambda file: os.path.getmtime(data_path + file) if not os.path.isdir(data_path + file) else 0)
@@ -241,9 +242,16 @@ 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:
+        read = False
+        with open(file_name, 'rb') as file:
+            while not read:
+                try:
+                    file.seek(0)
                     data = cPickle.load(file)
+                    read = True
+                except Exception as e:
+                    print(e)
+                    time.sleep(1)
         history = deque(maxlen=self.history_length)
         states = []
         probs = []

AlphaGo/play.py

@@ -50,6 +50,12 @@ if __name__ == '__main__':
     # print "Start Name Sever : " + str(start_new_server.pid)  # + str(start_new_server.wait())
     # time.sleep(1)
 
+    # start a name server if no name server exists
+    if len(os.popen('ps aux | grep pyro4-ns | grep -v grep').readlines()) == 0:
+        start_new_server = subprocess.Popen(['pyro4-ns', '&'])
+        print "Start Name Sever : " + str(start_new_server.pid)  # + str(start_new_server.wait())
+        time.sleep(1)
+
     # start two different player with different network weights.
     black_role_name = 'black' + str(args.id)
     white_role_name = 'white' + str(args.id)
@@ -137,7 +143,7 @@ if __name__ == '__main__':
                 file_list.sort(key=lambda file: os.path.getmtime(args.result_path + file) if not os.path.isdir(
                     args.result_path + file) else 0)
                 data_num = eval(file_list[-1][:-4]) + 1
-            with open("./data/" + str(data_num) + ".pkl", "w") as file:
+            with open("./data/" + str(data_num) + ".pkl", "wb") as file:
                 picklestring = cPickle.dump(data, file)
             data.reset()
             game_num += 1