提交 030a3db2 编写于 作者: X xzl

the groups default should be None

上级 ccd46d1b
...@@ -2269,7 +2269,7 @@ def img_depthwise_conv_layer(input, ...@@ -2269,7 +2269,7 @@ def img_depthwise_conv_layer(input,
name=None, name=None,
num_channels=None, num_channels=None,
act=None, act=None,
groups=1, groups=None,
stride=1, stride=1,
padding=0, padding=0,
bias_attr=None, bias_attr=None,
...@@ -2286,6 +2286,8 @@ def img_depthwise_conv_layer(input, ...@@ -2286,6 +2286,8 @@ def img_depthwise_conv_layer(input,
assert input.num_filters is not None assert input.num_filters is not None
num_channels = input.num_filters num_channels = input.num_filters
groups = num_channels
if filter_size_y is None: if filter_size_y is None:
if isinstance(filter_size, collections.Sequence): if isinstance(filter_size, collections.Sequence):
assert len(filter_size) == 2 assert len(filter_size) == 2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册