fastnlp / fastHan

fastHan是基于fastNLP与pytorch实现的中文自然语言处理工具,像spacy一样调用方便。
Apache License 2.0
752 stars 86 forks source link

Tesla-T4(16G)finetune修改batch_size时总是被Killed #40

Open LuoFanA595 opened 1 year ago

LuoFanA595 commented 1 year ago

image 实际显存占用率也不高,自身分词数据只有120M大小, 修改batch_size=16或者32时均被Killed,只能使用源码batch_size=8. 使用Tesla-T4训练1个Epoch都需要一周时间~~ 有没有大佬遇到这种问题该怎么解