Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 2fab2492 编写于 作者: C chenzomi
浏览文件
操作

change Q.BNTrainingReduce() to P.BNTrainingReduce()

上级 11caa3ae
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mindspore/nn/layer/quant.py
浏览文件 @ 2fab2492
...@@ -79,7 +79,7 @@ class BatchNormFoldCell(Cell): ...@@ -79,7 +79,7 @@ class BatchNormFoldCell(Cell):
self.bn_train = Q.BatchNormFold(momentum, epsilon, is_training=True, freeze_bn=freeze_bn) self.bn_train = Q.BatchNormFold(momentum, epsilon, is_training=True, freeze_bn=freeze_bn)
self.bn_infer = Q.BatchNormFold(momentum, epsilon, is_training=False, freeze_bn=freeze_bn) self.bn_infer = Q.BatchNormFold(momentum, epsilon, is_training=False, freeze_bn=freeze_bn)
else: else:
self.bn_reduce = Q.BNTrainingReduce() self.bn_reduce = P.BNTrainingReduce()
self.bn_update = Q.BatchNormFoldD(momentum, epsilon, is_training=True, freeze_bn=freeze_bn) self.bn_update = Q.BatchNormFoldD(momentum, epsilon, is_training=True, freeze_bn=freeze_bn)
def construct(self, x, mean, variance, global_step): def construct(self, x, mean, variance, global_step):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或