CPM_LM_bert4keras icon indicating copy to clipboard operation
CPM_LM_bert4keras copied to clipboard

Published 20 hours ago • verified publisher icon bojone

用CPM做分类任务报OOM

Open lambert-lan opened this issue 3 years ago • 2 comments

苏老师您好 我尝试用bert4keras和CPM做下游分类任务的时候发现GPU显存不够 报了OOM 设备是V100 32G显存 想请教一下您这边有没有遇到过同样的问题？ 或者有什么解决方法？

lambert-lan avatar Nov 05 '21 09:11 lambert-lan

如何做下游任务？finetune？

bojone avatar Nov 05 '21 09:11 bojone

如何做下游任务？finetune？

对的 finetune 做文本分类
哪怕batch_size调到1了 依然OOM

lambert-lan avatar Nov 08 '21 01:11 lambert-lan