diff --git a/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py b/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py index 8db9e01b4f211bc880e096a5fea88b8d486184e3..56e45f933b59337d901376095c7e9156c0017107 100644 --- a/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py +++ b/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py @@ -89,7 +89,7 @@ def abstract_model_xy(sess, hps, feeds, train_iterator, test_iterator, data_init saver = tf.train.Saver() saver_ema = tf.train.Saver(ema.variables_to_restore()) m.save_ema = lambda path: saver_ema.save( - sess, path, write_meta_graph=False) + sess, path, max_to_keep=5,write_meta_graph=False) m.save = lambda path: saver.save(sess, path, write_meta_graph=False) m.restore = lambda path: saver.restore(sess, path)