Is there any excellent tool to visualize the pytorch model?

2

Have a look at this post Print Autograd Graph

2 Likes