minor fixed on self play

AlphaGo/network_small.py

@@ -192,6 +192,9 @@ class Network(object):
         # checkpoint_path = "/home/tongzheng/tianshou/AlphaGo/checkpoints/"
         # sess = multi_gpu.create_session()
         # sess.run(tf.global_variables_initializer())
+        if checkpoint_path is None:
+            self.sess.run(tf.global_variables_initializer())
+        else:
             ckpt_file = tf.train.latest_checkpoint(checkpoint_path)
             if ckpt_file is not None:
             # print('Restoring model from {}...'.format(ckpt_file))

AlphaGo/self-play.py

@@ -14,7 +14,7 @@ args = parser.parse_args()
 if not os.path.exists(args.result_path):
     os.makedirs(args.result_path)
 
-game = Game()
+game = Game(checkpoint_path="./checkpoints/")
 engine = GTPEngine(game_obj=game)
 history = deque(maxlen=8)
 for i in range(8):