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

fix initialization bug of override cfg (#918)

Browse Source
This commit is contained in:
Wenwei Zhang 2021-04-02 09:43:16 +08:00 committed by GitHub
parent 9d80f56ae8
commit 3ae1b257fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mmcv/cnn/utils/weight_init.py
View File

@ -1,4 +1,5 @@
# Copyright (c) Open-MMLab. All rights reserved. # Copyright (c) Open-MMLab. All rights reserved.
import copy
import warnings import warnings
import numpy as np import numpy as np
@ -445,12 +446,17 @@ def initialize(module, init_cfg):
init_cfg = [init_cfg] init_cfg = [init_cfg]
for cfg in init_cfg: for cfg in init_cfg:
override = cfg.pop('override', None) # should deeply copy the original config because cfg may be used by
_initialize(module, cfg) # other modules, e.g., one init_cfg shared by multiple bottleneck
# blocks, the expected cfg will be changed after pop and will change
# the initialization behavior of other modules
cp_cfg = copy.deepcopy(cfg)
override = cp_cfg.pop('override', None)
_initialize(module, cp_cfg)
if override is not None: if override is not None:
cfg.pop('layer', None) cp_cfg.pop('layer', None)
_initialize_override(module, override, cfg) _initialize_override(module, override, cp_cfg)
else: else:
# All attributes in module have same initialization. # All attributes in module have same initialization.
pass pass