错误的提示信息如下:
先说解决方法:
1.将batchsize继续调小,直到不出现问题
2.添加如下代码:
torch.backends.cudnn.enabled = False
原因分析:
可能是cudnn的一个bug,由Conv3d layer引起的
参考:https://github.com/pytorch/pytorch/issues/27588
错误的提示信息如下:
先说解决方法:
1.将batchsize继续调小,直到不出现问题
2.添加如下代码:
torch.backends.cudnn.enabled = False
原因分析:
可能是cudnn的一个bug,由Conv3d layer引起的
参考:https://github.com/pytorch/pytorch/issues/27588