We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 943d972 commit 46a6cacCopy full SHA for 46a6cac
paddle/fluid/operators/batch_norm_op.cc
@@ -589,8 +589,10 @@ class BatchNormGradMaker : public framework::SingleGradOpDescMaker {
589
op->SetInput("SavedVariance", Output("SavedVariance"));
590
591
// used when setting use_global_stats True during training
592
- op->SetInput("Mean", Output("MeanOut"));
593
- op->SetInput("Variance", Output("VarianceOut"));
+ if (boost::get<bool>(GetAttr("use_global_stats"))) {
+ op->SetInput("Mean", Output("MeanOut"));
594
+ op->SetInput("Variance", Output("VarianceOut"));
595
+ }
596
597
op->SetAttrMap(Attrs());
598
0 commit comments