Metadata-Version: 2.1
Name: thop
Version: 0.0.23-1907171430
Summary: A tool to count the FLOPs of PyTorch model.
Home-page: https://github.com/Lyken17/pytorch-OpCounter/
Author: Ligeng Zhu
Author-email: lykensyu+github@gmail.com
License: MIT
Description: # 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 
            ```python
            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.
            
            ```python
            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
        
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Description-Content-Type: text/markdown
