qingkongzhiqian / GPT2-Summary

基于GPT2的中文摘要生成模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cuda memory 超了

jiezouguihuafu opened this issue · comments

commented

我用这个模型训练的时候,文本长度超过400就一直报cuda内存超了,你训练的时候句子一般都多长啊,我用的GPU内存是32G的