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

[Enhance] add deleting params info in swinv2. (#1142)

Browse Source
This commit is contained in:
Hubert 2022-11-14 17:07:21 +08:00 committed by GitHub
parent 1b98fc13d9
commit 992d13e772
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmcls/models/backbones/swin_transformer_v2.py
View File

@ -546,6 +546,13 @@ class SwinTransformerV2(BaseBackbone):
def _delete_reinit_params(self, state_dict, prefix, *args, **kwargs):
# delete relative_position_index since we always re-init it
from mmengine.logging import MMLogger
logger = MMLogger.get_current_instance()
logger.info(
'Delete `relative_position_index` and `relative_coords_table` '
'since we always re-init these params according to the '
'`window_size`, which might cause unwanted but unworried '
'warnings when loading checkpoint.')
relative_position_index_keys = [
k for k in state_dict.keys() if 'relative_position_index' in k
]