I’ve recently created a tool for visualization pytorch models and released a python package today. Just thought I’d share here, feel free to check out!
Github: https://github.com/vliu15/visunn 8
PyPi: https://pypi.org/project/visunn/ 1