![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 87
001号
才疏学浅 请多指教
展开
-
深度学习中卷积层和pooling层的输出计算公式
1.卷积层的输出计算公式class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)参数:in_channels(int) – 输入信号的通道 out_channels(int) – 卷积产生的通道 kerner_...原创 2018-09-04 13:35:48 · 22625 阅读 · 0 评论 -
解决ubantu下:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
最终环境:ubantu18.04,2080ti ,cuda10,cudnn7.5.0,python3.7.3,pytorch1.1.0查看cuda版本(打开终端,输入):nvcc -V查看cudnn版本:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看python版本:python查看pyt...原创 2019-07-11 20:59:26 · 7359 阅读 · 1 评论