Equivalent of TensorFlow's Sigmoid Cross Entropy With Logits in Pytorch

vision
9

Hi @varunagrawal,

did you get MultiLabelSoftMarginLoss to work on a multi-label classification test problem?

I tried applying it to a multi-label MNIST test, (each image is label by it’s original class, and the class-1), but it didn’t work?