Created by: Biaocsu

利用fine-tune好的ERNIE做文本分类任务，推理时关闭CUDA，占用内存12.8G，无法想象，是哪里出现了内存泄漏吗？望检查！相同情况pytorch_bert占用内存900M, tensorflow_bert占用内存850M 若打开CUDA，即CUDA=True，则占用内存836M