Skip to content

Commit 0adcd83

Browse files
authored
Remove edge direction for saved_tensors (#63)
Fixes #61
1 parent 1e81953 commit 0adcd83

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchviz/dot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def add_nodes(fn):
139139
if hasattr(fn, 'saved_tensors'):
140140
for t in fn.saved_tensors:
141141
seen.add(t)
142-
dot.edge(str(id(t)), str(id(fn)))
142+
dot.edge(str(id(t)), str(id(fn)), dir="none")
143143
dot.node(str(id(t)), get_var_name(t), fillcolor='orange')
144144

145145

0 commit comments

Comments
 (0)