我是靠谱客的博主 清新蜗牛,这篇文章主要介绍ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a,现在分享给大家,希望可以做个参考。
用Bert做命名实体识别的时候报的错,百度了一下说是可能存在
(1)显卡占用的问题,换一个空闲的显卡。
(2)程序问题,最后一层全连接层太大。
试了一下换了一个显卡,多跑了几步还是报错了。
后来将训练数据的batch_size 改小以后没有问题了。batch_size从64改到32
最后
以上就是清新蜗牛最近收集整理的关于ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a的全部内容,更多相关ResourceExhaustedError内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复