To fix VoxelNeXt training bugs in spconv 2.2 / 2.3 (open-mmlab#1313)

* VoxelNeXt * Update .gitignore * Update voxelnext_head.py * Update spconv_utils.py
xq198109 · Apr 9, 2023 · a68aaa6 · a68aaa6
1 parent 53c259f
commit a68aaa6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pcdet/models/dense_heads/voxelnext_head.py b/pcdet/models/dense_heads/voxelnext_head.py
@@ -5,7 +5,7 @@
 from ..model_utils import centernet_utils
 from ..model_utils import model_nms_utils
 from ...utils import loss_utils
-import spconv.pytorch as spconv
+from ...utils.spconv_utils import replace_feature, spconv
 import copy
 from easydict import EasyDict
 

diff --git a/pcdet/utils/spconv_utils.py b/pcdet/utils/spconv_utils.py
@@ -1,5 +1,9 @@
 from typing import Set
 
+import spconv
+if float(spconv.__version__[2:]) >= 2.2:
+    spconv.constants.SPCONV_USE_DIRECT_TABLE = False
+
 try:
     import spconv.pytorch as spconv
 except: