ResourceExhaustedError

一开始没有用GPU在跑，直接用了CPU，模型最后接了两个2048的全连接层，运行直接报错。
解决办法：
1、减少全连接层参数。

x = Dense(512,activation='relu')(x)

2、batch改小一点。

batch_size=10

3、用GPU跑代码训练。

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"#设置根据电脑对应的接口ID选择设备
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"#调用编号为0和1的gpu

参考博文

ResourceExhaustedError相关推荐

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape
报错解决:ResourceExhaustedError: OOM when allocating tensor with shape 早上在使用tensorflow时遇到如下报错: Traceback ...
FakeAPP训练时错误【ResourceExhaustedError: OOM】解决方法
背景在使用FakeApp训练的时候出错,表现为点击"train"按钮后一段时间就报错,且没有训练预览窗口弹出. log文件里显示 undefinedUsing GPU0 for ...
ResourceExhaustedError 解决方案
ResourceExhaustedError 解决方案参考文章: (1)ResourceExhaustedError 解决方案 (2)https://www.cnblogs.com/heiao10d ...
OOM ResourceExhaustedError
错误:OOM ResourceExhaustedError 本文记录ResourceExhaustedError出错过程,提出解决方案. 1.问题描述在学习keras时,运行从GitHub上down ...
报错：ResourceExhaustedError OOM when allocating
日萌社人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新) 报错:ResourceExhaustedError OOM w ...
关于错误：OOM ResourceExhaustedError 的完美解决方法
遇到的问题: 我使用MobileNet训练图片分类器,参数大概有几百万个并不是很多,然而报错OOM ResourceExhaustedError 分析: 这样的话可能有两个原因: (1)batchsi ...
Mask_RCNN安装与踩过的坑
一.Mask_RCNN下载 https://www.bilibili.com/video/BV1M7411x7is?t=629&p=5 按照上述教程的话,安装的是ballon例子的Mask_R ...
Windows Tensorflow GPU安装
GPU资源对神经网络模型的训练很重要,应充分利用电脑的显卡资源,加快模型的训练速度.这里是本人安装tensorflow-gpu的过程,记录了安装的步骤以及在每个步骤中参考的资料以及所遇到的坑. 大体步 ...
Tensorflow一些常用基本概念与函数
参考文献 Tensorflow一些常用基本概念与函数 http://www.cnblogs.com/wuzhitj/archive/2017/03.html Tensorflow笔记:常用函数说明: ...
使用tensorflow object detection API 训练自己的目标检测模型（三）
在上一篇博客"使用tensorflow object detection API 训练自己的目标检测模型 (二)"中介绍了如何使用LabelImg标记数据集,生成.xml文件,经过 ...

ResourceExhaustedError

ResourceExhaustedError相关推荐

最新文章

热门文章