Update dot.py #33

z-huabao · 2019-07-12T09:34:35Z

1, remove Backend of node(inner tensor) name
2, add shape to the node
3, add input and input shape
4, change make_dot function args to make_dot(model, x)

1, remove `Backend` of node 2, add shape to the node 3, add input and input shape

lizhaoliu-Lec · 2020-03-26T16:10:53Z

The new features added by you is fantastic!!! Can you find a way to support multiple inputs? Thanks very much!

txntxn · 2020-06-14T21:13:51Z

torchviz/dot.py

    """
+    var = model(x)


This doesn't allow you to use multiple inputs.

I propose:

var = model(*x)

Update dot.py

98d8652

1, remove `Backend` of node 2, add shape to the node 3, add input and input shape

txntxn reviewed Jun 14, 2020

View reviewed changes

torchviz/dot.py

"""

var = model(x)

Copy link

txntxn Jun 14, 2020

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This doesn't allow you to use multiple inputs.

I propose:

var = model(*x)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Update dot.py #33

Update dot.py #33

z-huabao commented Jul 12, 2019 •

edited

Loading

lizhaoliu-Lec commented Mar 26, 2020

txntxn Jun 14, 2020

Update dot.py #33

Are you sure you want to change the base?

Update dot.py #33

Conversation

z-huabao commented Jul 12, 2019 • edited Loading

lizhaoliu-Lec commented Mar 26, 2020

txntxn Jun 14, 2020

Choose a reason for hiding this comment

z-huabao commented Jul 12, 2019 •

edited

Loading