diff options
author | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-01-02 23:07:32 +0300 |
commit | df5925a2f094faf0d3cfd804cd7bea1617aab7df (patch) | |
tree | 07b47a5f8eca80230cdc713ee912c5e955de86bf /SpatialSoftMax.lua | |
parent | e2595a3abfaea264ccbae70557ea3013961927be (diff) |
adding inplace non-linearities
Diffstat (limited to 'SpatialSoftMax.lua')
-rw-r--r-- | SpatialSoftMax.lua | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/SpatialSoftMax.lua b/SpatialSoftMax.lua index 1586882..87af4d5 100644 --- a/SpatialSoftMax.lua +++ b/SpatialSoftMax.lua @@ -57,7 +57,6 @@ 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, |