pytorch显存报错,显示还有5G空闲,代码卡在测试部分和保存模型参数这,运行不了,batchsize 减小也一样报错,输入尺寸减小也报错
你好,我是有问必答小助手,非常抱歉,本次您提出的有问必答问题,技术专家团超时未为您做出解答
本次提问扣除的有问必答次数,将会以问答VIP体验卡(1次有问必答机会、商城购买实体图书享受95折优惠)的形式为您补发到账户。
因为有问必答VIP体验卡有效期仅有1天,您在需要使用的时候【私信】联系我,我会为您补发。
显存不够呗,12G应该被其他用户占用了一些了,batchsize填写1,input size写小点看看行不行,如果这还不行的话那就是你的网络太大了,把测试部分改的小点看看,有些训练和测试参数是不一致的,看你的的是否满足,是在不行测试部分用cpu跑,慢是慢了点,但是只要内存足够就能运行