diff options
| author | Clement Farabet <clement.farabet@gmail.com> | 2011-09-27 23:41:35 +0400 |
|---|---|---|
| committer | Clement Farabet <clement.farabet@gmail.com> | 2011-09-27 23:41:35 +0400 |
| commit | 07a8194cf5d9c737af1b291d2b5b057a1f369437 (patch) | |
| tree | 1eae39f3761646165463ac1c94ce97144ffe340d /DistMarginCriterion.lua | |
| parent | 9d5ffa2a97b32286af7b690884f86ffce7ad3cc2 (diff) | |
Added new criterions.
Diffstat (limited to 'DistMarginCriterion.lua')
| -rw-r--r-- | DistMarginCriterion.lua | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/DistMarginCriterion.lua b/DistMarginCriterion.lua new file mode 100644 index 0000000..6b38771 --- /dev/null +++ b/DistMarginCriterion.lua @@ -0,0 +1,14 @@ +local DistMarginCriterion, parent = torch.class('nn.DistMarginCriterion', 'nn.Criterion') + +function DistMarginCriterion:__init() + parent.__init(self) + self.sizeAverage = true +end + +function DistMarginCriterion:forward(input, target) + return input.nn.DistMarginCriterion_forward(self, input, target) +end + +function DistMarginCriterion:backward(input, target) + return input.nn.DistMarginCriterion_backward(self, input, target) +end |