Pretty model visualization

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
PyPi: https://pypi.org/project/visunn/