66import torch .nn as nn
77from mmcv .cnn import build_norm_layer
88from mmcv .cnn .bricks import DropPath
9- from mmcv .cnn .utils .weight_init import (constant_init , normal_init ,
10- trunc_normal_init )
11- from mmcv .runner import BaseModule
9+ from mmengine .model import (BaseModule , constant_init , normal_init ,
10+ trunc_normal_init )
1211from torch .nn .modules .utils import _pair as to_2tuple
1312
1413from mmrotate .registry import MODELS
@@ -27,7 +26,7 @@ class Mlp(BaseModule):
2726 Default: dict(type='GELU').
2827 drop (float, optional): Probability of an element to be
2928 zeroed in FFN. Default 0.0.
30- init_cfg (obj:`mmcv .ConfigDict`): The Config for initialization.
29+ init_cfg (obj:`mmengine .ConfigDict`): The Config for initialization.
3130 Default: None.
3231 """
3332
@@ -62,7 +61,7 @@ class LSKmodule(BaseModule):
6261
6362 Args:
6463 dim (int): Number of input channels.
65- init_cfg (obj:`mmcv .ConfigDict`): The Config for initialization.
64+ init_cfg (obj:`mmengine .ConfigDict`): The Config for initialization.
6665 Default: None.
6766 """
6867
@@ -99,7 +98,7 @@ class Attention(BaseModule):
9998
10099 Args:
101100 d_model (int): Number of input channels.
102- init_cfg (obj:`mmcv .ConfigDict`): The Config for initialization.
101+ init_cfg (obj:`mmengine .ConfigDict`): The Config for initialization.
103102 Default: None.
104103 """
105104
@@ -132,9 +131,9 @@ class Block(BaseModule):
132131 drop_path (float): Stochastic depth rate. Defaults to 0.1.
133132 act_layer (dict, optional): The activation config for FFNs.
134133 Default: dict(type='GELU').
135- norm_cfg (obj:`mmcv .ConfigDict`): The Config for normalization.
134+ norm_cfg (obj:`mmengine .ConfigDict`): The Config for normalization.
136135 Default: None.
137- init_cfg (obj:`mmcv .ConfigDict`): The Config for initialization.
136+ init_cfg (obj:`mmengine .ConfigDict`): The Config for initialization.
138137 Default: None.
139138 """
140139
@@ -186,9 +185,9 @@ class OverlapPatchEmbed(BaseModule):
186185 stride (int): OverlapPatchEmbed stride. Defaults to 4
187186 in_chans (int): Number of input channels. Defaults to 3.
188187 embed_dim (int): The hidden dimension of OverlapPatchEmbed.
189- norm_cfg (obj:`mmcv .ConfigDict`): The Config for normalization.
188+ norm_cfg (obj:`mmengine .ConfigDict`): The Config for normalization.
190189 Default: None.
191- init_cfg (obj:`mmcv .ConfigDict`): The Config for initialization.
190+ init_cfg (obj:`mmengine .ConfigDict`): The Config for initialization.
192191 Default: None.
193192 """
194193
0 commit comments