Is there any excellent tool to visualize the pytorch model?

1

Hi,

So far as I know , I can just browse the model by enumerating the model.modules().

Any good tool to visualize the model ?

4 Likes
2

Have a look at this post Print Autograd Graph

2 Likes
3

For now, there is a naive graphviz solution. Print Autograd Graph

1 Like