Open SuziKim opened 2 years ago
Thanks for the PR? Could you describe the issue it addresses?
Sure, sure. Currently, only webGL mode allows text to be displayed for the AlexNet. It is a hotfix that displays the text of the tensor and conv dimensions in SVG mode.
Ah! That's quite useful!
When I test the PR on my machine, switching to SVG mode on the AlexNet vis still disables the "show dimensions" radio buttons. I think you can fix that here?
@alexlenail I updated your comment in https://github.com/alexlenail/NN-SVG/pull/44/commits/0978d1af0f275c50bc1daf147c99a89c4b092d77. Thank you for your comment!
Hi @SuziKim
On your fork, I can't turn on dimension labels in SVG rendering:
Is that what you're seeing also?
May I ask about your environment? @alexlenail For me, it works on Chrome and Safari on macOS Big Sur, as seen below:
Can you make it such that it will make a visualisation for any PyTorch model? also in the conv layers, instead of making big cubes, can you segment it into grids, with multiple filters not connected into a cube but as different layers and also, a small number showing the output dimensions at a chosen layer, with the 2p-k/s equation? Would be really helpful, Thanks :)
If there's a better way, please feel free to discuss!