Skip to content

GitLab

Commit ca8ef7c0 authored by traveller59's avatar traveller59
Browse files

fix SparseInverseConv bugs

parent b9bc9d4e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ ca8ef7c0
...@@ -93,7 +93,7 @@ class ExampleNet(nn.Module): ...@@ -93,7 +93,7 @@ class ExampleNet(nn.Module):
super().__init__() super().__init__()
self.net = spconv.SparseSequential( self.net = spconv.SparseSequential(
spconv.SparseConv3d(32, 64, 3, 2, indice_key="cp0"), 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 self.shape = shape
......
spconv/conv.py
View file @ ca8ef7c0
...@@ -53,7 +53,7 @@ class SparseConvolution(SparseModule): ...@@ -53,7 +53,7 @@ class SparseConvolution(SparseModule):
ndim, ndim,
in_channels, in_channels,
out_channels, out_channels,
kernel_size, kernel_size=3,
stride=1, stride=1,
padding=0, padding=0,
dilation=1, 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