Loading

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  阅读(363)  评论(0)    收藏  举报