diff --git a/mmseg/models/backbones/stdc.py b/mmseg/models/backbones/stdc.py index ece7da172..fefaecbdf 100644 --- a/mmseg/models/backbones/stdc.py +++ b/mmseg/models/backbones/stdc.py @@ -4,7 +4,7 @@ import torch import torch.nn as nn import torch.nn.functional as F from mmcv.cnn import ConvModule -from mmcv.runner.base_module import BaseModule, ModuleList, Sequential +from mmengine.model import BaseModule, ModuleList, Sequential from mmseg.ops import resize from mmseg.registry import MODELS diff --git a/mmseg/models/backbones/swin.py b/mmseg/models/backbones/swin.py index ca8a71f0d..8d4aad657 100644 --- a/mmseg/models/backbones/swin.py +++ b/mmseg/models/backbones/swin.py @@ -11,10 +11,10 @@ from mmcv.cnn import build_norm_layer from mmcv.cnn.bricks.transformer import FFN, build_dropout from mmcv.cnn.utils.weight_init import (constant_init, trunc_normal_, trunc_normal_init) -from mmcv.runner import (BaseModule, CheckpointLoader, ModuleList, - load_state_dict) +from mmcv.runner import CheckpointLoader, load_state_dict from mmcv.utils import to_2tuple from mmengine.logging import print_log +from mmengine.model import BaseModule, ModuleList from mmseg.registry import MODELS from ..utils.embed import PatchEmbed, PatchMerging diff --git a/mmseg/models/backbones/vit.py b/mmseg/models/backbones/vit.py index 7757d5064..3707ff529 100644 --- a/mmseg/models/backbones/vit.py +++ b/mmseg/models/backbones/vit.py @@ -9,9 +9,9 @@ from mmcv.cnn import build_norm_layer from mmcv.cnn.bricks.transformer import FFN, MultiheadAttention from mmcv.cnn.utils.weight_init import (constant_init, kaiming_init, trunc_normal_) -from mmcv.runner import (BaseModule, CheckpointLoader, ModuleList, - load_state_dict) +from mmcv.runner import CheckpointLoader, load_state_dict from mmengine.logging import print_log +from mmengine.model import BaseModule, ModuleList from torch.nn.modules.batchnorm import _BatchNorm from torch.nn.modules.utils import _pair as to_2tuple diff --git a/mmseg/models/utils/embed.py b/mmseg/models/utils/embed.py index 1515675e1..d29183e78 100644 --- a/mmseg/models/utils/embed.py +++ b/mmseg/models/utils/embed.py @@ -5,8 +5,8 @@ from typing import Sequence import torch.nn as nn import torch.nn.functional as F from mmcv.cnn import build_conv_layer, build_norm_layer -from mmcv.runner.base_module import BaseModule from mmcv.utils import to_2tuple +from mmengine.model import BaseModule class AdaptivePadding(nn.Module):