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