diff options
author | Jean A. Senellart <jean.senellart@systrangroup.com> | 2017-02-05 20:52:46 +0300 |
---|---|---|
committer | Jean A. Senellart <jean.senellart@systrangroup.com> | 2017-02-05 20:52:46 +0300 |
commit | b02074db825f72833e151f3f8ca9bb27358251aa (patch) | |
tree | 9f3b7ceb89a873b284d906e3893958017166e2a2 | |
parent | ae794a24e1ec671c4ec974970050204a4a15be4a (diff) |
fix pointer substraction error
-rw-r--r-- | RNN.lua | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |