storageOffset is calculated in element not bytes

1 files changed, 1 insertions, 1 deletions

diff --git a/RNN.lua b/RNN.lua
index 096e316..edf16bc 100644
--- a/RNN.lua
+++ b/RNN.lua
@@ -613,7 +613,7 @@ local function retrieveLinearParams(self, cuDNNMethod)
                 nbDims:data(),
                 filterDimA:data())
 
-            local offset = tonumber(matrixPointer[0]) - tonumber(self.weight:data())
+            local offset = (tonumber(matrixPointer[0]) - tonumber(self.weight:data()))/self.weight:elementSize()
             local params = torch.CudaTensor(self.weight:storage(), offset + self.weight:storageOffset(), filterDimA:prod())
             table.insert(layerInfo, params)
         end