未验证 提交 ffb1a225 编写于 作者: L LielinJiang 提交者: GitHub

fix load inference model (#139)

上级 f743afdd
......@@ -38,11 +38,10 @@ class BasePredictor(object):
param_file = file_name
self.program, self.feed_names, self.fetch_targets = paddle.static.load_inference_model(
dirname=self.weight_path,
self.weight_path,
executor=self.exe,
model_filename=model_file,
params_filename=param_file)
print(self.feed_names)
def base_forward(self, inputs):
if paddle.in_dynamic_mode():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册