从TensorFlow模型中提取具体参数值

最新推荐文章于 2022-12-29 15:43:33 发布
最新推荐文章于 2022-12-29 15:43:33 发布
阅读量2.5k 收藏 2
点赞数
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/NeverGx/article/details/88919543
版权

代码
import tensorflow as tf
from tensorflow.python import pywrap_tensorflow

model_dir = “/home/model”

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()
print(param_dict)
print(reader.get_tensor(‘batch_normalization/gamma’).shape)

NeverGx
关注
  • 0
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
使用Tensorflow框架完美保存并实现调用训练好的模型（是模型不是参数哦、全网首篇）
Rainmaple‘s Blog
05-26 3万+
众所周知，Tensorflow在机器学习、深度学习领域有着广泛的应用。博主最近就Tensorflow进行了简单的学习，当我完成简单的回归模型以及CNN的训练后，一个想法油然而生，怎样将训练好的模型保存呢？保存之后又该怎么调用呢？ 于是，我查阅了很多相关资料，经过很多的测试，终于…成功了！现将如何保存和调用训练好的模型总结如下（提一下，本文以CNN为例了，数据集是众所周知的MNIST）： 1...
TensorFlow模型保存和提取的方法
09-20
以下是对TensorFlow模型保存和提取的详细解释： 首先，TensorFlow提供了`tf.train.Saver`类来实现模型的保存和恢复。在保存模型时，我们创建一个`Saver`对象，并调用其`save`方法，如`saver.save(sess,"Model/model...
keras.callbacks.ModelCheckpoint()
baixue0729的博客
07-17 5277
每个纪元后保存模型。 filepath可以包含命名格式化选项，它将填充日志中的纪元和键的值（在on_epoch_end中传递）。 例如：如果filepath是权重。{epoch：02d} - {val_loss：.2f} .hdf5，那么模型检查点将与文件名中的纪元号和验证丢失一起保存。 def __init__(self, filepath: Any, ...
【TensorFlow】从训练好的模型中获取参数值
热门推荐
zwwhsxq的博客
05-31 1万+
checkpoint_file = tf.train.latest_checkpoint(FLAGS.checkpoint_dir)graph = tf.Graph()with graph.as_default():    session_conf = tf.ConfigProto(      allow_soft_placement=FLAGS.allow_soft_placement,    ...
获得tensorflow模型的参数量
zseqsc_asd的博客
03-15 2086
tensorflow模型训练好之后，通常会保存为.ckpt文件，有时我们想了解一下模型保存了多少参数，这固然可以手动计算，但是速度太慢，这里我写了一个程序可以直接获得模型的参数量，下面是源代码： 转自： https://blog.csdn.net/chnguoshiwushuang/article/details/81588794 from tensorflow.python import pyw...
tensorflow1.X读取checkpoint(加载训练好的模型及参数)的两种方式
weixin_43220532的博客
03-25 2863
在模型训练的保存路径下面会保存有多个模型（如下图），下面介绍两种读取方式 1，获取最新的保存的一个模型，比如上图的model-5200 # model_path是checkpoints的路径，下面方法会自动寻找最新保存的一个模型 ckpt_file = tf.train.latest_checkpoint(model_path) saver.restore(sess,ckpt_file) 2，获取某个迭代次数的模型，比如要指定获取上图的model-5199 ckpt_file=os.path.join(
提取tensorflow中checkpoint模型参数到npy
01-03
tensorflow将训练好的网络模型保存到了checkpoint中，如果要利用训练好的网络模型去fintune另一个网络，需要为前置层的参数赋值，因此把checkpoint中的参数保存到npy文件中方便赋值。
浅谈tensorflow中张量的提取值和赋值
09-18
- `tf.gather(params, indices, validate_indices=None, name=None, axis=0)`：这个函数允许你根据提供的索引`indices`从`params`张量中提取值。`axis`参数指定了沿着哪个轴进行索引。如果`indices`是标量，它将直接...
从训练好的tensorflow模型中打印训练变量实例
09-18
本文将详细讲解如何从训练好的TensorFlow模型中提取并打印这些训练变量。 首先，我们需要导入必要的库，即TensorFlow库，用于操作和读取模型： ```python import tensorflow as tf ``` 当你完成了一次TensorFlow...
【Pytorch】查看模型某一层的参数数值（自用）
weixin_43799388的博客
04-13 7929
参考自：pytorch获取模型某一层参数名及参数值方式 import os import torch import torch.nn as nn # 设置GPU os.environ['CUDA_VISIBLE_DEVICES'] = '1' device = torch.device('cuda:0') if torch.cuda.is_available() else 'cpu' # 创建模型 model = nn.Sequential(nn.Conv2d(3, 16, kernel_size=1)
tensorflow 获取模型所有参数总和数量的方法
09-20
今天小编就为大家分享一篇tensorflow 获取模型所有参数总和数量的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
tensorflow2.x 索引与切片
lllfor的博客
03-14 80
举例说明 索引 数据中存有5张28x28的照片的三通道分析数据 data:[5, 28, 28, 3] #使用随机分布进行模拟图片数据 data = tf.random.normal([5, 28, 28, 3]) #索引至第5张照片的所有信息 data1 = data[4] 切片 省略号(…) 数据中有多个维度时，只需要在其中个别维度中进行获取数据，即可用省略号代表其他维度不填写 #以上述照片的数据为例 data2 = data [3,...,2]#取第一维度中的下标为3且第四维度中下标为2的
python3 timedelta的参数及其使用
yutu75的博客
11-10 3134
timedelta 只能计算如下参数的 timedelta参数如下: days=0 天数, seconds=0 秒数, microseconds=0 微秒;百万分之一秒, milliseconds=0 毫秒;千分之一秒, minutes=0 分钟数, hours=0 小时数, weeks=0 周数; 参数为1表示七天， from datetime import datetime, timedelta now = datetime.now() # 获取当
TensorFlow-断点续训
LMengi000的博客
05-13 399
import tensorflow as tf import os mnist = tf.keras.datasets.mnist (x_train,y_train),(x_test,y_test) = mnist.load_data() x_trian,x_test = x_train / 255.0,x_test / 255.0 model = tf.keras.models.Sequential([ tf.keras.layers.Flatten(), tf.keras....
tensorflow 查看模型中的参数列表及对应的值
beginerToBetter的博客
01-04 3697
tensorflow的checkpoint文件存储模型使用的变量对象，其是一个二进制文件，将变量名映射到对用的tensor值，因此可通过该文件查看参数对应的值 #使用NewCheckpointReader来读取ckpt里的变量 from tensorflow.python import pywrap_tensorflow # 以下XXX为模型保存chechpoint文件的相对路径，如"model/bestParMTCR_gateBehavior.ckpt" checkpoint_path = "XXX".
tensorflow2.0中model.fit 函数参数详解
最新发布
xiaofeixia002X的博客
12-29 1796
【代码】tensorflow2.0中model.fit 函数参数详解。
07-TensorFlow 模型参数提取
鸣鼓ming的博客
04-18 990
之前介绍了神经网络可以生成模型文件进行存储, 其实模型中的参数是可以查看的, 详情请看代码 import tensorflow as tf from tensorflow.keras.layers import Dense from tensorflow.keras import Model from sklearn import datasets import numpy as np x_train = datasets.load_iris().data y_train = datasets.load_
tensorflow获取和设置每层的参数的方法
华少的博客
03-17 5520
tensorflow获取和设置每层的参数的方法，看到一篇文章，转载一下，防止忘记！！！ 方法一：通过get_layer()函数先获取要获取权重对应的层；接着通过get_weights() 函数获取该层对应的全部参数（是一个长的为2的列表，列表中的每一个元素是一个numpy数组，第一存放该层的权重W的参数，第一个存放偏置b） layer的两个函数： get_weights(), set_weights(weights) model = load_model('vgg.h5') layer1 = m.
【tensorflow2caffe】二 、tensorflow模型介绍以及参数提取
zl3090的博客
04-05 880
在博客开始前，首先郑重感谢博主jiongnima的博客对我的帮助，我按照博主的方法顺利的把bcnn模型的tensorflow版转为caffe版，关于bcnn模型，它是个细粒度分类模型，我将在后续的博客中详细介绍。 首先隆重呈上博主的链接： https://blog.csdn.net/jiongnima/article/details/78326515 转模型系列博文的操作主要基于博主...
TensorFlow计算模型与深度学习目标识别详解
"本资料详细介绍了TensorFlow计算模型和深度学习在目标识别中的应用，特别提到了LSTM网络的计算过程。文档涵盖了图计算模型、损失函数优化、梯度下降法、反向传播以及LSTM在序列到序列任务中的应用，并涉及到CNN的...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值