search

wind91725 / gpt2-ml-finetune-

根据gpt2-ml中文模型finetune自己的数据集
Apache License 2.0
43 stars 15 forks source link

加载原gpt2-ml的大模型，9G的显存也会满了,报OOM，请问作者是使用多大显存？ #17

Open huangdacheng opened 4 years ago

huangdacheng commented 4 years ago

![image](https://user-images.githubusercontent.com/6139188/93421785-e49dcc00-f8e4-11ea-9d64-2471fa1e9d6a.png 如题

wind91725 commented 4 years ago

单块v-100 32G显存

在 2020-09-17 12:55:05，"huangdacheng" notifications@github.com 写道：

![image](https://user-images.githubusercontent.com/6139188/93421785-e49dcc00-f8e4-11ea-9d64-2471fa1e9d6a.png 如题

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.