From 837fcb054873fdd21be803188891af505a8ca0dd Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Tue, 22 Aug 2023 19:06:56 -0700 Subject: [PATCH 1/2] use join --- mmdnn/conversion/pytorch/pytorch_emitter.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmdnn/conversion/pytorch/pytorch_emitter.py b/mmdnn/conversion/pytorch/pytorch_emitter.py index ca6d6e0e..e1d6305a 100644 --- a/mmdnn/conversion/pytorch/pytorch_emitter.py +++ b/mmdnn/conversion/pytorch/pytorch_emitter.py @@ -52,8 +52,7 @@ def run(self, dstNetworkPath, dstWeightPath = None, phase = 'test'): def add_init(self, indent, codes): if isinstance(codes, _string_types): codes = [codes] - for code in codes: - self.init_code += (" " * indent) + code + '\n' + self.init_code += "\n".join([(" " * indent) + code for code in codes]) def parent_variable_name(self, IR_node, path=[0], weight_type='weights'): if not IR_node.in_edges and IR_node.name in self.weights_dict.keys(): From 647f8fbe9c2e8907fc859758744cec763fa0be94 Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Wed, 6 Sep 2023 22:09:59 -0700 Subject: [PATCH 2/2] fix bug --- mmdnn/conversion/pytorch/pytorch_emitter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdnn/conversion/pytorch/pytorch_emitter.py b/mmdnn/conversion/pytorch/pytorch_emitter.py index e1d6305a..6c6a5a18 100644 --- a/mmdnn/conversion/pytorch/pytorch_emitter.py +++ b/mmdnn/conversion/pytorch/pytorch_emitter.py @@ -52,7 +52,7 @@ def run(self, dstNetworkPath, dstWeightPath = None, phase = 'test'): def add_init(self, indent, codes): if isinstance(codes, _string_types): codes = [codes] - self.init_code += "\n".join([(" " * indent) + code for code in codes]) + self.init_code += "\n".join([(" " * indent) + code for code in codes]) + "\n" def parent_variable_name(self, IR_node, path=[0], weight_type='weights'): if not IR_node.in_edges and IR_node.name in self.weights_dict.keys():