PyTorch——错误集锦

1、KeyError: class ‘torch.cuda.ByteTensor’

解决
关于这个错误网上介绍的并不多，只找到一个解答：ByteTensor not working with F.conv2d?。PyTorch中的大部分操作都是针对FloatTensor和DoubleTensor的。

2、RuntimeError: CUDNN_STATUS_BAD_PARAM

解决
输入的size不正确，卷积层的输入size为(N, C, H, W)。

3、TypeError: max() got an unexpected keyword argument ‘keepdim

原因不明。
解决
torch.max(input, dim)而不用torch.max(input, dim, keepdim)

４、RuntimeError: getCudnnDataType() not supported for B

调用Module.forward()方法，在计算Conv2d时出现这个错误。