How to get the graph of your DNN drawn?
Hi all, I am wondering if there is an easy plug-and-play way in order for me to have a nice visualization of my DNN that I have designed? Im after something simple to be honest, just something where I can see the various layers connected to each other to get a sense of what I have coded. Thanks
Kalamaya
fmassa
ido4848
For the moment, we need to wait until the new autograd is merged into master (which should happen next week). Once that’s done, one can start working on a tensorboard integration, as the underlying graph representation from PyTorch shouldn’t change much
fmassa
Any updates on the tensorboard integration? How can I contribute?
varun-suresh
hhsecond
fmassa
Sounds good, i’ll wait for that then. Thanks
hhsecond
Do we have any plans to build own visualization tool based on jit.trace or ONNX? I and my colleagues had built one basic visualizer and we were planning to make a better one. If somebody is working on it already, would like to discuss.
hhsecond