1.从checkpoint中获取全部的变量名和变量值
tf.contrib.framework.list_variables(model_dir)
tf.contrib.framework.load_variable(model_dir, var_name)
2.清除 tf.Session
tf.reset_default_graph() 重置计算图
3.
原文地址:https://www.cnblogs.com/wb-learn/p/11596980.html
时间: 2024-10-10 15:26:12