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