diff options
| author | Jonathan Tompson <jonathantompson@gmail.com> | 2016-06-07 18:44:11 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2016-06-07 18:44:11 +0300 |
| commit | 1b7d3473be65285755d478da19d0b679a7acba71 (patch) | |
| tree | defa1fd1c6af81c7e69c015dc02bde393ae20faa | |
| parent | 531a0636093de27e29771e513180b120a95889ae (diff) | |
Quick fix for type. (#118)
| -rw-r--r-- | gmodule.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/gmodule.lua b/gmodule.lua index 7bdd13d..551baae 100644 --- a/gmodule.lua +++ b/gmodule.lua @@ -260,7 +260,8 @@ function gModule:type(type, tensorCache) for i,node in ipairs(self.backwardnodes) do if node.data.gradOutputBuffer ~= nil then - node.data.gradOutputBuffer = node.data.gradOutputBuffer:type(type) + node.data.gradOutputBuffer = + recursiveType(node.data.gradOutputBuffer, type) end end |