Is there any excellent tool to visualize the pytorch model?

3

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

1 Like