Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--RNN.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/RNN.lua b/RNN.lua
index 09690ab..c6a1bbf 100644
--- a/RNN.lua
+++ b/RNN.lua
@@ -613,7 +613,7 @@ local function retrieveLinearParams(self, cuDNNMethod)
nbDims:data(),
filterDimA:data())
- local offset = matrixPointer[0] - self.weight:data()
+ local offset = tonumber(matrixPointer[0]) - tonumber(self.weight:data())
local params = torch.CudaTensor(self.weight:storage(), offset + self.weight:storageOffset(), filterDimA:prod())
table.insert(layerInfo, params)
end