hans

hans

【Caffe】ステータスのチェックに失敗しました status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM


最近、いくつかのことをやっていて、Python で Caffe モデルを呼び出すときにエラーが発生することがあります(Caffe のコンパイル時のエラーではありません)。

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

インターネットで多くの情報を調べましたが、問題を解決することはできませんでした。

偶然、以前 GitHub のページで見た一つの文を思い出しました。それが私の問題を解決しました。

それは、畳み込み層の convolution_param に engine: CAFFE を追加することです。

読み込み中...
文章は、創作者によって署名され、ブロックチェーンに安全に保存されています。