This makes gModules behave consistently with nn.Modules and nn.Container, and solves the issues of trying to change the behaviour of layers like nn.Dropout which may be in nested gModules.

1 files changed, 8 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index c9f669b..e69e79f 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -113,6 +113,14 @@ function gModule:apply(func)
 	end
 end
 
+function gModule:training()
+	self:apply(function(module) module:training() end)
+end
+
+function gModule:evaluate()
+	self:apply(function(module) module:evaluate() end)
+end
+
 function gModule:type(type)
 	self:apply(function(module) module:type(type) end)
 	return self