[Fix] fix LSKNet error in import (#839)

* fix lsknet * fix some comment
open-mmlab · May 8, 2023 · b55be23 · b55be23
1 parent e3c87de
commit b55be23
Showing 1 changed file with 9 additions and 10 deletions.
diff --git 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.
     """