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

在實際項目中出現的該問題,起初以為是cudnn版本的問題,后來才定位到在網絡進行reshape操作的時候 如上所示,第一個參數是輸入圖片的尺寸,在實際的視頻中,輸入的圖片尺寸有可能為 ,那么在reshape操作的時候就會報錯。 在外層加一個保護就好了。 ...

2018-06-04 18:00 0 6226 推薦指數:

查看詳情

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