diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-02-06 10:23:07 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-02-06 10:23:07 +0300 |
commit | 99592ae1a5d7afbbd98d75e10aa7f6c3fae806ef (patch) | |
tree | fab4f993afca0ab9d5c78bb8d9d068cb36bae530 | |
parent | 8b32e18b815fb94f7723038e95d3ae3aab1a3d64 (diff) | |
parent | 0ff6917a73db120a2eb5ce40e871f31411bd03a3 (diff) |
Merge pull request #324 from jsenellart-systran/fix-NYI
storageOffset is calculated in element not bytes
-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 = 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 |