To make different spconv version compatible for focalsconv (open-mmla…

…b#1034) * To make different spconv version compatible for focalsconv
ajunlonglive · Jul 19, 2022 · c233477 · c233477
1 parent 8c6e889
commit c233477
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pcdet/models/backbones_3d/focal_sparse_conv/focal_sparse_conv.py b/pcdet/models/backbones_3d/focal_sparse_conv/focal_sparse_conv.py
@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-import spconv.pytorch as spconv
+from pcdet.utils.spconv_utils import spconv
 from pcdet.ops.roiaware_pool3d.roiaware_pool3d_utils import points_in_boxes_gpu
 from pcdet.models.backbones_3d.focal_sparse_conv.focal_sparse_utils import split_voxels, check_repeat, FocalLoss
 from pcdet.utils import common_utils

diff --git a/pcdet/models/backbones_3d/spconv_backbone_focal.py b/pcdet/models/backbones_3d/spconv_backbone_focal.py
@@ -1,7 +1,7 @@
 from functools import partial
 
 import torch
-import spconv.pytorch as spconv
+from pcdet.utils.spconv_utils import spconv
 import torch.nn as nn
 
 from .focal_sparse_conv.focal_sparse_conv import FocalSparseConv