added Module:evaluate/training

1 files changed, 12 insertions, 0 deletions

diff --git a/Parallel.lua b/Parallel.lua
index 3c625bc..547f444 100644
--- a/Parallel.lua
+++ b/Parallel.lua
@@ -108,6 +108,18 @@ function Parallel:updateParameters(learningRate)
    end
 end
 
+function Parallel:training()
+   for i=1,#self.modules do
+      self.modules[i]:training()
+   end
+end
+
+function Parallel:evaluate()
+   for i=1,#self.modules do
+      self.modules[i]:evaluate()
+   end
+end
+
 function Parallel:share(mlp,...)
    for i=1,#self.modules do
       self.modules[i]:share(mlp.modules[i],...);