Skip to content

Commit b55be23

Browse files
authored
[Fix] fix LSKNet error in import (#839)
* fix lsknet * fix some comment
1 parent e3c87de commit b55be23

1 file changed

Lines changed: 9 additions & 10 deletions

File tree

projects/LSKNet/lsknet/lsknet.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,8 @@
66
import torch.nn as nn
77
from mmcv.cnn import build_norm_layer
88
from 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)
1211
from torch.nn.modules.utils import _pair as to_2tuple
1312

1413
from 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

Comments
 (0)