How to hide the backward node?

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

MIT License

3.24k stars 279 forks source link

Closed ray-lee-94 closed 5 years ago

ray-lee-94 commented 5 years ago

I plot the graph in a PDF which is a long file. Is there any method to hide the backward so that the graph is smaller and more clear ?

ray-lee-94 commented 5 years ago

I find the HidenLayer predict the forward graph. Thank your for the code.