diff --git a/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py b/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py index 6c902d5fb8e7725b81d9433a9cc0afe42f4ab855..7263e4b7f61965bb36d2653ebc79cb891bd60621 100644 --- a/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py +++ b/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py @@ -71,9 +71,9 @@ def do_save_inference_model(args): assert (args.init_from_params) or (args.init_from_pretrain_model) if args.init_from_params: - fluid.load(test_prog, args.init_from_params) + fluid.load(test_prog, args.init_from_params, executor=exe) elif args.init_from_pretrain_model: - fluid.load(test_prog, args.init_from_pretrain_model) + fluid.load(test_prog, args.init_from_pretrain_model, executor=exe) # saving inference model fluid.io.save_inference_model( diff --git a/PaddleNLP/dialogue_system/dialogue_general_understanding/inference_model.py b/PaddleNLP/dialogue_system/dialogue_general_understanding/inference_model.py index d1786b32b8861aa62030aedc9ff4eda7f7312787..f645429c91c6ae4436723dbcbc653f07aab0f3b6 100644 --- a/PaddleNLP/dialogue_system/dialogue_general_understanding/inference_model.py +++ b/PaddleNLP/dialogue_system/dialogue_general_understanding/inference_model.py @@ -99,7 +99,7 @@ def do_save_inference_model(args): assert (args.init_from_params) if args.init_from_params: - fluid.load(test_prog, args.init_from_params) + fluid.load(test_prog, args.init_from_params, executor=exe) # saving inference model fluid.io.save_inference_model( diff --git a/PaddleNLP/dialogue_system/dialogue_general_understanding/predict.py b/PaddleNLP/dialogue_system/dialogue_general_understanding/predict.py index 0c530e99493c375f3a61d85b76fa0f0125fe8429..78cbe61060254919a1b808facfa9c1a547999008 100644 --- a/PaddleNLP/dialogue_system/dialogue_general_understanding/predict.py +++ b/PaddleNLP/dialogue_system/dialogue_general_understanding/predict.py @@ -109,7 +109,7 @@ def do_predict(args): assert (args.init_from_params) if args.init_from_params: - fluid.load(test_prog, args.init_from_params) + fluid.load(test_prog, args.init_from_params, executor=exe) compiled_test_prog = fluid.CompiledProgram(test_prog)