diff --git a/tensorflow2fluid/src/paddle_emitter.py b/tensorflow2fluid/src/paddle_emitter.py index 257b0a4f6592f43571f6d58c6dd19975c4947ba3..01cd1c1e89718c0b4257da17638849d11a186a5d 100644 --- a/tensorflow2fluid/src/paddle_emitter.py +++ b/tensorflow2fluid/src/paddle_emitter.py @@ -177,7 +177,7 @@ class PaddleEmitter(object): code.append("import paddle.fluid as fluid") code.append("import numpy") code.append("") - code.append("def KitModel():") + code.append("def Model():") return code def add_codes(self, indent, codes): @@ -233,7 +233,6 @@ class PaddleEmitter(object): # merge all the generated python codes for node in translate_nodes: - print(self.graph.get_node(node).output_name, node) codes = self.graph.get_node(node).code.gen_codes() self.add_codes(1, codes)