Tensorflow restore 恢复/载入模型.
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Tensorflow restore 恢复/载入模型.相关的知识，希望对你有一定的参考价值。
Tensorflow 恢复/载入模型。可以有两种方法：
1、首先利用tensorflow建立和以前一样的图。再利用restore恢复参数。
.....定义图......
sess=tf.Session()
saver=tf.train.Saver()
sess.run(tf.global_variables_initializer())
saver.restore(sess, weigths_path)
2、或者直接加载以前的图，将saver修改为：
sess=tf.Session()
saver=tf.train.import_meta_graph(graph_path)
sess.run(tf.global_variables_initializer())
saver.restore(sess, weigths_path)
不过第二种方法需要通过tf.get_tensor_by_name和tf.get_operation_by_name来获取以前的tensor和operation
以上是关于Tensorflow restore 恢复/载入模型.的主要内容，如果未能解决你的问题，请参考以下文章