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):
self.output.metadata = input.metadata
self.output.spatialSize = input.spatialSize
typed_fn(input, 'BatchwiseMultiplicativeDropout_updateOutput')(
typed_fn(input.features, 'BatchwiseMultiplicativeDropout_updateOutput')(
input.features,
self.output.features,
self.noise,
......@@ -61,7 +61,7 @@ class BatchwiseDropout(SparseModule):
if self.inplace:
self.gradInput = gradOutput
typed_fn(input, 'BatchwiseMultiplicativeDropout_updateGradInput')(
typed_fn(input.features, 'BatchwiseMultiplicativeDropout_updateGradInput')(
input.features,
self.gradInput,
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