diff options
| author | soumith <soumith@fb.com> | 2014-12-27 00:33:38 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2014-12-27 00:33:38 +0300 |
| commit | 7b0d52e68e07dee37ed347a607f6d9fa94802449 (patch) | |
| tree | d7cd591eff8a741100a9df5e06431430ea500b59 /SpatialSoftMax.lua | |
| parent | 7fbf2cdd07cc95962a80a02add85c79562724ca7 (diff) | |
fix for R2-RC1 beta=0 bug
Diffstat (limited to 'SpatialSoftMax.lua')
| -rw-r--r-- | SpatialSoftMax.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialSoftMax.lua b/SpatialSoftMax.lua index 87af4d5..1586882 100644 --- a/SpatialSoftMax.lua +++ b/SpatialSoftMax.lua @@ -57,6 +57,7 @@ function SpatialSoftMax:updateGradInput(input, gradOutput) assert((gradOutput:dim() == 4 or gradOutput:dim() == 3) and gradOutput:isContiguous()); self:createIODescriptors(input) + self.gradInput:fill(1) -- to get around bug in R2-RC1 https://github.com/soumith/cudnn.torch/issues/9 errcheck('cudnnSoftmaxBackward', cudnn.handle[cutorch.getDevice()-1], self.algorithm, self.mode, |