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