tensorflow 使用VGG16模型VGG_16.tfmodel 报错 GraphDef cannot be larger than 2GB.

最新推荐文章于 2022-06-05 11:06:22 发布

原创最新推荐文章于 2022-06-05 11:06:22 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

debug 专栏收录该内容

3 篇文章

订阅专栏

本文探讨了在使用TensorFlow时遇到的变量管理问题，特别是在连续运行多个模型时，默认图中的变量未被正确清理导致的问题。文章提供了在运行新模型前清除默认图中残留变量的方法，即使用tf.reset_default_graph()函数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://hehuihui1994.github.io/hehuihui1994.github.io/2017/07/19/TensorFlow常见问题/

爆掉应该是因为所有的变量都加载到了默认图里。

可能因为我跑完第一个model之后，再跑第二个model，我以为会覆盖的，但是第一个model中的变量还在图中…

解决方法：

在跑之前清除以下默认图。

#清空默认图
tf.reset_default_graph()

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

LYROOO

关注关注

1
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

专栏目录

Tensorflow使用的预训练的resnet_v2_50，resnet_v2_101，resnet_v2_152等模型预测，训练

MachineLP的专栏

10-18

4万+

你要的答案或许都在这里：小鹏的博客目录tensorflow 实现：Inception，ResNet，VGG，MobileNet，Inception-ResNet；tf官方地址： https://github.com/tensorflow/models/tree/master/research/slim我的GitHub：https://github.com/MachineLP/train_...

Tensorflow ResnetV2代码解读

qq_43258953的博客

11-08

562

一，神经网络逐层加深有Degradiation问题，准确率先上升到饱和，再加深会下降，这不是过拟合，是测试集和训练集同时下降的。网络过深导致的梯度弥散或消失。为了解决深度网络的退化问题，何凯明等人提出了残差结构，这个结构解决了网络越深，训练误差反而提升的问题，使得网络理论上可以无限深。二，残差网络的核心是bottleneck网络结构及恒等映射，注意Channel维度变化：，宛如一个中间细两端...

参与评论您还未登录，请先登录后发表或查看评论

VGG16模型tensorflow实现

05-05

包括数据集预处理以及生成tfrecord文件，用自己数据集训练VGG16，亲测可用

ValueError: GraphDef cannot be larger than 2GB.

van的博客

09-18

2677

solution: 不是用默认graph 用例： import tensorflow as tf g1 = tf.Graph() with g1.as_default(): c1 = tf.constant([1.0]) with tf.Graph().as_default() as g2: c2 = tf.constant([2.0]) with tf.Session(graph=g1) as...

深度学习笔记——深度学习框架TensorFlow之Model（十三）

Dream come ture~

07-11

929

TensorFlow的模型存储，恢复与使用在tensorflow中保存模型恢复模型的类是tf.train.Saver() 默认是所有的变量当不传参数，默认就是所有的变量保存模型：#save(sess,save_path,...) save_path = saver.save(sess,"/root/alexnet.tfmodel")从文件中恢复模型（此时restore恢复

tensorflow 在加载大型的embedding模型参数时，会遇到cannot be larger than 2GB

weixin_34056162的博客

11-21

1009

这种问题是，对于每一个变量 variable 由于是基于protobuf存在这大小限制(2G),这个时候，我们需要将embedding拆开，拆分成N等分，来使得每一个 variable都在2G以下; 　　 1 # !/usr/bin/env/python 2 # coding=utf-8 3 import tensorflow as tf 4 import n...

解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题

09-16

主要介绍了解决Tensorflow2.0 tf.keras.Model.load_weights() 报错处理问题，具有很好的参考价值，希望对大家有所帮助。一起跟随想过来看看吧

deeplearning.ai课程知识点整理

mikelkl的博客

12-29

4620

Neural Networks and Deep Learning Introduction to deep learning Neural Networks Basics Logistic Regression as a Neural Network Computation graph 神经网络的计算过程由正向传播（forward propagation ）来进行前向计...

RES2NET

qq_40962619的博客

04-09

1039

Abstract—Representing features at multiple scales is of great importance for numerous vision tasks. Recent advances in backbone convolutional neural networks (CNNs) continually demonstrate stronger mu...

动手学深度学习实现DAY-3

weixin_40054643的博客

02-25

4135

