Is there any excellent tool to visualize the pytorch model?

[liygcheng (Lechao Cheng)]

Hi,

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

Any good tool to visualize the model ?

[fmassa (Francisco Massa)]

Have a look at this post Print Autograd Graph

[wangg12 (Gu Wang)]

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