diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index 0bf0e7c3e919bb6219f8ee7f99eff6c47656bec4..d899363cbaef9da703f520b10b7e107e8706cb57 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -272,11 +272,15 @@ def name_scope(prefix=None): g = f - 1 """ # TODO(panyx0718): Only [0-9a-z]. - assert prefix, "namescope prefix cannot be empty." - global _name_scope - _name_scope = _name_scope.child(prefix) - yield - _name_scope = _name_scope.parent() + # in dygraph we don't need namescope since it will cause mem leak + if not in_dygraph_mode(): + assert prefix, "namescope prefix cannot be empty." + global _name_scope + _name_scope = _name_scope.child(prefix) + yield + _name_scope = _name_scope.parent() + else: + yield def _full_name_scope():