Skip to content

GitLab

Commit c74697aa authored by Ed Ng's avatar Ed Ng
Browse files

Small Fix

parent 3ff930b7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
PyTorch/sparseconvnet/legacy/batchwiseDropout.py
View file @ c74697aa
...@@ -48,7 +48,7 @@ class BatchwiseDropout(SparseModule): ...@@ -48,7 +48,7 @@ class BatchwiseDropout(SparseModule):
self.output.metadata = input.metadata self.output.metadata = input.metadata
self.output.spatialSize = input.spatialSize self.output.spatialSize = input.spatialSize
typed_fn(input, 'BatchwiseMultiplicativeDropout_updateOutput')( typed_fn(input.features, 'BatchwiseMultiplicativeDropout_updateOutput')(
input.features, input.features,
self.output.features, self.output.features,
self.noise, self.noise,
...@@ -61,7 +61,7 @@ class BatchwiseDropout(SparseModule): ...@@ -61,7 +61,7 @@ class BatchwiseDropout(SparseModule):
if self.inplace: if self.inplace:
self.gradInput = gradOutput self.gradInput = gradOutput
typed_fn(input, 'BatchwiseMultiplicativeDropout_updateGradInput')( typed_fn(input.features, 'BatchwiseMultiplicativeDropout_updateGradInput')(
input.features, input.features,
self.gradInput, self.gradInput,
gradOutput, gradOutput,
......
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