From 3f5c56c7c4a421efba312d3ae983a034995ce9c0 Mon Sep 17 00:00:00 2001 From: wjj19950828 Date: Wed, 30 Mar 2022 17:42:03 +0800 Subject: [PATCH] update optimize log --- x2paddle/optimizer/optimizer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/x2paddle/optimizer/optimizer.py b/x2paddle/optimizer/optimizer.py index 25e496f..49c6238 100644 --- a/x2paddle/optimizer/optimizer.py +++ b/x2paddle/optimizer/optimizer.py @@ -42,17 +42,22 @@ class GraphOptimizer(object): self.passes = [] def optimize(self, graph): + show_pass_log = False for pass_name in self.passes: pass_ = PassManager.lookup(pass_name)() if pass_name.endswith("_eliminate_pass") or pass_name.endswith( "conv2d_add_fuse_pass"): pass_.apply(graph) + show_pass_log = True else: while True: before_len = len(graph.layers) pass_.apply(graph) after_len = len(graph.layers) + if after_len < before_len: + show_pass_log = True if before_len == after_len: break - print("{} done!".format(pass_name)) + if show_pass_log: + print("{} done!".format(pass_name)) return graph -- GitLab