提交 a974e65b 编写于 作者: Z zhaojichen

fix batchnorm bug

上级 6d3709eb
......@@ -376,7 +376,7 @@ class GlobalBatchNorm(_BatchNorm):
use_batch_statistics,
group)
self.group = check_int_positive(group)
if self.group <=1:
if self.group <= 1:
raise ValueError("the number of group must be greater than 1.")
def _check_data_dim(self, x):
if x.dim == 0:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册