diff options
-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 |