Browse Source

fix(mge/network): fix Network not returning dump_info to align trace.dump

GitOrigin-RevId: 865adf75e2
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
5e80f0211e
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/utils/network.py

+ 2
- 1
imperative/python/megengine/utils/network.py View File

@@ -241,7 +241,7 @@ class Network:
metadata.optimize_options = optimize_options metadata.optimize_options = optimize_options


G.set_priority_to_id([o._node if isinstance(o, G.VarNode) else o for o in out]) 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, out,
keep_var_name=keep_var_name, keep_var_name=keep_var_name,
keep_opr_name=keep_opr_name, keep_opr_name=keep_opr_name,
@@ -255,6 +255,7 @@ class Network:
permission = "wb" if append == False else "ab" permission = "wb" if append == False else "ab"
file = open(file, permission) file = open(file, permission)
file.write(dump_content) file.write(dump_content)
return dump_info


def make_const(self, data, name=None, device=None): def make_const(self, data, name=None, device=None):
r"""Makes an ImmutableTensor OpNode to provide a parameter for the network.""" r"""Makes an ImmutableTensor OpNode to provide a parameter for the network."""


Loading…
Cancel
Save