代码在反向传播时报错,loss.backward()
报错如下:
查阅很多文章,大多在说CUDA版本与PyTorch版本不匹配
排查下来,发现是在使用nn.BCELoss时输入范围不在[0,1]之间,修改后即可正确运行
RuntimeError: CUDA error: device-side assert triggered
最新推荐文章于 2023-05-05 18:16:27 发布
代码在反向传播时报错,loss.backward()
报错如下:
查阅很多文章,大多在说CUDA版本与PyTorch版本不匹配
排查下来,发现是在使用nn.BCELoss时输入范围不在[0,1]之间,修改后即可正确运行