报错

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

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

pytorch debug：RuntimeError: reduce failed to synchronize: device-side assert triggered