Bert-TextClassification icon indicating copy to clipboard operation
Bert-TextClassification copied to clipboard

Published 20 hours ago • verified publisher icon songyingxin

GPU 显存问题

Open ume-technology opened this issue 3 years ago • 1 comments

我想走一下 Bert + CNN 的SST-2 数据的分类效果。 但是上来就报出 GPU 溢出，我换了公司服务器还是溢出，公司显存 10.75 G 我想知道这是为什么，是确实溢出还是说代码哪里我需要调整呢？

ume-technology avatar Apr 11 '21 06:04 ume-technology