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

add inference program

上级 f3275539
......@@ -30,6 +30,7 @@ class Task(object):
self.graph_var_dict = graph_var_dict
self._main_program = main_program
self._startup_program = startup_program
self._inference_program = main_program.clone(for_test=True)
def variable(self, var_name):
if var_name in self.graph_var_dict:
......@@ -42,3 +43,6 @@ class Task(object):
def startup_program(self):
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.
先完成此消息的编辑!
想要评论请 注册