RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

[ericyq]

    def stem(x):
        for conv, bn in [(self.conv1, self.bn1), (self.conv2, self.bn2), (self.conv3, self.bn3)]:
            x = self.relu(bn(conv(x)))
        x = self.avgpool(x)
        return x

error: self.bn1 = nn.BatchNorm2d(width // 2) RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED