pytorch bug: Unable to find a valid cuDNN algorithm to run convolution

解决方法:
在代码前加上torch.backends.cudnn.enabled = False

参考:https://blog.csdn.net/qq_33866063/article/details/121009069

解决这个问题后又报 CUDA error: device-side assert triggered

经过调式发现是unable to get repr for 错误信息定位在损失函数处。

出错原因:使用BCE损失函数的tensor没有进行(0-1)之间的映射。将input进行sigmoid处理即可。

参考:unable to get repr for <class ‘torch.Tensor‘>错误可能原因_peacefairy的博客-CSDN博客

你可能感兴趣的:(pytorch,bug,深度学习)