问题解决：RuntimeError: CUDA out of memory.(....; 5.83 GiB reserved in total by PyTorch)

https://blog.csdn.net/weixin_41587491/article/details/105488239
可以改batch_size 通常有64、32啥的

posted @ 2021-08-19 16:56 ice猫猫3 阅读(1301) 评论(0) 收藏举报

刷新页面返回顶部