从训练好的tensorflow模型中打印训练变量

最新推荐文章于 2024-06-06 10:02:18 发布
最新推荐文章于 2024-06-06 10:02:18 发布
阅读量2.7k 收藏 3
点赞数
分类专栏： 学习笔记 tensorflow
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/aiseu001/article/details/79851176
版权

从tensorflow 训练后保存的模型中打印训变量：使用tf.train.NewCheckpointReader()

import tensorflow as tf
reader = tf.train.NewCheckpointReader('path/alexnet/model-330000')
dic = reader.get_variable_to_shape_map()
print dic

打印变量

w = reader.get_tensor("fc1/W")
print type(w)
print w.shape
print w[0]



aiseu001
关注
  • 0
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
tensorflow实现打印ckpt模型保存下的变量名称及变量值
09-18
今天小编就为大家分享一篇tensorflow实现打印ckpt模型保存下的变量名称及变量值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
【tensorflow】打印Tensorflow graph中的所有需要训练的变量--tf.trainable_variables()
Shwan_ma的博客
12-23 2万+
一般来说，打印tensorflow变量的函数有两个： tf.trainable_variables () 和 tf.all_variables() 不同的是： tf.trainable_variables () 指的是需要训练的变量 tf.all_variables() 指的是所有变量一般而言，我们更关注需要训练的训练变量： 值得注意的是，在输出变量名时，要对整个graph进行初始化一、打
使用TensorFlow实现的"迭代剪枝"：高效神经网络优化的新视角
最新发布
gitblog_00037的博客
06-06 284
使用TensorFlow实现的"迭代剪枝"：高效神经网络优化的新视角 项目地址:https://gitcode.com/younghwanoh/impl-pruning-TF 在这个开源项目中，开发者提供了一个基于TensorFlow的实现，名为"迭代剪枝"（Iterative Pruning），旨在探索如何有效地对神经网络进行压缩，以提高效率并减少存储空间...
tensorflow 中的变量为何会被训练
zlq1233217的专栏
01-29 1117
之前一直疑惑，在TensorFlow中构建神经网络的时候仅仅做了几个矩阵乘法，其中W B在run的时候就会迭代训练。经过一番查找资料， 声明Variable变量时，默认trainable=True，会自动训练。 在http://blog.csdn.net/shenxiaoming77/article/details/77197708 中有更详细说明。
pytorch 实现打印模型的参数值
09-18
今天小编就为大家分享一篇pytorch 实现打印模型的参数值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch打印模型参数，冻结训练等操作
yanghao201607030101的博客
12-23 5061
import torch.optim as optim import torch import torchvision.models as models device=torch.device("cuda" if torch.cuda.is_available() else "cpu") model=models.resnet50(pretrained=False).to(device) optimizer=optim.Adam(model.parameters(),0.01) scheduler = op
从训练好的tensorflow模型中打印训练变量实例
09-18
本文将详细讲解如何从训练好的TensorFlow模型中提取并打印这些训练变量。 首先，我们需要导入必要的库，即TensorFlow库，用于操作和读取模型： ```python import tensorflow as tf ``` 当你完成了一次TensorFlow...
打印tensorflow恢复模型中所有变量与操作节点方式
09-16
`tf.trainable_variables()`和`tf.global_variables()`分别返回可训练变量和全局变量的列表。`sess.run(variable_names)`执行运行，获取所有变量的值。`get_operations()`则用于获取计算图中的所有操作节点，并打印...
tensorflow 打印内存中的变量方法
09-20
总的来说，理解和掌握在TensorFlow中打印变量的方法对于调试模型、观察学习过程以及诊断潜在问题至关重要。无论是简单的循环打印还是指定变量名的打印，都是实用的工具，能够帮助我们更好地理解模型的内部运作。
TensorFlow实现打印每一层的输出
09-18
今天小编就为大家分享一篇TensorFlow实现打印每一层的输出，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
erwin模型怎么保存_当我们保存模型的时候我们在保存什么 [TF 笔记 3]
weixin_39723248的博客
11-25 396
前言模型保存是模型训练整个流程不可缺少的一环。在模型训练结束时我们需要保存，从而方便我们进行部署使用，或者未来用来继续训练，例如迁移学习或者增量学习；在训练过程中我们往往也会定期保存，因为机器训练时间过长，难免会出现出现一些意想不到的故障需要重启。有了这些定期保存的模型，我们能够进行断点恢复；还有时我们希望导出仅仅对于部署有用的部分，删掉不必要的部分，例如和优化器 (optimizer) 相关的一...
tensorflow保存加载模型查看训练参数
James_Ying的专栏
04-18 1万+
TensorFlow的checkpoint机制使得其能够同时支持Online Learning和Continuous Learning，首先，通过tf.train.Saver()将训练好的或者训练过程中的模型保存成checkpoint:_, loss_value, step = sess.run([train_op, loss, global_step]) saver.save(sess,"./c
tensorflow 获取模型所有参数总和数量
初学者的博客
04-15 4489
参数数量： np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()]) 浮点运算量： tf.contrib.tfprof.model_analyzer.print_model_analysis(tf.get_default_graph(), tfprof_options=tf.contrib.tfp...
tensorflow打印模型结构_TensorFlow入门：第一个机器学习Demo
weixin_39625586的博客
12-01 2675
张量(Tensor)TensorFlow 内部的计算都是基于张量的，因此我们有必要先对张量有个认识。张量是在我们熟悉的标量、向量之上定义的，详细的定义比较复杂，我们可以先简单的将它理解为一个多维数组：3 # 这个 0 阶张量就是标量，shape=[][1., 2., 3.] ...
TensorFlow打印tensor值
Panda
07-13 2万+
最近一直在用TF做CNN的图像分类，当softmax层得到预测结果后，我希望能够看到预测结果，以便和标签之间进行比较。特此补上，以便自己记忆。 我现在通过softmax层得到变量train_logits，如果我直接执行print(train_logits)时，得到的结果如下（因为我是134类分类，所以结果是（1,134）维）： 这貌似什么都看不出来。 其实tensorflow提供输出...
TensorFlow保存训练模型为pb文件，读取pb恢复模型，C++调用pb模型
热门推荐
08-22 3万+
这个问题我是看了网上好多博客，自己也踩了好多坑才搞定的，看我这一篇文章，可能不能完全解决你们的问题，但是希望对你们有所帮助！ python线上训练模型比较方便，有时候想要在C++环境下调用python训练好的模型，应该怎么做呢？折腾了好几天，终于搞定了！ 一，第一步，训练保存模型 贴自己源代码太复杂，贴个简化版，表明主要意思就行，别人做的东西不可能和你的完全一样，需要在理解别人意思的基础上，...
tensorflow之让变量不可训练
qq_33345917的博客
12-13 3895
import tensorflow as tf import numpy as np with tf.variable_scope(“xyz”) as scop: a=tf.get_variable(“a”,shape=[2],trainable=False) b=tf.get_variable(“b”,shape=[2]) c=tf.get_variable(“c”,shape=[2]) pri...
TensorFlow入门 - 变量(Variables)
Deephome的博客
03-23 1万+
训练模型时，需要使用变量(Variables)保存和更新参数。Variables是包含张量(tensor)的内存缓冲。变量必须要先被初始化(initialize)，而且可以在训练时和训练后保存(save)到磁盘中。之后可以再恢复(restore)保存的变量值来训练和测试模型。 主要参考一下两类： - The tf.Variable class. - The tf.train.Saver c
tensorflow 怎么只训练一部分变量
Do not mess it up
12-25 1978
https://stackoverflow.com/questions/37326002/is-it-possible-to-make-a-trainable-variable-not-trainable/37327561 After looking at the documentation and the code, I was not able to find a way to remove...
Tensorflow实现Vgg19预训练模型加载与应用详解
本文主要介绍了在Tensorflow中如何加载预训练的VGG19模型，以实现诸如风格迁移、目标检测和图像标注等计算机视觉任务。作者假设读者已经具备基本的Tensorflow和深度学习背景，且已安装了相应的GPU（如GTX1050-ti）和...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值