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

查看chekpoit文件

时间:2018-08-25 21:18:34      阅读:156      评论:0      收藏:0      [点我收藏+]

标签:poi   col   mod   from   rar   import   port   int   tmp   

使用tf.train.Saver()保存到checkpoint文件,我们可以用tensorflow查看。

# import the inspect_checkpoint library
from tensorflow.python.tools import inspect_checkpoint as chkp

# print all tensors in checkpoint file
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name=‘‘, all_tensors=True)

# tensor_name:  v1
# [ 1.  1.  1.]
# tensor_name:  v2
# [-1. -1. -1. -1. -1.]

# print only tensor v1 in checkpoint file
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name=v1, all_tensors=False)

# tensor_name:  v1
# [ 1.  1.  1.]

# print only tensor v2 in checkpoint file
chkp.print_tensors_in_checkpoint_file("/tmp/model.ckpt", tensor_name=v2, all_tensors=False)

# tensor_name:  v2
# [-1. -1. -1. -1. -1.]

 

查看chekpoit文件

标签:poi   col   mod   from   rar   import   port   int   tmp   

原文地址:https://www.cnblogs.com/superxuezhazha/p/9535279.html

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