错误的提示信息如下:Conv3d引起的错误
先说解决方法:
1.将batchsize继续调小,直到不出现问题
2.添加如下代码:

torch.backends.cudnn.enabled = False

原因分析:
可能是cudnn的一个bug,由Conv3d layer引起的
参考:https://github.com/pytorch/pytorch/issues/27588

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