1. 程式人生 > >tensorflow的模型檔案ckpt引數獲取

tensorflow的模型檔案ckpt引數獲取

已知模型檔案的ckpt檔案,通過pywrap_tensorflow獲取模型的各引數名

import tensoflow as tf
from tensorflow.python import pywrap_tensorflow

model_dir = "./ckpt/"

ckpt = tf.train.get_checkpoint_state(model_dir)
ckpt_path = ckpt.model_checkpoint_path

reader = pywrap_tensorflow.NewCheckpointReader(ckpt_path)
param_dict = reader.get_variable_to_shape_map()

for key, val in param_dict.items():
    try:
        print key, val
    except:
        pass