multimargin supports p=2

1 files changed, 7 insertions, 0 deletions

diff --git a/test.lua b/test.lua
index f555b50..a8f81f5 100644
--- a/test.lua
+++ b/test.lua
@@ -693,6 +693,13 @@ function nntest.MarginCriterion()
    criterionJacobianTest1D(cri, input, target)
 end
 
+function nntest.MultiMarginCriterion()
+   local input = torch.rand(100)
+   local target = math.random(1,100)
+   local cri = nn.MultiMarginCriterion(math.random(1,2))
+   criterionJacobianTest1D(cri, input, target)
+end
+
 function nntest.WeightedMSECriterion()
    local input = torch.rand(10)
    local target = input:clone():add(torch.rand(10))