3.1 使用迁移学习的方法训练模型

技术博客与活动 历史活动 GPU Hackathon
1

训练时cuda内存不足如何解决

2

调整一下batchsize 设小一点， 在模型配置文件中的trainer部分gradient_clip_val参数调小一点 ， 关掉占用显存的进程重启一下电脑试试