diff options
Diffstat (limited to 'MultiMarginCriterion.lua')
| -rw-r--r-- | MultiMarginCriterion.lua | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/MultiMarginCriterion.lua b/MultiMarginCriterion.lua new file mode 100644 index 0000000..e8de9d9 --- /dev/null +++ b/MultiMarginCriterion.lua @@ -0,0 +1,14 @@ +local MultiMarginCriterion, parent = torch.class('nn.MultiMarginCriterion', 'nn.Criterion') + +function MultiMarginCriterion:__init() + parent.__init(self) + self.sizeAverage = true +end + +function MultiMarginCriterion:updateOutput(input, target) + return input.nn.MultiMarginCriterion_updateOutput(self, input, target) +end + +function MultiMarginCriterion:updateGradInput(input, target) + return input.nn.MultiMarginCriterion_updateGradInput(self, input, target) +end |