fix bug

hustvl · Sep 1, 2023 · e1e6d64 · e1e6d64
1 parent 695319f
commit e1e6d64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/projects/mmdet3d_plugin/maptr/detectors/maptrv2.py b/projects/mmdet3d_plugin/maptr/detectors/maptrv2.py
@@ -9,7 +9,7 @@
 from mmcv.runner import force_fp32, auto_fp16
 from mmdet3d.ops import Voxelization, DynamicScatter
 from mmdet3d.models import builder
-
+from mmcv.utils import TORCH_VERSION, digit_version
 @DETECTORS.register_module()
 class MapTRv2(MVXTwoStageDetector):
     """MapTR.