From 450160e85f8dc17050a25e46d890c50eb1e32379 Mon Sep 17 00:00:00 2001
From: 0YuanZhang0 <953963890@qq.com>
Date: Thu, 21 May 2020 17:17:41 +0800
Subject: [PATCH] upgrade ade  dgu load api (#4659)

* fix_ade_readme

* fix_load_api
---
 .../auto_dialogue_evaluation/inference_model.py               | 4 ++--
 .../dialogue_general_understanding/inference_model.py         | 2 +-
 .../dialogue_system/dialogue_general_understanding/predict.py | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py b/PaddleNLP/dialogue_system/auto_dialogue_evaluation/inference_model.py
index 6c902d5f..7263e4b7 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 d1786b32..f645429c 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 0c530e99..78cbe610 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)
 
-- 
GitLab