Skip to content

Commit a68aaa6

Browse files
authored
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
1 parent 53c259f commit a68aaa6

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

pcdet/models/dense_heads/voxelnext_head.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from ..model_utils import centernet_utils
66
from ..model_utils import model_nms_utils
77
from ...utils import loss_utils
8-
import spconv.pytorch as spconv
8+
from ...utils.spconv_utils import replace_feature, spconv
99
import copy
1010
from easydict import EasyDict
1111

pcdet/utils/spconv_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,9 @@
11
from typing import Set
22

3+
import spconv
4+
if float(spconv.__version__[2:]) >= 2.2:
5+
spconv.constants.SPCONV_USE_DIRECT_TABLE = False
6+
37
try:
48
import spconv.pytorch as spconv
59
except:

0 commit comments

Comments
 (0)