From 7e2b43d2c1d7ac51d123b37935497a052d6db54a Mon Sep 17 00:00:00 2001 From: chenguowei01 Date: Thu, 6 Aug 2020 11:11:03 +0800 Subject: [PATCH] update utils.py --- dygraph/utils/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dygraph/utils/utils.py b/dygraph/utils/utils.py index fa995d27..3d4fc62e 100644 --- a/dygraph/utils/utils.py +++ b/dygraph/utils/utils.py @@ -88,6 +88,7 @@ def resume(model, optimizer, resume_model): if resume_model is not None: logging.info('Resume model from {}'.format(resume_model)) if os.path.exists(resume_model): + resume_model = os.path.normpath(resume_model) ckpt_path = os.path.join(resume_model, 'model') para_state_dict, opti_state_dict = fluid.load_dygraph(ckpt_path) model.set_dict(para_state_dict) -- GitLab