查看TensorFlow checkpoint文件中的变量名和对应值方法
更新时间:2018年06月14日 10:15:55 投稿:jingxian
今天小编就为大家分享一篇查看TensorFlow checkpoint文件中的变量名和对应值方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
实例如下所示:
from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join(model_dir, "model.ckpt") reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_shape_map() for key in var_to_shape_map: print("tensor_name: ", key) print(reader.get_tensor(key)) # Remove this is you want to print only variable names
以上这篇查看TensorFlow checkpoint文件中的变量名和对应值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Python Django模板之模板过滤器与自定义模板过滤器示例
这篇文章主要介绍了Python Django模板之模板过滤器与自定义模板过滤器,结合实例形式分析了Django框架模板过滤器与自定义模板过滤器相关功能、原理、使用方法及相关操作注意事项,需要的朋友可以参考下2019-10-10
最新评论