zhengli97/PromptKD

教师模型训练条件

Closed this issue · 2 comments

jiachengzhong commented

请问，预训练教师模型大约需要什么样的硬件资源？

zhengli97 commented

3090 24g应该可以，我具体没多少概念了，具体推荐你试试。11g的卡可能跑不动，可以试着缩小bs

jiachengzhong commented

好的，谢谢！