torch model parameter(params) 개수 구하기

from torchvision import models


def get_n_params(model):
    pp = 0
    for p in list(model.parameters()):
        nn = 1
        for s in list(p.size()):
            nn = nn * s
        pp += nn
    return pp

resnet50_pretrained = models.resnet50(pretrained=True)
parms_num = get_n_params(resnet50_pretrained)  # 25557032 = 25.55M