How to visualize GraphModule

1

Hi all,

I want to visualize the torch.fx.GraphModule , how to do it?

2

You could give torchviz a go and see if that works?

1 Like
3

You can use this: pytorch/graph_drawer.py at master · pytorch/pytorch · GitHub

1 Like
4

Thank you so much, it worked for me.