Merge pull request #62 from hughperkins/fix-gradOutputBuffer-type-change

"nesting.lua:36: bad argument #1 to 'resizeAs'"

1 files changed, 7 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index 3a3d9fe..4c37525 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -192,6 +192,13 @@ function gModule:type(type)
 
 	-- Loop through modules and convert data
 	self:apply(function(module) module:type(type) end)
+
+  for i,node in ipairs(self.backwardnodes) do
+    if node.data.gradOutputBuffer ~= nil then
+      node.data.gradOutputBuffer = node.data.gradOutputBuffer:type(type)
+    end
+  end
+
 	return self
 end