提交 34855d8e 编写于 作者: W wuzewu

add inference program

上级 f3275539
...@@ -30,6 +30,7 @@ class Task(object): ...@@ -30,6 +30,7 @@ class Task(object):
self.graph_var_dict = graph_var_dict self.graph_var_dict = graph_var_dict
self._main_program = main_program self._main_program = main_program
self._startup_program = startup_program self._startup_program = startup_program
self._inference_program = main_program.clone(for_test=True)
def variable(self, var_name): def variable(self, var_name):
if var_name in self.graph_var_dict: if var_name in self.graph_var_dict:
...@@ -42,3 +43,6 @@ class Task(object): ...@@ -42,3 +43,6 @@ class Task(object):
def startup_program(self): def startup_program(self):
return self._startup_program return self._startup_program
def inference_program(self):
return self._inference_program
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册