浏览代码

save model

丁云鹏 5 月之前
父节点
当前提交
7c71f77829
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

+ 2 - 1
recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

@@ -247,7 +247,8 @@ class Main(object):
 
 
             # trans to new format
-            paddle.static.load_inference_model(os.path.join(model_dir, "dnn_plugin"))
+            # {"model_filename":"", "params_filename":""} fleet每个参数一个文件,需要同这种方式加载
+            paddle.static.load_inference_model(os.path.join(model_dir, "dnn_plugin"), self.exe, {"model_filename":"", "params_filename":""})
             paddle.static.save_inference_model(
                     os.path.join(model_dir, "dnn_plugin_new"),
                     [feed.name for feed in self.inference_feed_var],