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