代码在反向传播时报错,loss.backward()
报错如下:
查阅很多文章,大多在说CUDA版本与PyTorch版本不匹配
排查下来,发现是在使用nn.BCELoss时输入范围不在[0,1]之间,修改后即可正确运行
“相关推荐”对你有帮助么?
-
非常没帮助
-
没帮助
-
一般
-
有帮助
-
非常有帮助
提交