BCELoss
BCEWithLogitsLoss
将sigmoid和BCELoss结合在一起,数据的稳定性更好。
torch.nn.functional:
binary_cross_entropy_with_logits: 同BCEWithLogitsLoss
BCELoss
BCEWithLogitsLoss
将sigmoid和BCELoss结合在一起,数据的稳定性更好。
torch.nn.functional:
binary_cross_entropy_with_logits: 同BCEWithLogitsLoss