From 7d245680c7ada27f9165fd499c7b23c20b44e85a Mon Sep 17 00:00:00 2001 From: Yanyi Liu Date: Fri, 5 May 2023 06:03:55 +0000 Subject: [PATCH 1/2] fix lsknet --- projects/LSKNet/lsknet/lsknet.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/projects/LSKNet/lsknet/lsknet.py b/projects/LSKNet/lsknet/lsknet.py index 655b98b28..c42b01012 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 From ad0e09fe2f66bd208e8a32e6facb14280e493e93 Mon Sep 17 00:00:00 2001 From: Yanyi Liu Date: Fri, 5 May 2023 06:07:51 +0000 Subject: [PATCH 2/2] fix some comment --- projects/LSKNet/lsknet/lsknet.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/projects/LSKNet/lsknet/lsknet.py b/projects/LSKNet/lsknet/lsknet.py index c42b01012..85552c023 100644 --- a/projects/LSKNet/lsknet/lsknet.py +++ b/projects/LSKNet/lsknet/lsknet.py @@ -26,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. """ @@ -61,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. """ @@ -98,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. """ @@ -131,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. """ @@ -185,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. """