from torchstat import stat import torchvision.models as models model = models.resnet152() stat(model, (3, 224, 224))