Skip to content

GitLab

Unverified Commit 26a61a99 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

Force to use mmdet3d's spconv (#1290) 


Co-authored-by: default avatarWenhao Wu <wenhaowu.chn@gmail.com>
parent e6d0e214
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 10 deletions
mmdet3d/ops/spconv/conv.py
View file @ 26a61a99
...@@ -205,7 +205,7 @@ class SparseConvolution(SparseModule): ...@@ -205,7 +205,7 @@ class SparseConvolution(SparseModule):
return out_tensor return out_tensor
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseConv2d(SparseConvolution): class SparseConv2d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -231,7 +231,7 @@ class SparseConv2d(SparseConvolution): ...@@ -231,7 +231,7 @@ class SparseConv2d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseConv3d(SparseConvolution): class SparseConv3d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -257,7 +257,7 @@ class SparseConv3d(SparseConvolution): ...@@ -257,7 +257,7 @@ class SparseConv3d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseConv4d(SparseConvolution): class SparseConv4d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -283,7 +283,7 @@ class SparseConv4d(SparseConvolution): ...@@ -283,7 +283,7 @@ class SparseConv4d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseConvTranspose2d(SparseConvolution): class SparseConvTranspose2d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -310,7 +310,7 @@ class SparseConvTranspose2d(SparseConvolution): ...@@ -310,7 +310,7 @@ class SparseConvTranspose2d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseConvTranspose3d(SparseConvolution): class SparseConvTranspose3d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -337,7 +337,7 @@ class SparseConvTranspose3d(SparseConvolution): ...@@ -337,7 +337,7 @@ class SparseConvTranspose3d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseInverseConv2d(SparseConvolution): class SparseInverseConv2d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -356,7 +356,7 @@ class SparseInverseConv2d(SparseConvolution): ...@@ -356,7 +356,7 @@ class SparseInverseConv2d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SparseInverseConv3d(SparseConvolution): class SparseInverseConv3d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -375,7 +375,7 @@ class SparseInverseConv3d(SparseConvolution): ...@@ -375,7 +375,7 @@ class SparseInverseConv3d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SubMConv2d(SparseConvolution): class SubMConv2d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -402,7 +402,7 @@ class SubMConv2d(SparseConvolution): ...@@ -402,7 +402,7 @@ class SubMConv2d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SubMConv3d(SparseConvolution): class SubMConv3d(SparseConvolution):
def __init__(self, def __init__(self,
...@@ -429,7 +429,7 @@ class SubMConv3d(SparseConvolution): ...@@ -429,7 +429,7 @@ class SubMConv3d(SparseConvolution):
indice_key=indice_key) indice_key=indice_key)
@CONV_LAYERS.register_module() @CONV_LAYERS.register_module(force=True)
class SubMConv4d(SparseConvolution): class SubMConv4d(SparseConvolution):
def __init__(self, def __init__(self,
......
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