tensorflow模型冻结

最新推荐文章于 2024-05-14 20:38:59 发布
最新推荐文章于 2024-05-14 20:38:59 发布
阅读量1.5k 收藏 2
点赞数
分类专栏： 模型 文章标签： tensorflow
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u013546508/article/details/90404017
版权

查看模型输入输出节点名称

saved_model_cli show --dir ./xxx --all
如：
在这里插入图片描述

冻结模型

freeze_graph --input_saved_model_dir --output_node_names --output_graph
在这里插入图片描述

问题

模型冻结时出现的BUG：
tensorflow.python.framework.errors_impl.InternalError: Dst tensor is not initialized.
[[{ {node save/RestoreV2/_379}} = _Recvclient_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device_incarnation=1, tensor_name=“edge_383_save/RestoreV2”, tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"<

最低0.47元/天 解锁文章
SUN_SU3
关注
专栏目录
Tensorflow 模型的保存、读取和冻结、执行
weixin_30352191的博客
05-01 1048
转载自https://www.jarvis73.cn/2018/04/25/Tensorflow-Model-Save-Read/ 本文假设读者已经懂得了 Tensorflow 的一些基础概念, 如果不懂, 则移步 TF官网. 在 Tensorflow 中我们一般使用tf.train.Saver()定义的存储器对象来保存模型, 并得到形如下面列表的文件: checkpointmo...
tensorflow导出冻结图模型
hzhj的博客
06-03 1070
导出推理图 python export_inference_graph.py \ --alsologtostderr \ --dataset_name=autohome \ #自定义数据集的名称 --model_name=mobilenet_v2_140 \ #模型名字 --image_size=224 \ --output_file=angle/mobilenet_v2_140_224_inference.pb #待输出的推理图 冻结模型文件 python -m tenso
tensoeflow 冻结网络模型
终结大笨狗的博客
04-21 1085
冻结模型文件本文基于mnist与lenet，讲述如下两个问题：1.如何将训练好的网络模型冻结，形成net.pb文件？2.如何将net.pb文件部署到TensorFlow中做inference？pb文件保存的步骤 1.需要给input与最终的预测值取个名字，便于部署时输入数据并输出数据 2.利用graph_util.convert_variables_to_constants将网络中模型参数变量转
Keras, Tensorflow: 冻结某层中的特定变量
PianGe_zyl的博客
04-24 956
创建的自定义层中含有W和m两个可训练变量： self.W = self.add_weight(name='W', shape=(input_shape[0][-1].value, self.n_classes), trainable=True) self.m = self.add_weight(name='m', shape=(1, 1), trainable=True) 现在想将m单独设置为不可训练，即单独冻结变量m： l = model.get_layer('layer_name') # 层的名字
输出tensorflow冻结模型的图操作节点特征
邹九的个人博客
10-07 198
importtensorflowastf model_path='/home/rkb/disk/project/ocr/compressyolo/output_yolo.pb' config=tf.ConfigProto() config.gpu_options.allow_growth=True detection_graph=tf.Graph() withtf.Graph().as_default(): graph_def=tf...
tensorflow设置部分参数可训练，冻结部分参数
mingo220的博客
12-16 5491
需求： 在利用bert进行训练时，需要指定模型中的部分参数不训练 方法： 设置一个可训练的变量list 代码： update_var_list = [] #该list中的变量参与参数更新 tvars = tf.trainable_variables() for tvar in tvars: if "bert" not in tvar.name: update_var_list.append...
tensorflow冻结模型为pb文件的各种方法
nkxiaolei的博客
12-05 4091
笔者最近因为工作需要将TensorFlow训练模型迁移到晟腾芯片平台上离线推理，所以需要将ckpt或者h5模型冻结成pb，再利用ATC模型转换工具将pb转为离线模型om文件，期间遇到一些问题和坑，总结一下，供大家参考。 1.Tensorflow1.x 训练好的模型Ckpt文件： DB_resnet_v1_50_adam_model.ckpt-16801.index DB_resnet_v1_50_adam_model.ckpt-16801.data-00000-of-00001 DB_resnet_v1_5
torch模型转tensorflow模型汇总
qq_40637261的博客
09-29 2168
torch模型转tensorflow模型汇总
TensorFlow2.x冻结模型，保存为.pb格式方便部署
布布汪的博客
06-21 8409
1、用tf.keras创建模型 inputs = tf.keras.Input(shape=(1000, 9,), name='input') input = tf.keras.layers.Flatten(name="flatten")(inputs) f1 = tf.keras.layers.Dense(512, activation='relu', name='dense_1')(input) d1 = tf.keras.layers.Dropout(0.2,name='dropout_1')(f1)
如何将tensorflow训练好的参数冻结成pb文件
一个做图像文本的深度学习人
07-03 1322
利用tensorflow冻结pb文件，虽然只是实现了，但是我还不会用，先记录下来吧，以后慢慢补充。
pytorch冻结模型
重剑无锋博客
03-08 2380
当您想要冻结部分模型时，或者您事先知道不会使用某些参数的梯度。 例如，如果要对预先训练的CNN进行优化，只要切换冻结模型中的requires_grad标志就足够了， 直到计算到最后一层才会保存中间缓冲区，其中的仿射变换（卷级操作等）将使用需要梯度的权重并且网络的输出也将需要它们。 import torchvision import torch.nn as nn from torch import optim model = torchvision.models.resnet18(pretrained
tensorflow运行错误及解决方法--一直更新
乌龟慢慢爬
03-02 1万+
1. FailedPreconditionError: Attempting to use uninitialized value beta1_power [[Node: beta1_power/read = Identity[T=DT_FLOAT, _class=["loc:@Variable"], _device="/job:localhost/replica:0/task:...
TensorFlow-GPU执行模型训练时报错
weixin_45746505的博客
10-13 1826
TensorFlow-GPU训练存储溢出
tensorflow冻结部分层，只训练某一层
keep forward, go, go, go
04-16 2万+
其实fine tune就是这样子的，如何把某些层冻结，然后训练最后一层。那么在tensorflow里如何实现呢？优化的时候，只选择优化特定层的参数即可。如下：-Python 代码1#定义优化算子2optimizer = tf.train.AdamOptimizer(1e-3)3#选择待优化的参数4output_vars = tf.get_collection(tf.GraphKyes.TRAINA...
TensorFlow模型冻结（ckpt转为pb）笔记
qq249356520的博客
07-06 2105
# -*- coding:utf-8 -*- """ This file used to freeze tensorflow .ckpt to .pb """ import tensorflow as tf #两种方式 方法1：函数方法，传入session def freeze_session(session, keep_var_name=None, output_names=None...
TensorFlow-模型的保存和调用（freeze方式）
samylee的博客
12-18 2200
TensorFlow-模型的保存和调用（freeze方式） 硬件：NVIDIA-GTX1080 软件：Windows7、python3.6.5、tensorflow-gpu-1.4.0 一、基础知识 freeze：将ckpt的三个文件融合为一个文件，将variables转换为constant，文件更小，更易于移植 二、代码展示 1、保存模型 import tensorflow as...
【深度学习中的“冻结”含义】
最新发布
fyc300的博客
05-14 1159
在深度学习领域，“冻结”的含义通常指的是在训练过程中保持网络模型中的某一层或多层的权重参数不变。这样做的目的可能是为了保留预训练模型在这些层上学到的特征，或者是因为这些层的参数对于当前任务来说已经足够好，不需要再进行训练。提示：以下是本篇文章正文内容，下面案例可供参考在深度学习中，"冻结"通常指的是在训练过程中保持模型的某些部分或参数不可更新。当我们冻结某些参数时，意味着它们在反向传播过程中不会被更新，即它们的梯度值将保持不变。
解决Tensorflow使用CPU报错[ /job:localhost/replica:0/task:0/device:CPU:0 ]
热门推荐
梦醒时分
11-21 2万+
最近在跟着&lt;21个项目玩转深度学习&gt;中第三章的模型,但是遇见一个问题,我认为应该是CPU和GPU的使用问题. 解决方法是将slim文件夹中的train_image_classifier.py文件中的 tf.app.flags.DEFINE_boolean('clone_on_cpu', Flase, 'Use CPUs to...
TensorFlow模型持久化：ckpt到pb转换详解
pb文件，即GraphDef，是TensorFlow模型的序列化形式，它包含了模型的完整计算图结构和权重。pb文件主要用于模型的发布和离线预测，因为它们提供了静态的图，不需要再次运行训练代码。将ckpt转换为pb文件通常使用`...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值