From 5e80f0211e5448508d74f547d40c1554cde36ac2 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 13 Jan 2022 17:45:06 +0800 Subject: [PATCH] fix(mge/network): fix Network not returning dump_info to align trace.dump GitOrigin-RevId: 865adf75e2549d53bda0cbb44d20f39276c1dc09 --- imperative/python/megengine/utils/network.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/imperative/python/megengine/utils/network.py b/imperative/python/megengine/utils/network.py index f0fbd4a33..88a1552da 100644 --- a/imperative/python/megengine/utils/network.py +++ b/imperative/python/megengine/utils/network.py @@ -241,7 +241,7 @@ class Network: metadata.optimize_options = optimize_options G.set_priority_to_id([o._node if isinstance(o, G.VarNode) else o for o in out]) - dump_content, _ = G.dump_graph( + dump_content, dump_info = G.dump_graph( out, keep_var_name=keep_var_name, keep_opr_name=keep_opr_name, @@ -255,6 +255,7 @@ class Network: permission = "wb" if append == False else "ab" file = open(file, permission) file.write(dump_content) + return dump_info def make_const(self, data, name=None, device=None): r"""Makes an ImmutableTensor OpNode to provide a parameter for the network.""" -- GitLab