Is there any API to visualize the architecture of model?

[Pinnh]

I know the 'print' method can show the graph of model,but is there any API to visualize(plot) the architecture of pytorch network model? thanks~

[apaszke]

There's @szagoruyko script that produces a dot file

[Pinnh]

OK，Thanks a lot.