PyTorch Forums
Why MultiLabelMarginLoss take torch.long has arguments?
ptrblck
November 8, 2018, 9:15am
4
I think
nn.BCELoss
could be a good alternative.
Here
is a small example.
show post in topic