Initialize modules table after read if necessary.

1 files changed, 18 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index a5c7d91..a360d2f 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -393,6 +393,24 @@ function gModule:accGradParameters(input,gradOutput,lr)
    end
 end
 
+function gModule:read(file)
+   local data = file:readObject()
+   for k, v in pairs(data) do
+      self[k] = v
+   end
+
+   -- Initialize the modules table if necessary.
+   if not self.modules then
+      self.modules = {}
+      for _, node in ipairs(self.forwardnodes) do
+         if node.data.module then
+            table.insert(self.modules, node.data.module)
+         end
+      end
+   end
+end
+
+
 function gModule:__tostring__()
    return self.name or torch.type(self)
 end