tensorflow 相关

Posted 2020-11-14 wb-learn

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tensorflow 相关相关的知识，希望对你有一定的参考价值。

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() 重置计算图

以上是关于tensorflow 相关的主要内容，如果未能解决你的问题，请参考以下文章