Return Grad Input on BatchwiseDropout

8060ca5c · Ed Ng · 7470fd58 · 8060ca5c
Commit 8060ca5c authored Sep 15, 2017 by Ed Ng
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

PyTorch/sparseconvnet/legacy/batchwiseDropout.py PyTorch/sparseconvnet/legacy/batchwiseDropout.py +1 -1

No files found.
--- a/PyTorch/sparseconvnet/legacy/batchwiseDropout.py
+++ b/PyTorch/sparseconvnet/legacy/batchwiseDropout.py
@@ -54,7 +54,6 @@ class BatchwiseDropout(SparseModule):
            self.noise,
            self.leakiness
            )
-
        return self.output

    def updateGradInput(self, input, gradOutput):
@@ -68,6 +67,7 @@ class BatchwiseDropout(SparseModule):
            self.noise,
            self.leakiness
            )
+        return self.gradInput

    def type(self, t, tensorCache=None):
        self.noise.type(t)