From cf38c676cad51acae111cdea6f18ac3148009a94 Mon Sep 17 00:00:00 2001 From: Xin Pan Date: Fri, 21 Sep 2018 21:58:41 +0800 Subject: [PATCH] fix --- python/paddle/fluid/io.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/io.py b/python/paddle/fluid/io.py index 01d5e7a11ea..604f3eacd75 100644 --- a/python/paddle/fluid/io.py +++ b/python/paddle/fluid/io.py @@ -618,10 +618,6 @@ def save_inference_model(dirname, if main_program is None: main_program = default_main_program() - if params_filename is not None: - params_filename = os.path.basename(params_filename) - save_persistables(executor, dirname, main_program, params_filename) - # if there is lookup table, the trainer 0 will notify all pserver to save. if main_program._is_distributed and main_program._is_chief and main_program._distributed_lookup_table: lookup_table_filename = os.path.join(dirname, "__lookup_table__") @@ -665,6 +661,10 @@ def save_inference_model(dirname, with open(model_basename + ".main_program", "wb") as f: f.write(main_program.desc.serialize_to_string()) + if params_filename is not None: + params_filename = os.path.basename(params_filename) + save_persistables(executor, dirname, main_program, params_filename) + def load_inference_model(dirname, executor, -- GitLab