Unverified Commit a68aaa65 authored by yukang's avatar yukang Committed by GitHub
Browse files

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

* VoxelNeXt

* Update .gitignore

* Update voxelnext_head.py

* Update spconv_utils.py
parent 53c259f4
......@@ -5,7 +5,7 @@ from torch.nn.init import kaiming_normal_
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
......
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:
......
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