From 2787041246f0e4df1eb8b4c3c64a6e4a7e5a26d1 Mon Sep 17 00:00:00 2001 From: songyouwei Date: Wed, 25 Mar 2020 13:45:47 +0800 Subject: [PATCH] bug fix (#23193) test=develop --- python/paddle/fluid/dygraph/base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/dygraph/base.py b/python/paddle/fluid/dygraph/base.py index 0bf620ea217..22b95dfd5f2 100644 --- a/python/paddle/fluid/dygraph/base.py +++ b/python/paddle/fluid/dygraph/base.py @@ -105,8 +105,9 @@ def enable_dygraph(place=None): print(fluid.in_dygraph_mode()) # False """ global _functional_dygraph_context_manager - _functional_dygraph_context_manager = guard(place=place) - _functional_dygraph_context_manager.__enter__() + if _functional_dygraph_context_manager is None: + _functional_dygraph_context_manager = guard(place=place) + _functional_dygraph_context_manager.__enter__() def disable_dygraph(): -- GitLab