Multi class SVM loss

deeplearner93 · April 20, 2018, 8:03pm

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

arturml · April 21, 2018, 1:21am

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

deeplearner93 · April 23, 2018, 10:31pm

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