tensorflow小技巧之查看保存模型参数的 name 和 value

tensorflow 同时被 2 个专栏收录
6 篇文章 0 订阅
9 篇文章 0 订阅

tensorflow小技巧之查看保存模型参数的 name 和 value

在将tensorflow训练好的模型保存之后,可能会遇见需要查看其中的权重和偏置的名称以及具体的值的大小,可以使用tensorflow.python 中的 pywrap_tensorflow库进行查看
具体程序如下:

import os
from tensorflow.python import pywrap_tensorflow
num_epoch = 199
model_path = r'checkpoint\Autoencoder_128_Pixel_Grad\Autoencoder.model-' + str(num_epoch)
model_path = os.path.join(os.getcwd(), model_path)
rule_num_epoch = 49
rule_model_path = './checkpoint/UNet_Fusion_Rule_128_MSE/RuleModel.model-' + str(rule_num_epoch)
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(rule_model_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
    print("name: ", key)
    print(reader.get_tensor(key))

在这里插入图片描述
模型保存的形式如上图所示。
查看结果则如下图所示。
在这里插入图片描述
因为值无法完全显示,所以只展示name。

  • 1
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值