解决方法:
在代码前加上torch.backends.cudnn.enabled = False
。
参考:https://blog.csdn/qq_33866063/article/details/121009069
解决这个问题后又报 CUDA error: device-side assert triggered
经过调式发现是Unable to get repr for <class 'torch.Tensor'> 错误信息定位在损失函数处。
出错原因:使用BCE损失函数的tensor没有进行(0-1)之间的映射。将input进行sigmoid处理即可。
参考:unable to get repr for <class ‘torch.Tensor‘>错误可能原因_peacefairy的博客-CSDN博客
更多推荐
pytorch bug: Unable to find a valid cuDNN algorithm to run convolution
发布评论