GitHub repository: vra/flopth
Primary language: Python
Description: A simple program to calculate the FLOPs of Pytorch models, multiple inputs supported, with cli tool and Python APIs.