节选自“ElitesAI·动手学深度学习PyTorch版”. （Task06：批量归一化和残差网络；凸优化；梯度下降（1天）） Task09：目标检测基础；图像风格迁移；图像分类案例1（1天） Task10：图像分类案例2；GAN；DCGAN（1天）批量归一化（BatchNormalization）对输入的标准化（浅层模型）处理后的任意一个特征在数据集中所有样本上的均值为0、标准差为1...

关于vgg16运行报class_names未定义错误

Polly001的博客

07-11

1056

关于运行报class_names未定义错误是因为本来含有imagenet_classes.py文件，但是项目中未包含。解决方法：添加imagenet_classes.py文件（github中不包含vgg16训练好的模型）源代码地址：https://github.com/polly000/CNNlearn 网盘地址：链接：https://pan.baidu.com/s/13epuXF7iL2akVSOgP4gOwA 提取码：qj4t import tensorflow as tf import nump

tensorflow从0开始（6）——保存加载模型

热门推荐

searobbers_duck的博客

06-20

5万+

目的学习tensorflow的目的是能够训练的模型，并且利用已经训练好的模型对新数据进行预测。下文就是一个简单的保存模型加载模型的过程。保存模型 import tensorflow as tf import os import numpy as np from tensorflow.python.platform import gfile flags = tf.a

tensorflow2.0使用VGG16预训练模型实现猫狗数据集2分类

weixin_43723269的博客

03-04

2568

tensorflow2.0使用VGG16预训练模型实现猫狗数据集2分类准备： tensorflow2.0.0或以上，最好是GPU版本 python3.7或以上环境，推荐使用Anaconda发行版，本站有相关教程数据集，这里放出网盘链接：猫狗数据集，提取码：ijtz 直接上代码： tensorboard使用方法：训练完成之后，进入对应文件的目录（logs文件夹的主目录），在终端输入：tensorboard --logdir=logs ''' 2021-3-4, edit by wyf python3.

[VGG16模型]现有网络模型的使用与修改的学习及ImageNet 数据集的下载

m0_51816252的博客

06-05

2158

本文以vgg16为例，来学习现有网络模型的使用与修改。除此之外还提及到了ImageNet 数据集的学习，和ImageNet 数据集的下载方面的一些问题。

tensorflow读取数据时报错：cannot create a tensor proto whose content is larger than 2GB

QQ704630835的博客

09-07

7588

一、背景我在做实验时，利用vangogh2photo数据集[1]实现图像风格转换时，由于数据集过大，其中trainB文件夹中包含由6287张图片，trainA包含整整400张图片，每张图片的大小均为256*256，如果直接将数据集全部读取到内存中，就会报错，报错的结果如下图所示： [1]下载地址：https://people.eecs.berkeley.edu/~taesung_park/C...

[tensorflow Debug]ValueError: Cannot create a tensor proto whose content is larger than 2GB

mingo220的博客

11-25

840

问题1: ValueError: Cannot create a tensor proto whose content is larger than 2GB. 解决方法：利用placeholder以及assign方法。参考 import tensorflow as tf import numpy as np w_init = np.random.randn(2**22, 256).astype...

Tensorflow矩阵过大问题的解决

医疗影像检索

07-18

1764

tensorflow保存模型时报错: ValueError: GraphDef cannot be larger than 2GB.

jinbeibei0606的博客

10-28

796

最终查出原因是因为输入网络的数据没有提前定义numpy的array接收，而是直接feed到模型里了，可能这样导致模型一次性保存的太大吧，在feed之前先用np.zeros()定义了数组，用来接收输入的图片数据，再传入network中，问题就能解决了。 ...

tensorflow 源码分析之 protobuf 文件

power的专栏

03-11

1086

a@ubuntu:~/Downloads/tensorflow$ git branch master * r1.13 a@ubuntu:~/Downloads/tensorflow$ find . -name "*proto" 。。。。 ./tensorflow/core/framework/tensor.proto 。。。编译安装后对应的文件 /usr/local/lib/py...

深入解析VGG_ILSVRC_16_layers_fc_reduced.caffemodel模型

从给出的文件信息中，我们可以看出关键词是“VGG_ILSVRC_16_layers_fc_reduced.caffemodel”，“VGG”，以及“压缩包子文件的文件名称列表”中的“vgg”。接下来，我们逐一详细解读这些知识点： ### 标题知识点 ##...