Skip to content

GitLab

Unverified Commit 14c3e175 authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Fix] num_channels can not be divisible by num_groups in test_build_norm_layer (#2085)

parent c03bba00
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_cnn/test_build_layers.py
View file @ 14c3e175
...@@ -160,7 +160,7 @@ def test_build_norm_layer(): ...@@ -160,7 +160,7 @@ def test_build_norm_layer():
for postfix in ['_test', 1]: for postfix in ['_test', 1]:
cfg = dict(type=type_name) cfg = dict(type=type_name)
if type_name == 'GN': if type_name == 'GN':
cfg['num_groups'] = 2 cfg['num_groups'] = 3
name, layer = build_norm_layer(cfg, 3, postfix=postfix) name, layer = build_norm_layer(cfg, 3, postfix=postfix)
assert name == abbr_mapping[type_name] + str(postfix) assert name == abbr_mapping[type_name] + str(postfix)
assert isinstance(layer, module) assert isinstance(layer, module)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment