diff --git a/projects/LSKNet/lsknet/lsknet.py b/projects/LSKNet/lsknet/lsknet.py index 655b98b28..85552c023 100644 --- a/projects/LSKNet/lsknet/lsknet.py +++ b/projects/LSKNet/lsknet/lsknet.py @@ -6,9 +6,8 @@ import torch.nn as nn from mmcv.cnn import build_norm_layer from mmcv.cnn.bricks import DropPath -from mmcv.cnn.utils.weight_init import (constant_init, normal_init, - trunc_normal_init) -from mmcv.runner import BaseModule +from mmengine.model import (BaseModule, constant_init, normal_init, + trunc_normal_init) from torch.nn.modules.utils import _pair as to_2tuple from mmrotate.registry import MODELS @@ -27,7 +26,7 @@ class Mlp(BaseModule): Default: dict(type='GELU'). drop (float, optional): Probability of an element to be zeroed in FFN. Default 0.0. - init_cfg (obj:`mmcv.ConfigDict`): The Config for initialization. + init_cfg (obj:`mmengine.ConfigDict`): The Config for initialization. Default: None. """ @@ -62,7 +61,7 @@ class LSKmodule(BaseModule): Args: dim (int): Number of input channels. - init_cfg (obj:`mmcv.ConfigDict`): The Config for initialization. + init_cfg (obj:`mmengine.ConfigDict`): The Config for initialization. Default: None. """ @@ -99,7 +98,7 @@ class Attention(BaseModule): Args: d_model (int): Number of input channels. - init_cfg (obj:`mmcv.ConfigDict`): The Config for initialization. + init_cfg (obj:`mmengine.ConfigDict`): The Config for initialization. Default: None. """ @@ -132,9 +131,9 @@ class Block(BaseModule): drop_path (float): Stochastic depth rate. Defaults to 0.1. act_layer (dict, optional): The activation config for FFNs. Default: dict(type='GELU'). - norm_cfg (obj:`mmcv.ConfigDict`): The Config for normalization. + norm_cfg (obj:`mmengine.ConfigDict`): The Config for normalization. Default: None. - init_cfg (obj:`mmcv.ConfigDict`): The Config for initialization. + init_cfg (obj:`mmengine.ConfigDict`): The Config for initialization. Default: None. """ @@ -186,9 +185,9 @@ class OverlapPatchEmbed(BaseModule): stride (int): OverlapPatchEmbed stride. Defaults to 4 in_chans (int): Number of input channels. Defaults to 3. embed_dim (int): The hidden dimension of OverlapPatchEmbed. - norm_cfg (obj:`mmcv.ConfigDict`): The Config for normalization. + norm_cfg (obj:`mmengine.ConfigDict`): The Config for normalization. Default: None. - init_cfg (obj:`mmcv.ConfigDict`): The Config for initialization. + init_cfg (obj:`mmengine.ConfigDict`): The Config for initialization. Default: None. """