{"library":"ptflops","title":"ptflops","description":"A Flops counter for neural networks in PyTorch. It computes the number of floating point operations (FLOPs) and parameters of a model. The current version is 0.7.5, with a slow release cadence.","language":"python","status":"active","last_verified":"Mon Apr 27","install":{"commands":["pip install ptflops"],"cli":null},"imports":["from ptflops import get_model_complexity_info"],"auth":{"required":false,"env_vars":[]},"quickstart":{"code":"import torch\nfrom ptflops import get_model_complexity_info\nimport torchvision.models as models\n\nnet = models.resnet18()\nmacs, params = get_model_complexity_info(net, (3, 224, 224), as_strings=True, print_per_layer_stat=True)\nprint('{:<30}  {:<8}'.format('Computational complexity: ', macs))\nprint('{:<30}  {:<8}'.format('Number of parameters: ', params))","lang":"python","description":"Compute FLOPs and parameters for ResNet-18 with a 224x224 input.","tag":null,"tag_description":null,"last_tested":null,"results":[]},"compatibility":null}