diff options
author | SeanNaren <taz838@hotmail.co.uk> | 2016-05-13 21:51:31 +0300 |
---|---|---|
committer | SeanNaren <taz838@hotmail.co.uk> | 2016-05-13 21:51:31 +0300 |
commit | a0306ea4172331a007c51d250b9e5bbd58074284 (patch) | |
tree | b99170716e816f10e732ebb1add9999b8c79cce8 /test | |
parent | c4006a5f32fae62f8e7bd261d93f99d47987b5f3 (diff) |
Updated Criterion name in tests
Diffstat (limited to 'test')
-rw-r--r-- | test/test-all.lua | 4 |
1 files 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}}, |