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:
return 'in'
else:
return 'norm'
return 'norm_layer'
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:
pass
assert infer_norm_abbr(FancyNorm) == 'norm'
assert infer_norm_abbr(FancyNorm) == 'norm_layer'
def test_build_norm_layer():