MSECriterion.lua


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

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

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

function MSECriterion:updateOutput(input, target)
   return input.nn.MSECriterion_updateOutput(self, input, target)
end

function MSECriterion:updateGradInput(input, target)
   return input.nn.MSECriterion_updateGradInput(self, input, target)
end