diff --git a/python/paddle/v2/fluid/clip.py b/python/paddle/v2/fluid/clip.py index f6ff83924f2516875f44b8b49ac75e36d7d1d407..9800ad7c5d024b21e564762c8079d40968cd97d2 100644 --- a/python/paddle/v2/fluid/clip.py +++ b/python/paddle/v2/fluid/clip.py @@ -124,11 +124,11 @@ class GradientClipByGlobalNorm(BaseGradientClipAttr): cls.global_norm_var = layers.fill_constant( shape=[1], dtype="float32", value=0.0) - cls.local_norm_var = framework.default_main_program().current_block( - ).create_var( - name=framework.unique_name("local_norm"), - dtype="float32", - persistable=False) + cls.local_norm_var = framework.default_main_program().block( + 0).create_var( + name=framework.unique_name("local_norm"), + dtype="float32", + persistable=False) cls.clip_norm_var = layers.fill_constant( shape=[1], dtype="float32", value=clip_norm)