CODE EXAMPLE FOR PYTHON

torch summary

from torchvision import models
from torchsummary import summary

vgg = models.vgg16()
summary(vgg, (3, 224, 224))