diff options
author | Soumith Chintala <soumith@gmail.com> | 2014-11-22 18:17:18 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2014-11-22 18:17:18 +0300 |
commit | 7f115ee237d12e9bd95967f92079f2345b7a6c37 (patch) | |
tree | f0fdcb3988136c06b0ddc42843345d0fce6cb1bb /WeightedEuclidean.lua | |
parent | 280be1532b1b9ffb5323b7434030813788cb44ce (diff) |
fix for euclidian patch from #108
Diffstat (limited to 'WeightedEuclidean.lua')
-rw-r--r-- | WeightedEuclidean.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/WeightedEuclidean.lua b/WeightedEuclidean.lua index 22dbd1d..3808db6 100644 --- a/WeightedEuclidean.lua +++ b/WeightedEuclidean.lua @@ -75,13 +75,13 @@ function WeightedEuclidean:accGradParameters(input, gradOutput, scale) self.temp:copy(self.templates:select(2,o)):add(-1,input) self.temp:cmul(self.diagCov:select(2,o)):cmul(self.diagCov:select(2,o)) self.temp:mul(gradOutput[o]/self.output[o]) - self.gradTemplates:select(2,o):add(scale * self.temp) + self.gradTemplates:select(2,o):add(scale, self.temp) self.temp:copy(self.templates:select(2,o)):add(-1,input) self.temp:cmul(self.temp) self.temp:cmul(self.diagCov:select(2,o)) self.temp:mul(gradOutput[o]/self.output[o]) - self.gradDiagCov:select(2,o):add(scale * self.temp) + self.gradDiagCov:select(2,o):add(scale, self.temp) end end end |