Giters

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to calculate the FLOPs of each type of layers?

TheSunWillRise opened this issue · comments

commented

I want to calculate the FLOPs of each type of different layers, how can I implement this?
For example, calculate the flops of CNN layers and ReLU layers in resnet50, respectively.