torchinfo (pypi): Model summary in PyTorch, based off of the original torchsummary.
Keywords: torch|pytorch|torchsummary|torch-summary|summary|keras|deep-learning|ml|torchinfo|torch-info|visualize|model|statistics|layer|stats|python|torchvision|visualization