GitHub repository: tiskw/pytorch-op-counter-layerwise
Primary language: Python
Description: Count the layer-wise MACs and the number of parameters of your PyTorch model.