From 2f34f935c02dcb599f94c61ba1bf2c63a897868d Mon Sep 17 00:00:00 2001 From: wangxiao Date: Fri, 29 Nov 2019 18:17:02 +0800 Subject: [PATCH] update saver.py --- paddlepalm/utils/saver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlepalm/utils/saver.py b/paddlepalm/utils/saver.py index 9277f0d..2ec4c9f 100644 --- a/paddlepalm/utils/saver.py +++ b/paddlepalm/utils/saver.py @@ -55,7 +55,7 @@ def init_pretraining_params(exe, print("Loading pretraining parameters from {}...".format( pretraining_params_path)) - with tarfile.open(os.path.join(pretraining_params_path, '__palmmodel__'), 'r:') as f: + with tarfile.open(os.path.join(pretraining_params_path, '__palmmodel__'), 'r:gz') as f: f.extractall(os.path.join(pretraining_params_path, '.temp')) log_path = os.path.join(pretraining_params_path, '__palmmodel__') -- GitLab