Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorkoray kavukcuoglu <koray@kavukcuoglu.org>2012-02-10 20:23:00 +0400
committerkoray kavukcuoglu <koray@kavukcuoglu.org>2012-02-10 20:23:00 +0400
commit688447df2cdccc44aa4be37991b04da489b2267b (patch)
tree413cf985b194a82dbafdd7d5b62a2834a23dc337 /SpatialSubtractiveNormalization.lua
parentb5aa64c8363af9e0c8e7143822400befbe9d46ab (diff)
dim check was correction on spatialsubtractivenorm
Diffstat (limited to 'SpatialSubtractiveNormalization.lua')
-rw-r--r--SpatialSubtractiveNormalization.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/SpatialSubtractiveNormalization.lua b/SpatialSubtractiveNormalization.lua
index 070e4b8..dfa8fd2 100644
--- a/SpatialSubtractiveNormalization.lua
+++ b/SpatialSubtractiveNormalization.lua
@@ -66,7 +66,7 @@ end
function SpatialSubtractiveNormalization:updateOutput(input)
-- compute side coefficients
- if (input:size(3) ~= self.coef:size(2)) or (input:size(2) ~= self.coef:size(1)) then
+ if (input:size(3) ~= self.coef:size(3)) or (input:size(2) ~= self.coef:size(2)) then
local ones = input.new():resizeAs(input):fill(1)
self.coef = self.meanestimator:updateOutput(ones)
self.coef = self.coef:clone()