torchsummaryX: Improved visualization tool of torchsummary - View it on GitHub