blob: 6b3877174e5ece082b12edc704d197ae5b1aae2d (
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:forward(input, target)
return input.nn.DistMarginCriterion_forward(self, input, target)
end
function DistMarginCriterion:backward(input, target)
return input.nn.DistMarginCriterion_backward(self, input, target)
end
|