Skip to content

Commit

Permalink
polish batch_norm api (#49508)
Browse files Browse the repository at this point in the history
  • Loading branch information
veyron95 authored Jan 5, 2023
1 parent 1168a17 commit 11f5848
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions python/paddle/nn/functional/norm.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
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

Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit 11f5848

Please sign in to comment.