在python上加载caffe 的网络时会报上述错误,解决方案如下
方式1
1.在报错的相应层添加 engine: CAFFE
结果:加载网络时没有报错,但是net.forward前向计算时,会报Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_BAD_PARAM
解决如下
2.重新编译caffe源码
编译时, USE_CUDNN := 0
结果:用cudnn编译和不用cudnn编译结果不一样,因此我采用的是方式2
方式2
所有卷积层有group参数的,都加上 engine: CAFFE