ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[8192,3072] a
用Bert做命名实体识别的时候报的错,百度了一下说是可能存在(1)显卡占用的问题,换一个空闲的显卡。(2)程序问题,最后一层全连接层太大。试了一下换了一个显卡,多跑了几步还是报错了。后来将训练数据的batch_size 改小以后没有问题了。batch_size从64改到32...