search

autoliuweijie / K-BERT

Source code of K-BERT (AAAI2020)
https://ojs.aaai.org//index.php/AAAI/article/view/5681
951 stars 213 forks source link

请问可以补充下硬件信息吗？ #6

Closed LinMu7177 closed 4 years ago

LinMu7177 commented 4 years ago

谢谢！！！

autoliuweijie commented 4 years ago

我想骗老师买GPU了 。。。。。 谢谢！！！

本文的结果是在P40 GPU上finetune得到的，推荐显存24G以上，内存32G以上。

如果你的GPU显存比较小，可以缩小batch_size，也能跑，但会有一点损失。