提交 9b567eb7 编写于 作者: S SunAhong1993

fix the Chinese

上级 4c8b9537
...@@ -510,7 +510,7 @@ class PaddleGraph(object): ...@@ -510,7 +510,7 @@ class PaddleGraph(object):
comment_list = list() comment_list = list()
comment_list.append("# There are {} inputs.".format(len(self.inputs_info))) comment_list.append("# There are {} inputs.".format(len(self.inputs_info)))
for k, v in self.inputs_info.items(): for k, v in self.inputs_info.items():
comment_list.append("# {}: shape-{},type-{}。".format(k, v[0], v[1])) comment_list.append("# {}: shape-{}, type-{}.".format(k, v[0], v[1]))
self.run_func.extend( self.run_func.extend(
gen_codes( gen_codes(
comment_list, comment_list,
......
...@@ -65,5 +65,6 @@ class TraceDecoder(Decoder): ...@@ -65,5 +65,6 @@ class TraceDecoder(Decoder):
exit(0) exit(0)
self.graph = self._optimize_graph(self.script.inlined_graph) self.graph = self._optimize_graph(self.script.inlined_graph)
self.input_examples = input_examples self.input_examples = input_examples
print(self.graph)
...@@ -385,7 +385,7 @@ class HierarchicalTree(Tree): ...@@ -385,7 +385,7 @@ class HierarchicalTree(Tree):
run_func_list.append("def main({}):".format(input_data_name)) run_func_list.append("def main({}):".format(input_data_name))
run_func_list.append(" # There are {} inputs.".format(len(self.pd_graph.inputs_info))) run_func_list.append(" # There are {} inputs.".format(len(self.pd_graph.inputs_info)))
for k, v in self.pd_graph.inputs_info.items(): for k, v in self.pd_graph.inputs_info.items():
run_func_list.append(" # {}: shape-{},type-{}。".format(k, v[0], v[1])) run_func_list.append(" # {}: shape-{}, type-{}.".format(k, v[0], v[1]))
run_func_list.extend( run_func_list.extend(
[" paddle.disable_static()", [" paddle.disable_static()",
" params = paddle.load('{}')".format(osp.join(osp.abspath(save_dir), "model.pdparams")), " params = paddle.load('{}')".format(osp.join(osp.abspath(save_dir), "model.pdparams")),
......
...@@ -351,7 +351,7 @@ class ModuleGraph(object): ...@@ -351,7 +351,7 @@ class ModuleGraph(object):
run_func_list.append("def main({}):".format(input_data_name)) run_func_list.append("def main({}):".format(input_data_name))
run_func_list.append(" # There are {} inputs.".format(len(self.pd_graph.inputs_info))) run_func_list.append(" # There are {} inputs.".format(len(self.pd_graph.inputs_info)))
for k, v in self.pd_graph.inputs_info.items(): for k, v in self.pd_graph.inputs_info.items():
run_func_list.append(" # {}: shape-{}type-{}.".format(k, v[0], v[1])) run_func_list.append(" # {}: shape-{}, type-{}.".format(k, v[0], v[1]))
run_func_list.extend( run_func_list.extend(
[" paddle.disable_static()", [" paddle.disable_static()",
" params = paddle.load('{}')".format(osp.join(osp.abspath(save_dir), "model.pdparams")), " params = paddle.load('{}')".format(osp.join(osp.abspath(save_dir), "model.pdparams")),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册