Skip to content

GitLab

Unverified Commit f0407b36 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Fix Sparsify

parent bb5de2f5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
sparseconvnet/sparsify.py
View file @ f0407b36
...@@ -28,8 +28,7 @@ class Sparsify(Module): ...@@ -28,8 +28,7 @@ class Sparsify(Module):
active = input.features[:,0]>0 active = input.features[:,0]>0
output.features=input.features[active] output.features=input.features[active]
active=active.type('torch.LongTensor') active=active.type('torch.LongTensor')
dim_fn(self.dimension, 'sparsifyMetadata')( input.metadata.sparsifyMetadata(
input.metadata,
output.metadata, output.metadata,
input.spatial_size, input.spatial_size,
active.byte(), active.byte(),
......
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