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:
authorJean A. Senellart <jean.senellart@systrangroup.com>2017-02-05 20:52:46 +0300
committerJean A. Senellart <jean.senellart@systrangroup.com>2017-02-05 20:52:46 +0300
commitb02074db825f72833e151f3f8ca9bb27358251aa (patch)
tree9f3b7ceb89a873b284d906e3893958017166e2a2
parentae794a24e1ec671c4ec974970050204a4a15be4a (diff)
fix pointer substraction error
-rw-r--r--RNN.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/RNN.lua b/RNN.lua
index 4bb5a98..096e316 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