blob: c435888d34fe72dd3bc4fec7b401ee7abdfee0ff (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
local MultiLabelMarginCriterion, parent = torch.class('nn.MultiLabelMarginCriterion', 'nn.Criterion')
function MultiLabelMarginCriterion:__init()
parent.__init(self)
self.sizeAverage = true
end
function MultiLabelMarginCriterion:updateOutput(input, target)
return input.nn.MultiLabelMarginCriterion_updateOutput(self, input, target)
end
function MultiLabelMarginCriterion:updateGradInput(input, target)
return input.nn.MultiLabelMarginCriterion_updateGradInput(self, input, target)
end
|