未验证 提交 3635fc64 编写于 作者: Q Qiyang Min 提交者: GitHub

Merge pull request #14994 from velconia/fix_net_drawer

Fix net_drawer
......@@ -26,7 +26,7 @@ logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
try:
from .graphviz import Digraph
from .graphviz import Graph
except ImportError:
logger.info(
'Cannot import graphviz, which is required for drawing a network. This '
......@@ -112,7 +112,7 @@ def draw_graph(startup_program, main_program, **kwargs):
filename = kwargs.get("filename")
if filename == None:
filename = str(graph_id) + ".gv"
g = Digraph(
g = Graph(
name=str(graph_id),
filename=filename,
graph_attr=GRAPH_STYLE,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册