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

[Bug] Avoid recursion error when using customized norm layer without _abbr_ (#732)

Browse Source 
* avoid recursion error when using customized norm layer without _abbr_

* fix unittest bug
This commit is contained in:
Rui Xu 2020-12-22 10:19:52 +08:00 committed by GitHub
parent adc59b9f28
commit 826d3a7b68
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcv/cnn/bricks/norm.py
View File

@ -65,7 +65,7 @@ def infer_abbr(class_type):
elif 'instance' in class_name: elif 'instance' in class_name:
return 'in' return 'in'
else: else:
return 'norm' return 'norm_layer'
def build_norm_layer(cfg, num_features, postfix=''): def build_norm_layer(cfg, num_features, postfix=''):

2
tests/test_cnn/test_build_layers.py
View File

@ -101,7 +101,7 @@ def test_infer_norm_abbr():
class FancyNorm: class FancyNorm:
pass pass
assert infer_norm_abbr(FancyNorm) == 'norm' assert infer_norm_abbr(FancyNorm) == 'norm_layer'
def test_build_norm_layer(): def test_build_norm_layer():