Commit ca8ef7c0 authored by traveller59's avatar traveller59
Browse files

fix SparseInverseConv bugs

parent b9bc9d4e
......@@ -93,7 +93,7 @@ class ExampleNet(nn.Module):
super().__init__()
self.net = spconv.SparseSequential(
spconv.SparseConv3d(32, 64, 3, 2, indice_key="cp0"),
spconv.SparseInverseConv3d(64, 32, 3, 2,indice_key="cp0"),
spconv.SparseInverseConv3d(64, 32, indice_key="cp0"),
)
self.shape = shape
......
......@@ -53,7 +53,7 @@ class SparseConvolution(SparseModule):
ndim,
in_channels,
out_channels,
kernel_size,
kernel_size=3,
stride=1,
padding=0,
dilation=1,
......
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