From eea40f49c792e10137ff722b638a2b9b271f9e60 Mon Sep 17 00:00:00 2001 From: xixiaoyao Date: Mon, 28 Oct 2019 17:40:48 +0800 Subject: [PATCH] fix bugs --- paddlepalm/utils/saver.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddlepalm/utils/saver.py b/paddlepalm/utils/saver.py index 95812b6..9277f0d 100644 --- a/paddlepalm/utils/saver.py +++ b/paddlepalm/utils/saver.py @@ -57,14 +57,15 @@ def init_pretraining_params(exe, with tarfile.open(os.path.join(pretraining_params_path, '__palmmodel__'), 'r:') as f: f.extractall(os.path.join(pretraining_params_path, '.temp')) - + + log_path = os.path.join(pretraining_params_path, '__palmmodel__') pretraining_params_path = os.path.join(pretraining_params_path, '.temp') def existed_params(var): if not isinstance(var, fluid.framework.Parameter): return False if not os.path.exists(os.path.join(pretraining_params_path, var.name)): - print('Warning: {} not found in {}.'.format(var.name, pretraining_params_path)) + print('Warning: {} not found in {}.'.format(var.name, log_path)) return os.path.exists(os.path.join(pretraining_params_path, var.name)) fluid.io.load_vars( -- GitLab