From 8aaecd38a156b0dc5cb58bb171cb0dac48e26f0a Mon Sep 17 00:00:00 2001 From: xiexinch Date: Tue, 15 Nov 2022 11:11:28 +0800 Subject: [PATCH] fix swin load statedict --- mmseg/models/backbones/swin.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmseg/models/backbones/swin.py b/mmseg/models/backbones/swin.py index 57ab99085a..c0ace3c139 100644 --- a/mmseg/models/backbones/swin.py +++ b/mmseg/models/backbones/swin.py @@ -13,7 +13,7 @@ from mmengine.model import BaseModule, ModuleList from mmengine.model.weight_init import (constant_init, trunc_normal_, trunc_normal_init) -from mmengine.runner import CheckpointLoader, load_state_dict +from mmengine.runner import CheckpointLoader from mmengine.utils import to_2tuple from mmseg.registry import MODELS @@ -732,7 +732,7 @@ def init_weights(self): nH2, L2).permute(1, 0).contiguous() # load state_dict - load_state_dict(self, state_dict, strict=False, logger=None) + self.load_state_dict(state_dict, strict=False) def forward(self, x): x, hw_shape = self.patch_embed(x)