make_dot without backward nodes?

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

MIT License

3.24k stars 279 forks source link

make_dot without backward nodes? #32

Closed zhangjing1997 closed 5 years ago

zhangjing1997 commented 5 years ago

For a huge or complicated network, directly drawing with backward nodes seems to be lengthy and jumbled. So I'm wondering does this package support drawing model graph without showing backward nodes (but only show forward layers/nodes)?

Thanks for your attention.

szagoruyko commented 5 years ago

check make_dot_from_trace. in make_dot there is no notion of 'forward graph', the graph shown is backward graph needed to do backprop

ray-lee-94 commented 5 years ago

So can i draw a forward graph?

szagoruyko commented 5 years ago

see above, there is no forward graph in pytorch, unless you do jit trace