THOP: PyTorch-OpCounter

How to install

Through PyPi

pip install thop
Using GitHub (always latest)

pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git

How to use

Basic usage

from torchvision.models import resnet50
from thop import profile
model = resnet50()
input = torch.randn(1, 3, 224, 224)
flops, params = profile(model, inputs=(input, ))

Define the rule for 3rd party module.

class YourModule(nn.Module):
    # your definition
def count_your_model(model, x, y):
    # your rule here

input = torch.randn(1, 3, 224, 224)
flops, params = profile(model, inputs=(input, ), 
                        custom_ops={YourModule: count_your_model})

Results on Recent Models

Model	Params(M)	FLOPs(G)
alexnet	61.10	0.71
vgg11	132.86	7.75
vgg11_bn	132.87	7.76
vgg13	133.05	11.46
vgg13_bn	133.05	11.48
vgg16	138.36	15.62
vgg16_bn	138.37	15.65
vgg19	143.67	19.79
vgg19_bn	143.68	19.82
densenet121	7.98	2.79
densenet161	28.68	7.69
densenet169	14.15	3.33
densenet201	20.01	4.28
resnet18	11.69	1.58
resnet34	21.80	3.44
resnet50	25.56	3.53
resnet101	44.55	7.26
resnet152	60.19	10.99
squeezenet1_0	1.25	0.70
squeezenet1_1	1.24	0.34

Metadata

Release files for thop 0.0.23-1907171430

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for thop 0.0.23-1907171430
File	Size	Uploaded
thop-0.0.23-1907171430.tar.gz	3.9 kB	Jul 17, 2019	Details

Built distribution (wheel)

Table of built distributions (wheels) for thop 0.0.23-1907171430
File	Interpreter	ABI	Platform	Reset
thop-0.0.23.post1907171430-py3-none-any.whl 4.5 kB Jul 17, 2019	Python 3	none	any	Details

Total release size: 8.5 kB

Release files / thop-0.0.23-1907171430.tar.gz

Download URL	thop-0.0.23-1907171430.tar.gz
Size	3.9 kB
Tags	Source
SHA-256 checksum How to use checksums	`e0148585f52b185d4bd30e427c03ecae3fd78519f393c3ff864170c3612eeaf2`
BLAKE2b-256 checksum How to use checksums	`ae78be411a42dd908285d7d037ac01cc76df84ea03dbe25b01d7d523fd55ddfc`
Upload date	Jul 17, 2019
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/1.13.0 pkginfo/1.4.2 requests/2.21.0 setuptools/39.1.0 requests-toolbelt/0.9.1 tqdm/4.26.0 CPython/3.6.5`

Release files / thop-0.0.23.post1907171430-py3-none-any.whl

Download URL	thop-0.0.23.post1907171430-py3-none-any.whl
Size	4.5 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`6ca9b375729e9718fd80e7f14666e400d25177d181a2242752b1d00e1f60a045`
BLAKE2b-256 checksum How to use checksums	`124556dcc90ece0a04a34b802541461e2b72bf1d33df8a4631f9cb26dcc99eec`
Upload date	Jul 17, 2019
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/1.13.0 pkginfo/1.4.2 requests/2.21.0 setuptools/39.1.0 requests-toolbelt/0.9.1 tqdm/4.26.0 CPython/3.6.5`