Loading

2021年9月17日
pytorch debug：RuntimeError: reduce failed to synchronize: device-side assert triggered
摘要： 报错 RuntimeError: reduce failed to synchronize: device-side assert triggered 原因：BCELoss 需要最后一层输出层为sigmod，可用 BCEWithLogitsLoss 进行替换，或直接在网络输出层添加sigmod层。 阅读全文
posted @ 2021-09-17 09:30 _yhwu 阅读(325) 评论(0) 推荐(0) 编辑