Chinese-Text-Classification
Chinese-Text-Classification copied to clipboard
内存溢出怎么解决?
在最初的代码中使用的时候没有发生内存溢出的问题,但是在使用中文数据集进行训练的时候发生内存溢出导致内存不足,有没有什么解决办法?
仔细看了一下不是内存溢出,是显存溢出了,有什么解决办法么?
batch size 设置小一点
我把batch size减小一倍以后依然会出现这个问题