diff options
author | Eric Cosatto <cosatto@nec-labs.com> | 2017-10-02 22:36:29 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-10-02 22:36:29 +0300 |
commit | d135408b965da4d9c2d7ec0de37f526f9d401ce0 (patch) | |
tree | af849e1e355c5a3da9936fcef4e3c6d5f487cf40 | |
parent | 7762e143d86e1664a2675065420d57a7a4195d07 (diff) |
fix condition logic for coef recalculation in SpatialSubtractiveNormalization (#1283)
* replace lon glong types with size_t
Work around bug in msvc compiler in win32 mode
* fix condition logic for coef recalculation in SpatialSubtractiveNormalization
Existing conditional expression always verified, resulting in overhead
as coeffcients were recalculated every time. Fix and simplify expression
by using chached input size.
-rw-r--r-- | SpatialSubtractiveNormalization.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/SpatialSubtractiveNormalization.lua b/SpatialSubtractiveNormalization.lua index e2da2c6..7fa4402 100644 --- a/SpatialSubtractiveNormalization.lua +++ b/SpatialSubtractiveNormalization.lua @@ -63,7 +63,8 @@ end function SpatialSubtractiveNormalization:updateOutput(input) -- compute side coefficients local dim = input:dim() - if input:dim()+1 ~= self.coef:dim() or (input:size(dim) ~= self.coef:size(dim)) or (input:size(dim-1) ~= self.coef:size(dim-1)) then + if not self._inpsz or not input:isSize(self._inpsz) then + self._inpsz = input:size() self.ones = self.ones or input.new() self._coef = self._coef or self.coef.new() if dim == 4 then |