Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.
MIT License
4.85k stars 528 forks source link

Adding Jit profile #150

Open HaoKang-Timmy opened 2 years ago

HaoKang-Timmy commented 2 years ago
  1. Adding class JitProfile, which can count macs and params
  2. fix some bugs in counter
HaoKang-Timmy commented 2 years ago
model macs params
alexnet 7.7406e+08 61100840
vgg11 8.1484e+09 132863336
vgg11_bn 8.1559e+09 132868840
vgg13 8.1559e+09 132868840
vgg13_bn 1.1860e+10 133053736
vgg16 1.6010e+10 138357544
vgg19 2.0171e+10 143667240
resnet18 1.8163e+09 11689512
resnet50 4.1024e+09 25557032
wide_resnet101_2 2.2776e+10 126886696
densenet121 2.8500e+09 7978856
squeezenet1_0 8.1893e+08 1248424
mnasnet0_75 2.2062e+08 3170208
mobilenet_v2 3.0781e+08 3504872