最近在搞一些事情,有些情況用python調(diào)用caffe模型時會報(bào)錯(注意不是編譯caffe時報(bào)的)
Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0)? CUDNN_STATUS_BAD_PARAM
Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)? CUDNN_STATUS_BAD_PARAM
在網(wǎng)上查了很多資料,不過都沒解決我的問題。
意外想到曾經(jīng)在某個Github主頁上看到過一句話。最終解決了我的問題。
就是在卷積層convolution_param中添加engine: CAFFE
原文:https://blog.csdn.net/renhanchi/article/details/78500665