fix batch norm. test=develop (#15597)

6 years ago · 46a6cac91f
parent 943d972878
commit 46a6cac91f
1 changed files with 4 additions and 2 deletions
--- a/paddle/fluid/operators/batch_norm_op.cc
+++ b/paddle/fluid/operators/batch_norm_op.cc
@ -589,8 +589,10 @@ class BatchNormGradMaker : public framework::SingleGradOpDescMaker {
    op->SetInput("SavedVariance", Output("SavedVariance"));

    // used when setting use_global_stats True during training
+    if (boost::get<bool>(GetAttr("use_global_stats"))) {
      op->SetInput("Mean", Output("MeanOut"));
      op->SetInput("Variance", Output("VarianceOut"));
+    }

    op->SetAttrMap(Attrs());