Multi class SVM loss

Is there any way we could implement this loss function in pytorch? I would love to test it out on my image dataset.

There is the torch.nn.MultiLabelMarginLoss. Is this the loss you’re looking for?

Ah! Sorry for the late reply. Missed this. I guess that could be it although I haven’t delved into it much. Thanks!

1 Like