diff options
Diffstat (limited to 'MSECriterion.lua')
-rw-r--r-- | MSECriterion.lua | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/MSECriterion.lua b/MSECriterion.lua new file mode 100644 index 0000000..655c74f --- /dev/null +++ b/MSECriterion.lua @@ -0,0 +1,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 |