From 86598c8227de0936a3bd13cb79026849ea3145ab Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 7 Apr 2021 12:53:11 +0800
Subject: [PATCH] fix(mge/utils): filter out parameter "arg_names" and
 "output_name" in network.dump

GitOrigin-RevId: 408f52ad2ddf7f542ba7ad5b0da2699ba3327b77
---
 imperative/python/megengine/utils/network.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/imperative/python/megengine/utils/network.py b/imperative/python/megengine/utils/network.py
index 3ba99219..4a3f5ac1 100644
--- a/imperative/python/megengine/utils/network.py
+++ b/imperative/python/megengine/utils/network.py
@@ -18,6 +18,7 @@ import numpy as np
 from ..core._imperative_rt import ComputingGraph
 from ..core._imperative_rt.core2 import SymbolVar
 from ..core.tensor import megbrain_graph as G
+from ..logger import get_logger
 from .comp_graph_tools import get_dep_vars, get_opr_type, get_oprs_seq
 from .network_node import (
     Host2DeviceCopy,
@@ -28,6 +29,8 @@ from .network_node import (
     str_to_mge_class,
 )
 
+logger = get_logger(__name__)
+
 
 class Network:
     def __init__(self):
@@ -164,6 +167,15 @@ class Network:
         self._compile()
         out = [G.VarNode(var.var) for var in self.output_vars]
 
+        if kwargs.pop("arg_names", False):
+            logger.warning(
+                '"arg_names" is not supported in Network.dump, rename input vars directly'
+            )
+        if kwargs.pop("output_names", False):
+            logger.warning(
+                '"output_names" is not supported in Network.dump, rename output vars directly'
+            )
+
         if optimize_for_inference:
             out = G.optimize_for_inference(out, **kwargs)
 
-- 
GitLab