search

songyingxin / Bert-TextClassification

Implemention some Baseline Model upon Bert for Text Classification
Other
677 stars 148 forks source link

GPU 显存问题 #26

Open ume-technology opened 3 years ago

ume-technology commented 3 years ago

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

songyingxin commented 3 years ago

https://zhuanlan.zhihu.com/p/65002487

文本长度适当放短点就ok