未验证 提交 11f5848b 编写于 作者: W Weilong Wu 提交者: GitHub

polish batch_norm api (#49508)

上级 1168a178
......@@ -20,7 +20,6 @@ import paddle.fluid as fluid
from paddle import _C_ops, in_dynamic_mode
from paddle.fluid.framework import in_dygraph_mode
from ...fluid import dygraph_utils
from ...fluid.data_feeder import check_type, check_variable_and_dtype
from ...fluid.layer_helper import LayerHelper
......@@ -210,10 +209,7 @@ def batch_norm(
use_global_stats,
trainable_statistics,
)
return dygraph_utils._append_activation_in_dygraph(
batch_norm_out, act=None
)
return batch_norm_out
else:
check_variable_and_dtype(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册