hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

no restrict on saving checkpoints

Browse Source
This commit is contained in:
rtz19970824 2017-12-22 13:05:01 +08:00
parent 2b1285143c
commit d281ecc6e0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
AlphaGo/model.py
View File

@ -133,7 +133,7 @@ class ResNet(object):
with tf.control_dependencies(self.update_ops): with tf.control_dependencies(self.update_ops):
self.train_op = tf.train.AdamOptimizer(1e-4).minimize(self.total_loss) self.train_op = tf.train.AdamOptimizer(1e-4).minimize(self.total_loss)
self.var_list = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES) 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 = multi_gpu.create_session()
self.sess.run(tf.global_variables_initializer()) self.sess.run(tf.global_variables_initializer())
if checkpoint_path is not None: if checkpoint_path is not None: