add read for BatchNorm

1 files changed, 10 insertions, 0 deletions

diff --git a/BatchNormalization.lua b/BatchNormalization.lua
index b2520be..a342063 100644
--- a/BatchNormalization.lua
+++ b/BatchNormalization.lua
@@ -118,6 +118,16 @@ function BatchNormalization:clearDesc()
    self.sDesc = nil
 end
 
+function BatchNormalization:read(file, version)
+   parent.read(self, file)
+   if version < 2 then
+      if self.running_std then
+         self.running_var = self.running_std:pow(-2):add(-self.eps)
+         self.running_std = nil
+      end
+   end
+end
+
 function BatchNormalization:write(f)
    self:clearDesc()
    local var = {}