/torchsummaryX

torchsummaryX: Improved visualization tool of torchsummary

Primary LanguagePython

Stargazers