基于GPT2的中文摘要生成模型
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
Fino2020 opened this issue 2 years ago · comments
设置batchsize为1的时候会有这个问题,batchsize为2的时候非常多的轮次都是run out of memory,请问这种情况应该怎么办