blob: deb903e3f02b54466457664eadf07c207a969a06 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
local MarginCriterion, parent =
torch.class('nn.MarginCriterion', 'nn.Module')
function MarginCriterion:__init(margin)
parent.__init(self)
margin=margin or 1
self.margin = margin
self.gradInput = torch.Tensor(1)
end
function MarginCriterion:updateOutput(input,y)
self.output=math.max(0, self.margin- y* input[1])
return self.output
end
function MarginCriterion:updateGradInput(input, y)
if (y*input[1])<self.margin then
self.gradInput[1]=-y
else
self.gradInput[1]=0;
end
return self.gradInput
end
|