码迷,mamicode.com
首页 > 其他好文 > 详细

Tensorflow restore 恢复/载入模型.

时间:2017-09-07 12:09:34      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:glob   oba   name   global   加载   两种方法   import   利用   参数   

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 恢复/载入模型.

标签:glob   oba   name   global   加载   两种方法   import   利用   参数   

原文地址:http://www.cnblogs.com/Tron/p/7488644.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!