未验证 提交 46a6cac9 编写于 作者: D dzhwinter 提交者: GitHub

fix batch norm. test=develop (#15597)

上级 943d9728
...@@ -589,8 +589,10 @@ class BatchNormGradMaker : public framework::SingleGradOpDescMaker { ...@@ -589,8 +589,10 @@ class BatchNormGradMaker : public framework::SingleGradOpDescMaker {
op->SetInput("SavedVariance", Output("SavedVariance")); op->SetInput("SavedVariance", Output("SavedVariance"));
// used when setting use_global_stats True during training // used when setting use_global_stats True during training
if (boost::get<bool>(GetAttr("use_global_stats"))) {
op->SetInput("Mean", Output("MeanOut")); op->SetInput("Mean", Output("MeanOut"));
op->SetInput("Variance", Output("VarianceOut")); op->SetInput("Variance", Output("VarianceOut"));
}
op->SetAttrMap(Attrs()); op->SetAttrMap(Attrs());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册