caffe测试时报错 [....syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out of memory

现象: 在caffe复现DenseNet的过程中,训练完成了,就到了测试阶段。当开始运行之后,过了一段时间发现报了一个错误。 F0523 12:55:04.591845 17608 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out of memory。爆内存了。如图所示。 分析一:batch_size 一般这个问题
相关文章
相关标签/搜索