基于GPT2的中文摘要生成模型
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
DingFong opened this issue 3 years ago · comments
我是自己拿新聞去做訓練,不過幾乎每個訓練都ran out of memory,請問有什麼參數要調嗎??
降低batch_size