From d9ccc90148278eed23f30559d45a90266c43a220 Mon Sep 17 00:00:00 2001
From: yukavio <67678385+yukavio@users.noreply.github.com>
Date: Wed, 9 Dec 2020 11:00:12 +0800
Subject: [PATCH] fix prune demo (#535)

---
 demo/prune/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/demo/prune/train.py b/demo/prune/train.py
index 138bf55c..08ec41e0 100644
--- a/demo/prune/train.py
+++ b/demo/prune/train.py
@@ -240,8 +240,8 @@ def compress(args):
         if args.save_inference:
             infer_model_path = os.path.join(args.model_path, "infer_models",
                                             str(i))
-            paddle.static.save_inference_model(infer_model_path, ["image"],
-                                               [out], exe, pruned_val_program)
+            paddle.fluid.io.save_inference_model(infer_model_path, ["image"],
+                                                 [out], exe, pruned_val_program)
             _logger.info("Saved inference model into [{}]".format(
                 infer_model_path))
 
-- 
GitLab