OSError: (External) Cuda error(2), out of memory.
收藏
请问这个怎么解决
0
收藏
请登录后评论
显存爆了,减小batch size试试
显存爆了只能够:
1. 改小batch_size
2. 换小模型;
3. 换大显卡。
这个错误什么也没改过了一会再运行就行了