From b584b1ba301c05236ea162a542f579506320ebf4 Mon Sep 17 00:00:00 2001
From: SunAhong1993 <48579383+SunAhong1993@users.noreply.github.com>
Date: Thu, 10 Dec 2020 14:55:36 +0800
Subject: [PATCH] Update merge_params.py

---
 tools/merge_params.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tools/merge_params.py b/tools/merge_params.py
index 1b2fa18..f85e7ab 100644
--- a/tools/merge_params.py
+++ b/tools/merge_params.py
@@ -1,15 +1,17 @@
+import paddle
 import paddle.fluid as fluid
 import sys
 
 model_dir = sys.argv[1]
 new_model_dir = sys.argv[2]
-exe = fluid.Executor(fluid.CPUPlace())
+paddle.enable_static()
+exe = paddle.static.Executor(paddle.CPUPlace())
 [inference_program, feed_target_names,
- fetch_targets] = fluid.io.load_inference_model(
+ fetch_targets] = paddle.static.load_inference_model(
      dirname=model_dir, executor=exe)
 
 print(feed_target_names)
-fluid.io.save_inference_model(
+paddle.static.save_inference_model(
     dirname=new_model_dir,
     feeded_var_names=feed_target_names,
     target_vars=fetch_targets,
-- 
GitLab