tensorflow 查看模型中的参数列表及对应的值
tensorflow的checkpoint文件存储模型使用的变量对象,其是一个二进制文件,将变量名映射到对用的tensor值,因此可通过该文件查看参数对应的值#使用NewCheckpointReader来读取ckpt里的变量from tensorflow.python import pywrap_tensorflow# 以下XXX为模型保存chechpoint文件的相对路径,如"model/bes
·
- tensorflow的checkpoint文件存储模型使用的变量对象,其是一个二进制文件,将变量名映射到对用的tensor值,因此可通过该文件查看参数对应的值
#使用NewCheckpointReader来读取ckpt里的变量
from tensorflow.python import pywrap_tensorflow
# 以下XXX为模型保存chechpoint文件的相对路径,如"model/bestParMTCR_gateBehavior.ckpt"
checkpoint_path = "XXX"
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
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)) # 参数值
- 以上版本仅支持tf1, 在tf2环境下可以使用以下方式进行查看:
import os
import tensorflow as tf
import tensorflow.compat.v1 as tfv1
from tensorflow.python import pywrap_tensorflow
path = "ckpt file path"
reader = tfv1.train.NewCheckpointReader(path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
print("tensor_name: ", key) # 参数名
print(reader.get_tensor(key)) # 参数值
更多推荐
所有评论(0)