type fix for forwardnodes

1 files changed, 6 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index 551baae..0a81da9 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -265,6 +265,12 @@ function gModule:type(type, tensorCache)
       end
    end
 
+   for i,node in ipairs(self.forwardnodes) do
+      if node.data.input ~= nil then
+         node.data.input = recursiveType(node.data.input, type)
+      end
+   end
+
    self._type = type
    return self
 end