网上查阅了一些资料,包括什么改虚拟内存,都没什么用,最后我修改了batch_size,把它改小,模型终于能跑了
pytorch模型提示超出内存cuda runtime error(2): out of memory
最新推荐文章于 2020-11-27 14:19:12 发布
网上查阅了一些资料,包括什么改虚拟内存,都没什么用,最后我修改了batch_size,把它改小,模型终于能跑了