AbsCriterion.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14

local AbsCriterion, parent = torch.class('nn.AbsCriterion', 'nn.Criterion')

function AbsCriterion:__init()
   parent.__init(self)
   self.sizeAverage = true
end

function AbsCriterion:updateOutput(input, target)
   return input.nn.AbsCriterion_updateOutput(self, input, target)
end

function AbsCriterion:updateGradInput(input, target)
   return input.nn.AbsCriterion_updateGradInput(self, input, target)
end