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