2020年11月12日
python 报错：reduce failed to synchronize: device-side assert triggered
摘要： 使用torch.nn.BCELoss()需要input >= 0. && input <= 1，但是我的网络最后没有加sigmoid，用nn.BCEWithLogitsLoss()替换torch.nn.BCELoss()即可。 阅读全文
posted @ 2020-11-12 20:22 Xycdada 阅读(537) 评论(0) 推荐(0) 编辑