hans

hans

【Caffe】檢查失敗 狀態 == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM


最近在搞一些事情,有些情況用 python 調用 caffe 模型時會報錯(注意不是編譯 caffe 時報的)

檢查失敗: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

檢查失敗: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM

在網上查了很多資料,不過都沒解決我的問題。

意外想到曾經在某個 Github 主頁上看到過一句話。最終解決了我的問題。

就是在卷積層 convolution_param 中添加 engine: CAFFE

載入中......
此文章數據所有權由區塊鏈加密技術和智能合約保障僅歸創作者所有。