tensorflow查看checkpoint变量值

import tensorflow as tf

from tensorflow.python import pywrap_tensorflow

#使用tensorflow自带的pywrap_tensorflow读取模型

model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt")

#变换成类似于dict形式的数据

var_dict = model_reader.get_variable_to_shape_map()

#最后，循环打印输出

for key in var_dict:

    print("variable name: ", key)

    print(model_reader.get_tensor(key))
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容