From 192ab39ad65db3a0b65906971e6b4121853ce52c Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Tue, 22 Sep 2020 15:08:34 +0000 Subject: [PATCH] fix scope problem for slim --- paddlex/cv/models/load_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlex/cv/models/load_model.py b/paddlex/cv/models/load_model.py index afccc44..71fdf4c 100644 --- a/paddlex/cv/models/load_model.py +++ b/paddlex/cv/models/load_model.py @@ -69,7 +69,7 @@ def load_model(model_dir, fixed_input_shape=None): if status == "Prune": from .slim.prune import update_program model.test_prog = update_program(model.test_prog, model_dir, - model.places[0]) + model.places[0], scope=model_scope) import pickle with open(osp.join(model_dir, 'model.pdparams'), 'rb') as f: load_dict = pickle.load(f) -- GitLab