diff options
Diffstat (limited to 'RNN.lua')
-rw-r--r-- | RNN.lua | 7 |
1 files changed, 6 insertions, 1 deletions
@@ -613,7 +613,12 @@ local function retrieveLinearParams(self, cuDNNMethod) nbDims:data(), filterDimA:data()) - local offset = (tonumber(matrixPointer[0]) - tonumber(self.weight:data()))/self.weight:elementSize() + local offset + if jit then + offset = matrixPointer[0] - self.weight:data() + else + offset = (tonumber(matrixPointer[0]) - tonumber(self.weight:data()))/self.weight:elementSize() + end local params = torch.CudaTensor(self.weight:storage(), offset + self.weight:storageOffset(), filterDimA:prod()) table.insert(layerInfo, params) end |