From 60c85023e872318904c0f53af30f93648c7258df Mon Sep 17 00:00:00 2001 From: soumith Date: Fri, 12 Feb 2016 09:29:13 -0500 Subject: fixing running_var to what it is --- SpatialBatchNormalization.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'SpatialBatchNormalization.lua') diff --git a/SpatialBatchNormalization.lua b/SpatialBatchNormalization.lua index c8adce7..c25cd92 100644 --- a/SpatialBatchNormalization.lua +++ b/SpatialBatchNormalization.lua @@ -94,7 +94,7 @@ function SpatialBatchNormalization:read(file, version) if version < 2 then if self.running_std then -- for models before https://github.com/soumith/cudnn.torch/pull/101 - self.running_var = self.running_std:pow(-2):add(-self.eps) + self.running_var = self.running_std self.running_std = nil end end -- cgit v1.2.3