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 · 22844 阅读 · 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 · 7405 阅读 · 1 评论