提交 3872ea05 编写于 作者: C chengduoZH

fix bug (conv2d groups )

上级 6da78d97
...@@ -661,7 +661,7 @@ def conv2d(input, ...@@ -661,7 +661,7 @@ def conv2d(input,
if groups is None: if groups is None:
num_filter_channels = num_channels num_filter_channels = num_channels
else: else:
if num_channels % groups is not 0: if num_channels % groups != 0:
raise ValueError("num_channels must be divisible by groups.") raise ValueError("num_channels must be divisible by groups.")
num_filter_channels = num_channels / groups num_filter_channels = num_channels / groups
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册