diff options
| author | soumith <soumith@gmail.com> | 2016-02-12 17:29:13 +0300 |
|---|---|---|
| committer | soumith <soumith@gmail.com> | 2016-02-12 17:29:13 +0300 |
| commit | 60c85023e872318904c0f53af30f93648c7258df (patch) | |
| tree | 067042d0bd99dddc53f45a8cce8e3b5b966b8fd2 /SpatialBatchNormalization.lua | |
| parent | 3f7c066fae27e202925a9f7c74eaec70130e6e2f (diff) | |
fixing running_var to what it is
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 |