qingkongzhiqian / GPT2-Summary

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

請問各位大神,遇到ran out of memory怎麼辦,我是拿自己的新聞資料去做訓練。

DingFong opened this issue · comments

1628694960562

我是自己拿新聞去做訓練,不過幾乎每個訓練都ran out of memory,請問有什麼參數要調嗎??

降低batch_size