本文介绍了UIE模型微调过程中遇到的显存问题，探讨了如何在显存有限的情况下调整batch size，并分享了使用在线GPU平台（如百度AI平台）进行按量计费训练的经验。此外，还提到了安装paddlepaddle-gpu和visualdl等必要包的步骤，以及运行微调脚本的详细流程。
参考：uie模型微调个人总结
github：https://github.com/PaddlePaddle/PaddleNLP/tree/develop/model_zoo/uie
1、显存问题
之前是在实验室的服务器上跑（2080ti，12G显存），频频出现以下报错：
原始的batch size是16，即使调为4，也会在跑了一会以后出现以上问题。经过分析，应该是显存不够的原因。
2、在线GPU：按量计费
购买链接：https://ai.baidu.com/bml/
新用户有50小时免费额度。额外的训练计费：
选择32
最低0.47元/天 解锁文章
扫一扫
929
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
