pytorch-OpCounter

Model profiler

An open-source tool for counting the number of multiplication operations (MACs) and floating-point operations (FLOPs) in PyTorch models.

Count the MACs / FLOPs of your PyTorch model.

GitHub

5k stars
29 watching
528 forks
Language: Python
last commit: 5 months ago

Related projects:

Repository Description Stars
titu1994/densenet An implementation of a deep learning network architecture with dense connectivity and bottleneck layers. 707
szagoruyko/wide-residual-networks An experimental study on residual networks to improve depth and width trade-offs in neural networks 1,299
tornadomeet/resnet Reproducing ResNet-v2 with MXNet on ImageNet and CIFAR10 datasets using Python. 559
idealo/image-super-resolution A project providing tools and frameworks for improving the quality of low-resolution images through deep learning-based techniques. 4,674
titu1994/keras-resnext An implementation of ResNeXt models in Keras, allowing for efficient deep neural networks for image classification. 224
nutszebra/resnet_in_resnet An implementation of Residual Networks In Residual Networks using Chainer. 3
liuzhuang13/densenet A deep learning implementation of a neural network architecture with interconnected layers, designed to improve accuracy on image classification tasks 4,757
n2cholas/jax-resnet Provides implementations and checkpoints for various ResNet variants using JAX and Flax. 105
hasnainraz/fc-densenet-tensorflow Re-implementation of a 100-layer fully convolutional network architecture for image segmentation 123
simjeg/fc-densenet An implementation of a fully convolutional densely connected neural network architecture for semantic segmentation tasks. 487
xternalz/wideresnet-pytorch An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets. 334
prlz77/resnext.pytorch Reproduces ResNet-V3 with PyTorch for computer vision tasks 511
taki0112/senet-tensorflow A TensorFlow implementation of Squeeze and Excitation Networks for image classification tasks 756
hli1221/imagefusion_resnet50 A deep learning-based approach to fuse infrared and visible images using ResNet50 architecture and zero-phase component analysis 81
raghakot/keras-resnet An implementation of Residual Networks using Keras' functional API 1,385