caffe cudnnSetConvolution2dDescriptor error: too few arguments in function call

cuDNN卷积配置详解

最新推荐文章于 2024-04-23 17:57:38 发布

原创最新推荐文章于 2024-04-23 17:57:38 发布 · 6.1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#caffe

Caffe 专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了如何使用cuDNN库中的cudnnSetConvolution2dDescriptor函数正确设置2D卷积描述符。根据cuDNN版本的不同，该函数的调用方式有所变化。对于cuDNN 6.0及以上版本，需要额外指定数据类型参数。

参考https://devtalk.nvidia.com/default/topic/1002826/question-about-cudnnsetconvolution2ddescriptor/
在cudnn.hpp中查看cudnnSetConvolution2dDescriptor的正确的调用方法：

template <typename Dtype>
inline void setConvolutionDesc(cudnnConvolutionDescriptor_t* conv,
    cudnnTensorDescriptor_t bottom, cudnnFilterDescriptor_t filter,
    int pad_h, int pad_w, int stride_h, int stride_w) {
#if CUDNN_VERSION_MIN(6, 0, 0)
  CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION,
      dataType<Dtype>::type));
#else
    CUDNN_CHECK(cudnnSetConvolution2dDescriptor(*conv,
      pad_h, pad_w, stride_h, stride_w, 1, 1, CUDNN_CROSS_CORRELATION));
#endif
}