thop_pytorch_model_flops_and_parameter_count_calculation.py

1import torch
2from torchvision.models import resnet50
3from thop import profile
4
5# Define the model
6model = resnet50()
7
8# Create a dummy input tensor matching the model's input shape
9input = torch.randn(1, 3, 224, 224)
10
11# Calculate FLOPs and Parameters
12macs, params = profile(model, inputs=(input, ))
13
14print(f"Total MACs (FLOPs): {macs}")
15print(f"Total Parameters: {params}")