From 2fab2492bcd69941d662a18272ab17cddb79a556 Mon Sep 17 00:00:00 2001 From: chenzomi Date: Sat, 20 Jun 2020 15:22:35 +0800 Subject: [PATCH] change Q.BNTrainingReduce() to P.BNTrainingReduce() --- mindspore/nn/layer/quant.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/quant.py b/mindspore/nn/layer/quant.py index e7e41670a..26d56689f 100644 --- a/mindspore/nn/layer/quant.py +++ b/mindspore/nn/layer/quant.py @@ -79,7 +79,7 @@ class BatchNormFoldCell(Cell): 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) 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) def construct(self, x, mean, variance, global_step): -- GitLab