caffe報錯:cudnn.hpp:86] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM 原因


在實際項目中出現的該問題,起初以為是cudnn版本的問題,后來才定位到在網絡進行reshape操作的時候

input_layer->Reshape({(int)imgin.size(), input_layer->shape(1),
                          input_layer->shape(2), input_layer->shape(3)});

如上所示,第一個參數是輸入圖片的尺寸,在實際的視頻中,輸入的圖片尺寸有可能為0,那么在reshape操作的時候就會報錯。

在外層加一個保護就好了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM