Skip to content

GitLab

Commit 014fed37 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Merge pull request #7 from gnedster/bug_fix 

Return Grad Input on BatchwiseDropout
parents 7470fd58 8060ca5c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
PyTorch/sparseconvnet/legacy/batchwiseDropout.py
View file @ 014fed37
...@@ -54,7 +54,6 @@ class BatchwiseDropout(SparseModule): ...@@ -54,7 +54,6 @@ class BatchwiseDropout(SparseModule):
self.noise, self.noise,
self.leakiness self.leakiness
) )
return self.output return self.output
def updateGradInput(self, input, gradOutput): def updateGradInput(self, input, gradOutput):
...@@ -68,6 +67,7 @@ class BatchwiseDropout(SparseModule): ...@@ -68,6 +67,7 @@ class BatchwiseDropout(SparseModule):
self.noise, self.noise,
self.leakiness self.leakiness
) )
return self.gradInput
def type(self, t, tensorCache=None): def type(self, t, tensorCache=None):
self.noise.type(t) self.noise.type(t)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment