torchinfo (conda): View model summaries in PyTorch!
Keywords: keras|python|pytorch|torch|torch-summary|torchinfo|torchsummary|torchvision|visualization