From e4e0f877e6289f6e6d4fe4f1ae498d4ca6149256 Mon Sep 17 00:00:00 2001 From: Soumith Chintala Date: Mon, 12 May 2014 12:07:28 -0400 Subject: criterionJacobianTest -> criterionJacobianTest1D --- test/test.lua | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index e3b6a50..c945d4e 100644 --- a/test/test.lua +++ b/test/test.lua @@ -403,13 +403,13 @@ function nntest.WeightedEuclidean() mytester:asserteq(berr, 0, torch.typename(module) .. ' - i/o backward err ') end -local function criterionJacobianTest(cri, input, target) +local function criterionJacobianTest1D(cri, input, target) local eps = 1e-6 local fx = cri:forward(input, target) local dfdx = cri:backward(input, target) -- for each input perturbation, do central difference - local centraldiff_dfdx = torch.Tensor(100) - for i=1,100 do + local centraldiff_dfdx = torch.Tensor():resizeAs(dfdx) + for i=1,input:size(1) do -- f(xi + h) input[i] = input[i] + eps local fx1 = cri:forward(input, target) @@ -433,21 +433,21 @@ function nntest.MSECriterion() local input = torch.rand(100) local target = input:clone():add(torch.rand(100)) local cri = nn.MSECriterion() - criterionJacobianTest(cri, input, target) + criterionJacobianTest1D(cri, input, target) end function nntest.WeightedMSECriterion() local input = torch.rand(100) local target = input:clone():add(torch.rand(100)) local cri = nn.WeightedMSECriterion(torch.rand(100)) - criterionJacobianTest(cri, input, target) + criterionJacobianTest1D(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) + criterionJacobianTest1D(cri, input, target) end function nntest.LogSigmoid() -- cgit v1.2.3