Bert-Multi-Label-Text-Classification icon indicating copy to clipboard operation
Bert-Multi-Label-Text-Classification copied to clipboard

Published 20 hours ago • verified publisher icon lonePatient

CUDA out of memory

Open unstory opened this issue 4 years ago • 2 comments

train_batch_size我设置成1了，为啥还是out of memory. 我的显卡 getforce 930MX , 2G显存 请问最低显存要求是多少？为什么会out of memory? 15万的训练数据，才5个标签，应该不会很吃显存吧

unstory avatar Aug 19 '20 17:08 unstory

2g太小了啊 ，bert本身有点大，另外跟你的seq_length有关。换卡吧 8G我都觉得玩耍不友好了。

lonePatient avatar Aug 20 '20 06:08 lonePatient

感谢回复。我没钱，不换显卡，搞完这个我就不弄nlp了。

unstory avatar Aug 20 '20 07:08 unstory