From a0306ea4172331a007c51d250b9e5bbd58074284 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Fri, 13 May 2016 19:51:31 +0100 Subject: Updated Criterion name in tests --- test/test-all.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test-all.lua b/test/test-all.lua index 7d1d13e..edc69aa 100644 --- a/test/test-all.lua +++ b/test/test-all.lua @@ -525,7 +525,7 @@ function nnxtest.TreeNLLCriterion() end function nnxtest.CTCCriterion() - local criterion = nn.CTCCriterionTest() + local criterion = nn.CTCCriterion() local acts = torch.Tensor({{{0,0,0,0,0}}}):transpose(1, 2):contiguous() -- input is seqLength x batch x inputDim local targets = {{1}} local sizes = torch.Tensor({1}) @@ -549,7 +549,7 @@ function nnxtest.CTCCriterion() local sizes = torch.Tensor({1,3,3}) mytester:eq(criterion:updateOutput(acts, targets, sizes), 13.904030799866, precision, "CTCCriterion.batchTest") local gradOutputNorm = criterion:updateGradInput(acts, targets, sizes) - criterion = nn.CTCCriterionTest(true) -- batchFirst true, input is batch x seqLength x inputDim + criterion = nn.CTCCriterion(true) -- batchFirst true, input is batch x seqLength x inputDim local batchFirstActs = torch.Tensor({ {{0,0,0,0,0},{0,0,0,0,0},{0,0,0,0,0}}, -- cgit v1.2.3