/flopth

A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.

Primary LanguagePythonMIT LicenseMIT

Issues