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

Fix Sparsify

parent bb5de2f5
......@@ -28,8 +28,7 @@ class Sparsify(Module):
active = input.features[:,0]>0
output.features=input.features[active]
active=active.type('torch.LongTensor')
dim_fn(self.dimension, 'sparsifyMetadata')(
input.metadata,
input.metadata.sparsifyMetadata(
output.metadata,
input.spatial_size,
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