diff --git a/mmdeploy/codebase/mmcls/models/backbones/shufflenet_v2.py b/mmdeploy/codebase/mmcls/models/backbones/shufflenet_v2.py index 2d26318af..80c7a4f20 100644 --- a/mmdeploy/codebase/mmcls/models/backbones/shufflenet_v2.py +++ b/mmdeploy/codebase/mmcls/models/backbones/shufflenet_v2.py @@ -1,6 +1,5 @@ # Copyright (c) OpenMMLab. All rights reserved. import torch -from mmcls.models.utils import channel_shuffle from mmdeploy.core import FUNCTION_REWRITER from mmdeploy.utils import Backend @@ -29,6 +28,7 @@ def shufflenetv2_backbone__forward__ncnn(ctx, self, x): out (Tensor): A feature map output from InvertedResidual. The tensor shape (N, Cout, H, W). """ + from mmcls.models.utils import channel_shuffle if self.stride > 1: out = torch.cat((self.branch1(x), self.branch2(x)), dim=1) else: diff --git a/mmdeploy/codebase/mmcls/models/backbones/vision_transformer.py b/mmdeploy/codebase/mmcls/models/backbones/vision_transformer.py index 21d99aa27..a31853912 100644 --- a/mmdeploy/codebase/mmcls/models/backbones/vision_transformer.py +++ b/mmdeploy/codebase/mmcls/models/backbones/vision_transformer.py @@ -1,6 +1,5 @@ # Copyright (c) OpenMMLab. All rights reserved. import torch -from mmcls.models.utils import resize_pos_embed from mmdeploy.core import FUNCTION_REWRITER from mmdeploy.utils import Backend @@ -25,6 +24,7 @@ def visiontransformer__forward__ncnn(ctx, self, x): out (Tensor): A feature map output from InvertedResidual. The tensor shape (N, Cout, H, W). """ + from mmcls.models.utils import resize_pos_embed B = x.shape[0] x, patch_resolution = self.patch_embed(x)