From b9463b72f5e203c778af104e707ed44002959caf Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Fri, 24 Apr 2020 12:47:20 +0000 Subject: [PATCH] refine save inference model --- hapi/model.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/hapi/model.py b/hapi/model.py index 6b800d2..71b08cc 100644 --- a/hapi/model.py +++ b/hapi/model.py @@ -1180,6 +1180,8 @@ class Model(fluid.dygraph.Layer): export_for_deployment=True, program_only=False): """ + Save inference model must in static mode. + Args: dirname(str): The directory path to save the inference model. model_filename(str|None): The name of file to save the inference program @@ -1201,7 +1203,9 @@ class Model(fluid.dygraph.Layer): Returns: list: The fetch variables' name list """ - # self.prepare(self.) + assert not fluid.in_dygraph_mode( + ), 'Save inference model must in static mode!' + prog = self._adapter._progs.get('test', None) assert prog, \ "Model is not ready, please call `model.prepare()` first" -- GitLab