解决‘Hint: If you want to see a list of allocated tensors when OOM happens, add ................‘

tensorflow-gpu运行ner模型的训练代码，出现报错如下图：

主要原因是显卡的显存不够。

解决方法：

1，降低训练的数据输入批次值大小：batch_size

2, 将模型放到gpu显存大的服务器上跑

Done！！！

解决‘Hint: If you want to see a list of allocated tensors when OOM happens, add ................‘相关推荐

Hint: If you want to see a list of allocated tensors when OOM happens,
问题描述: 使用keras搭建siamese网络时,遇到错误如下: OOM when allocating tensor with shape[129024,4096] and type float ...
报错解决：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 ...
使用tensorflow object detection API 训练自己的目标检测模型（三）
在上一篇博客"使用tensorflow object detection API 训练自己的目标检测模型 (二)"中介绍了如何使用LabelImg标记数据集,生成.xml文件,经过 ...
TensorFlow精进之路（十六）：使用slim模型库对图片分类
1.概述 TF-slim是tensorflow的一个轻量级库,它将很多常见tensorflow函数进行封装,使的模型的构建.训练.测试都更加简洁,特别适用于构建结构复杂的深度神经网络.github地址 ...
tensorflow-gpu版本使用问题和方法汇总
tensorflow-gpu版本使用问题和方法汇总训练内存溢出问题问题描述解决方法 keras和tensorflow混合使用的条件下发生内存溢出的问题 batch_size设置导致tensorf ...
Jetson TX2实现EfficientDet推理加速（二）
一.参考资料 TensorRT实现EfficientDet推理加速(一) 二.可能出现的问题 infer推理错误 [TensorRT] ERROR: 2: [pluginV2DynamicExtRun ...
NLP之BERT英文阅读理解问答SQuAD 2.0超详细教程
环境 linux python 3.6 tensorflow 1.12.0 文件准备工作下载bert源代码 : https://github.com/google-research/bert 下载b ...
NVIDIA之TLT迁移学习训练自己的数据集
0 背景 NVIDIA Transfer Learning Toolkit是英伟达推出的迁移学习工具包,使用提供的预训练模型来训练自己的数据,并可以完成模型裁剪.导出到deepstream框架中,实现 ...

解决‘Hint: If you want to see a list of allocated tensors when OOM happens, add ................‘

解决‘Hint: If you want to see a list of allocated tensors when OOM happens, add ................‘相关推荐

最新文章

热门文章