fix batch norm. test=develop (#15597)

46a6cac9 · dzhwinter · GitHub · 943d9728 · 46a6cac9
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

paddle/fluid/operators/batch_norm_op.cc paddle/fluid/operators/batch_norm_op.cc +4 -2

未找到文件。
--- 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
-    op->SetInput("Mean", Output("MeanOut"));
-    op->SetInput("Variance", Output("VarianceOut"));
+    if (boost::get<bool>(GetAttr("use_global_stats"))) {
+      op->SetInput("Mean", Output("MeanOut"));
+      op->SetInput("Variance", Output("VarianceOut"));
+    }

    op->SetAttrMap(Attrs());