我是靠谱客的博主 清新蜗牛，这篇文章主要介绍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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![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)
发表评论 取消回复