diff options
Diffstat (limited to 'test/test.lua')
-rw-r--r-- | test/test.lua | 7 |
1 files changed, 7 insertions, 0 deletions
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) |