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})

Improve the output readability

Call thop.clever_format to give a better format of the output.

from thop import clever_format
flops, params = clever_format([flops, params], "%.3f")

Results on Recent Models

Model	Params(M)	FLOPs(G)
alexnet	58.27	0.72
densenet121	7.61	2.70
densenet161	27.35	7.31
densenet169	13.49	3.20
densenet201	19.09	4.09
resnet18	11.15	1.70
resnet34	20.79	3.43
resnet50	24.37	3.85
resnet101	42.49	7.33
resnet152	57.40	10.81

Model	Params(M)	FLOPs(G)
squeezenet1_0	1.19	0.77
squeezenet1_1	1.18	0.33
vgg11	126.71	7.21
vgg11_bn	126.71	7.24
vgg13	126.88	10.66
vgg13_bn	126.89	10.70
vgg16	131.95	14.54
vgg16_bn	131.96	14.59
vgg19	137.01	18.41
vgg19_bn	137.02	18.47

Metadata

Release files for thop 0.0.30-190806

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.30-190806
File	Size	Uploaded
thop-0.0.30-190806.tar.gz	4.1 kB	Aug 6, 2019	Details

Built distribution (wheel)

Table of built distributions (wheels) for thop 0.0.30-190806
File	Interpreter	ABI	Platform	Reset
thop-0.0.30.post190806-py3-none-any.whl 5.6 kB Aug 6, 2019	Python 3	none	any	Details

Total release size: 9.7 kB

Release files / thop-0.0.30-190806.tar.gz

Download URL	thop-0.0.30-190806.tar.gz
Size	4.1 kB
Tags	Source
SHA-256 checksum How to use checksums	`126df0082e13d21164959e2087ad10f160b076a1654a1c5e6e48979a4c225dd4`
BLAKE2b-256 checksum How to use checksums	`cbb0dc61f22c849c5fb8a273e26750665d5e1fc02d74b76e5e96c82167c0093f`
Upload date	Aug 6, 2019
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.3`

Release files / thop-0.0.30.post190806-py3-none-any.whl

Download URL	thop-0.0.30.post190806-py3-none-any.whl
Size	5.6 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`f9b0f4579ff010cc0b07f175786c993a08aa6073a9874e18ab54df7e424b3d34`
BLAKE2b-256 checksum How to use checksums	`4a0827efe0dee0478e534d993362a764692d71303f91dbb7ac8bdae2652c124c`
Upload date	Aug 6, 2019
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/1.13.0 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.0.1 requests-toolbelt/0.9.1 tqdm/4.32.1 CPython/3.7.3`