From 8f9fb7a7a923ea747087ca4f250c02e931cc2554 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=94=90=E7=92=9E?= <1039183305@qq.com> Date: Fri, 30 Dec 2022 02:09:28 +0000 Subject: [PATCH] update TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 唐璞 <1039183305@qq.com> --- TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py b/TensorFlow/contrib/cv/Glow_ID2085_for_TensorFlow/model.py index 8db9e01b4..56e45f933 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) -- Gitee