set self.train=[true,false] for containers

mys007 · Jun 17, 2015 · 8007e9c · 8007e9c
1 parent fdfcd12
commit 8007e9c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Container.lua b/Container.lua
@@ -36,12 +36,14 @@ function Container:training()
     for i=1,#self.modules do
         self.modules[i]:training()
     end
+    parent.training(self)
 end
 
 function Container:evaluate()
     for i=1,#self.modules do
         self.modules[i]:evaluate()
     end
+    parent.evaluate(self)
 end
 
 function Container:share(mlp, ...)