Caffe FCN Test | Check failed: error == cudaSuccess (2 vs. 0) out of memory

17 篇文章 0 订阅
17 篇文章 0 订阅

在使用caffe跑FCN测试的时候,提示如下错误:
Check failed: error == cudaSuccess (2 vs. 0) out of memory
是内存溢出,而且是cuda报出的,那么就是gpu的缓存溢出了。
遇到这种情况,首先要看的就是训练与测试的batch_size了,而训练过程没有问题,说明训练的batch_size没有问题;
测试过程出现的这个问题,查看的fcn的源码中关于测试的部分:
这里写图片描述
这里的reshape(1,*),其中1就表示batch_size,所以已经没法调整了。
于是又怀疑是不是使用的原图太大,占用空间多,而修改后依然报出这个错误,于是只能先不测试,直接训练了。
希望有同样问题的小伙伴一起讨论。

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值