cuDNN error: CUDNN_STATUS_BAD_PARAM
cuDNN error: CUDNN_STATUS_BAD_PARAM
nn.Conv3d(in_channels=conv3dIn, out_channels=conv3dOut,
kernel_size=conv3dKer_size, stride=conv3dStride, padding=conv3dPad)
self.adaptive2d = nn.AdaptiveAvgPool2d((H,W))
kernel_size设置错误出现的
用
import torchsnooper
with torchsnooper.snoop():
代码块
@torchsnooper.snoop():
函数