输出TensorFlow 模型的变量名和对应值

最新推荐文章于 2022-03-25 10:57:10 发布

原创最新推荐文章于 2022-03-25 10:57:10 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorlow模型 #变量名和对应值 #tensorflow模型变量 #tensorflow变量值

tensorflow 专栏收录该内容

4 篇文章

订阅专栏

本文介绍如何使用TensorFlow从checkpoint文件加载模型并读取所有参数的名称及形状，通过具体代码示例展示了参数的加载过程。

import tensorflow as tf

# checkpoint为模型保存地址
checkpoint_path = './model'
with tf.Session() as sess:
    # 得到checkpoint文件中所有的参数（名字，形状）元组
    for var_name, _ in tf.contrib.framework.list_variables(checkpoint_path): 
        # 得到上述参数的值
        var = tf.contrib.framework.load_variable(checkpoint_path, var_name) 
        # var_name为变量的name scope
        # var 是该name scope对应的值
        print(var_name,var)