I’ve read a lot of posts about graph visualization in pytorch, but i still can’t find something really usable.
I’ve tried with https://github.com/szagoruyko/functional-zoo/blob/master/visualize.py but graphviz’s image conversion seems too heavy if we have a large model (i.e. resnet34) and my process goes on for hours if i try to save a png or any image format.
I’ve also read about the tensorboard integration, but i use visdom and would like to stick with it.
So are there any new ways to get the pytorch computational graph and get an image from it?
My final goal is to send the image to visdom in order to visualize it with all my other logs.