diff options
Diffstat (limited to 'SpatialBatchNormalization.lua')
| -rw-r--r-- | SpatialBatchNormalization.lua | 2 |
1 files changed, 1 insertions, 1 deletions
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 |