Pytorch模型flops, params計算

thop

import thop
from torchvision.models import resnet50

if __name__ == "__main__":

    model = resnet50()
    input = torch.randn(1, 3, 224, 224)
    flops, params = thop.profile(model, inputs=(input, ))
    flops, params = thop.clever_format([flops, params], "%.3f")  # 提升結果可讀性
    

Naive

1

1