From 900e8522fd2cd759803babde7bf8bae43daacaf3 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Mon, 12 May 2014 11:27:54 -0400 Subject: fixing BCECriterion, adding test, closing https://github.com/torch/nn/pull/10 --- test/test.lua | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index 7eb2d44..e3b6a50 100644 --- a/test/test.lua +++ b/test/test.lua @@ -443,6 +443,13 @@ function nntest.WeightedMSECriterion() criterionJacobianTest(cri, input, target) end +function nntest.BCECriterion() + local input = torch.rand(100) + local target = input:clone():add(torch.rand(100)) + local cri = nn.BCECriterion() + criterionJacobianTest(cri, input, target) +end + function nntest.LogSigmoid() local ini = math.random(10,20) local inj = math.random(10,20) -- cgit v1.2.3