pytorch統計模型參數量

用resnet50 來舉例子

from torchvision.models import resnet18, resnet34, resnet50

model = resnet50()
print("resnet50 have {} paramerters in total".format(sum(x.numel() for x in model.parameters())))

其中numel表示含有多少element，通過此操作可以統計模型的參數量有多少

 

另外，兩個是一樣的，方便debug看其中到底有什么東西

    num_params = 0
    for param in model.parameters():
        num_params += param.numel()
    print(num_params / 1e6)