how to calculate the number of parameters and computational complexity for each model

[wangluoWSG]

Hello, I want to calculate the number of parameters and computational complexity for each model (like U-Net, U-Net++, etc.). How can I implement this?

[qubvel]

Hi @wangluoWSG

The number of parameters could be computed with

num_params = sum(p.numel() for p in model.parameters())

To compute model FLOPs you can try the following tools

• https://github.com/Lyken17/pytorch-OpCounter/
• https://github.com/sovrasov/flops-counter.pytorch
• https://github.com/Swall0w/torchstat