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()
  flops, params = profile(model, input_size=(1, 3, 224,224))

• Define the rule for 3rd party module.

  class YourModule(nn.Module):
      # your definition
  def count_your_model(model, x, y):
      # your rule here
  flops, params = profile(model, input_size=(1, 3, 224,224), 
                          custom_ops={YourModule: count_your_model})

Results on Recent Models

参考：https://github.com/Lyken17/pytorch-OpCounter