Fix import (#861)

open-mmlab · Aug 11, 2021 · 90e6378 · 90e6378
1 parent 057ad6e
commit 90e6378
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tools/analysis_tools/benchmark.py b/tools/analysis_tools/benchmark.py
@@ -3,11 +3,10 @@
 import torch
 from mmcv import Config
 from mmcv.parallel import MMDataParallel
-from mmcv.runner import load_checkpoint
+from mmcv.runner import load_checkpoint, wrap_fp16_model
 
 from mmdet3d.datasets import build_dataloader, build_dataset
 from mmdet3d.models import build_detector
-from mmdet.core import wrap_fp16_model
 from tools.misc.fuse_conv_bn import fuse_module