报错

RuntimeError: reduce failed to synchronize: device-side assert triggered

原因：BCELoss 需要最后一层输出层为sigmod，可用 BCEWithLogitsLoss 进行替换，或直接在网络输出层添加sigmod层。

原文：https://www.cnblogs.com/-yhwu/p/15303265.html