added Module:evaluate/training

1 files changed, 12 insertions, 0 deletions

diff --git a/Sequential.lua b/Sequential.lua
index a368964..b43bd99 100644
--- a/Sequential.lua
+++ b/Sequential.lua
@@ -83,6 +83,18 @@ function Sequential:updateParameters(learningRate)
    end
 end
 
+function Sequential:training()
+   for i=1,#self.modules do
+      self.modules[i]:training()
+   end
+end
+
+function Sequential:evaluate()
+   for i=1,#self.modules do
+      self.modules[i]:evaluate()
+   end
+end
+
 function Sequential:share(mlp,...)
    for i=1,#self.modules do
       self.modules[i]:share(mlp.modules[i],...);