|
@@ -248,7 +248,7 @@ class Main(object):
|
|
|
|
|
|
# trans to new format
|
|
|
# {"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.load_inference_model(os.path.join(model_dir, "dnn_plugin"), model_filename=None, params_filename=None})
|
|
|
paddle.static.save_inference_model(
|
|
|
os.path.join(model_dir, "dnn_plugin_new"),
|
|
|
[feed.name for feed in self.inference_feed_var],
|