mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Import BaseModule from mmengien (#1862)

Browse Source
This commit is contained in:
Miao Zheng 2022-08-04 09:35:05 +08:00 committed by GitHub
parent fdaf8ea7d6
commit e2f26e3fde
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/models/backbones/stdc.py
View File

@ -4,7 +4,7 @@ import torch
import torch.nn as nn
import torch.nn.functional as F
from mmcv.cnn import ConvModule
from mmcv.runner.base_module import BaseModule, ModuleList, Sequential
from mmengine.model import BaseModule, ModuleList, Sequential
from mmseg.ops import resize
from mmseg.registry import MODELS

4
mmseg/models/backbones/swin.py
View File

@ -11,10 +11,10 @@ from mmcv.cnn import build_norm_layer
from mmcv.cnn.bricks.transformer import FFN, build_dropout
from mmcv.cnn.utils.weight_init import (constant_init, trunc_normal_,
trunc_normal_init)
from mmcv.runner import (BaseModule, CheckpointLoader, ModuleList,
load_state_dict)
from mmcv.runner import CheckpointLoader, load_state_dict
from mmcv.utils import to_2tuple
from mmengine.logging import print_log
from mmengine.model import BaseModule, ModuleList
from mmseg.registry import MODELS
from ..utils.embed import PatchEmbed, PatchMerging

4
mmseg/models/backbones/vit.py
View File

@ -9,9 +9,9 @@ from mmcv.cnn import build_norm_layer
from mmcv.cnn.bricks.transformer import FFN, MultiheadAttention
from mmcv.cnn.utils.weight_init import (constant_init, kaiming_init,
trunc_normal_)
from mmcv.runner import (BaseModule, CheckpointLoader, ModuleList,
load_state_dict)
from mmcv.runner import CheckpointLoader, load_state_dict
from mmengine.logging import print_log
from mmengine.model import BaseModule, ModuleList
from torch.nn.modules.batchnorm import _BatchNorm
from torch.nn.modules.utils import _pair as to_2tuple

2
mmseg/models/utils/embed.py
View File

@ -5,8 +5,8 @@ from typing import Sequence
import torch.nn as nn
import torch.nn.functional as F
from mmcv.cnn import build_conv_layer, build_norm_layer
from mmcv.runner.base_module import BaseModule
from mmcv.utils import to_2tuple
from mmengine.model import BaseModule
class AdaptivePadding(nn.Module):