用户
 找回密码
 立即注册
发表于 2021-11-25 20:45:42
7131
训练时cuda内存不足如何解决
Screenshot from 2021-11-25 20-37-28.png
使用道具 举报 回复
发表于 2021-11-25 21:02:52
调整一下batchsize 设小一点, 在模型配置文件中的trainer部分gradient_clip_val参数调小一点 , 关掉占用显存的进程重启一下电脑试试
使用道具 举报 回复 支持 反对
发新帖
您需要登录后才可以回帖 登录 | 立即注册