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:
authorSoumith Chintala <soumith@gmail.com>2017-02-06 03:27:30 +0300
committerGitHub <noreply@github.com>2017-02-06 03:27:30 +0300
commit8b32e18b815fb94f7723038e95d3ae3aab1a3d64 (patch)
treefe55d4c302e42a4f58ae83b47d4fc328c6436bf9
parent79b11b4f870a9ee11eceb30c1087de6652390723 (diff)
parentb02074db825f72833e151f3f8ca9bb27358251aa (diff)
Merge pull request #323 from jsenellart-systran/fix-NYI
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 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