text_plain_output_1.png
from torchvision import models

model_ft = models.resnet50(pretrained=True)

print('Number of trainable parameters: ', sum((p.numel() for p in model_ft.parameters() if p.requires_grad)))