A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
-
Updated
Nov 23, 2024 - Python
A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
Add a description, image, and links to the flop topic page so that developers can more easily learn about it.
To associate your repository with the flop topic, visit your repo's landing page and select "manage topics."