我是靠谱客的博主 清新蜗牛，最近开发中收集的这篇文章主要介绍ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a，觉得挺不错的，现在分享给大家，希望可以做个参考。
概述
用Bert做命名实体识别的时候报的错，百度了一下说是可能存在
（１）显卡占用的问题，换一个空闲的显卡。
（２）程序问题，最后一层全连接层太大。
试了一下换了一个显卡，多跑了几步还是报错了。
后来将训练数据的batch_size 改小以后没有问题了。batch_size从64改到32
最后
以上就是清新蜗牛为你收集整理的ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a的全部内容，希望文章能够帮你解决ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096]](/uploads/reation/bcimg6.png)
![ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a](/uploads/reation/bcimg8.png)
发表评论 取消回复