本文主要是介绍tensorflow的模型文件ckpt参数获取，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
已知模型文件的ckpt文件，通过pywrap_tensorflow获取模型的各参数名
import tensoflow as tf
from tensorflow.python import pywrap_tensorflowmodel_dir = "./ckpt/"ckpt = tf.train.get_checkpoint_state(model_dir)
ckpt_path = ckpt.model_checkpoint_pathreader = pywrap_tensorflow.NewCheckpointReader(ckpt_path)
param_dict = reader.get_variable_to_shape_map()for key, val in param_dict.items():try:print key, valexcept:pass
这篇关于tensorflow的模型文件ckpt参数获取的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
