From 60c5b6a34e70daf7dde6d96f35203bb14bd57ea1 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Fri, 8 Apr 2016 22:30:32 +0100 Subject: Fixed CPU gradInputs float error --- CTCCriterion.lua | 1 + 1 file changed, 1 insertion(+) diff --git a/CTCCriterion.lua b/CTCCriterion.lua index 22f13af..66ddde4 100644 --- a/CTCCriterion.lua +++ b/CTCCriterion.lua @@ -47,6 +47,7 @@ function CTCCriterion:updateGradInput(output, labels) if (output:type() == 'torch.CudaTensor') then gpu_ctc(acts, grads, labels, sizes) else + grads = grads:float() cpu_ctc(acts:float(), grads:float(), labels, sizes) end self.gradInput = self:revertBatching(grads, tensorSizes):typeAs(output) -- cgit v1.2.3 From 7e75a910842690630b9bde5101ef35c9851c4771 Mon Sep 17 00:00:00 2001 From: SeanNaren Date: Fri, 8 Apr 2016 22:31:05 +0100 Subject: removed type change of float --- CTCCriterion.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CTCCriterion.lua b/CTCCriterion.lua index 66ddde4..0aa4ae5 100644 --- a/CTCCriterion.lua +++ b/CTCCriterion.lua @@ -48,7 +48,7 @@ function CTCCriterion:updateGradInput(output, labels) gpu_ctc(acts, grads, labels, sizes) else grads = grads:float() - cpu_ctc(acts:float(), grads:float(), labels, sizes) + cpu_ctc(acts:float(), grads, labels, sizes) end self.gradInput = self:revertBatching(grads, tensorSizes):typeAs(output) return self.gradInput -- cgit v1.2.3