From 5beaf46308387e77478263fa677650812dfe7bc7 Mon Sep 17 00:00:00 2001 From: rensilin Date: Wed, 4 Sep 2019 19:45:27 +0800 Subject: [PATCH] save_inference_model Change-Id: I8dadf474be1ceb263e3e498bae5e130e72b7ffde --- .../train/custom_trainer/feed/scripts/create_programs.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/paddle/fluid/train/custom_trainer/feed/scripts/create_programs.py b/paddle/fluid/train/custom_trainer/feed/scripts/create_programs.py index e1fad394..1adeb704 100644 --- a/paddle/fluid/train/custom_trainer/feed/scripts/create_programs.py +++ b/paddle/fluid/train/custom_trainer/feed/scripts/create_programs.py @@ -121,6 +121,14 @@ class ModelBuilder: with open(os.path.join(self._save_path, name), 'w') as f: f.write(program.desc.serialize_to_string()) + fluid.io.save_inference_model(self._save_path, + [var.name for var in inputs], + outputs, + executor=None, + main_program=test_program, + model_filename='inference_program', + program_only=True) + params = filter(fluid.io.is_parameter, main_program.list_vars()) vars = [] sums=[] -- GitLab