p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Merge pull request !2407 from chenzhongming/r0.3
This commit is contained in:
mindspore-ci-bot 2020-06-20 15:32:19 +08:00 committed by Gitee
parent 11caa3aec8 2fab2492bc
commit e519317622
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/quant.py
View File

@ -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):