|
@@ -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],
|