Adding a :applyToModules() method to gModule, and :training() and :evaluate() should be applied 'self' as well.

Author: yozw

gmodule.lua

@@ -177,17 +177,25 @@ function gModule:share(gm, ...)
 end
 
 function gModule:training()
+   parent.training(self)
    for _, m in ipairs(self.modules) do
       m:training()
    end
 end
 
 function gModule:evaluate()
+   parent.evaluate(self)
    for _, m in ipairs(self.modules) do
       m:evaluate()
    end
 end
 
+function gModule:applyToModules(func)
+   for _, m in ipairs(self.modules) do
+      func(m)
+   end
+end
+
 --[[ Recursively applies type(type_str) to any tensors in the argument. If the
 argument is a tensor, type(type_str) is applied; if the argument is an array,
 this function recurses into it. ]]