From 7d04d019779bf1fc53b7031961324ab0a609fa96 Mon Sep 17 00:00:00 2001 From: SigureMo Date: Mon, 5 Sep 2022 21:38:14 +0800 Subject: [PATCH] add a wrap_decorator decorator to keep the __doc__ attr --- python/paddle/fluid/dygraph/jit.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/fluid/dygraph/jit.py b/python/paddle/fluid/dygraph/jit.py index f96cbb82a7359..856a21881c233 100644 --- a/python/paddle/fluid/dygraph/jit.py +++ b/python/paddle/fluid/dygraph/jit.py @@ -634,6 +634,7 @@ def _remove_save_pre_hook(hook): _save_pre_hooks_lock.release() +@wrap_decorator def _run_save_pre_hooks(func): def wrapper(layer, path, input_spec=None, **configs):