diff --git a/paddlex/cv/models/load_model.py b/paddlex/cv/models/load_model.py index afccc44506079eea4b6043610dedefb0a8be5334..71fdf4c1d0fe5e8d0b1f90ac0142efa0a4b4b3d6 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)