The calflops is designed to calculate FLOPs、MACs and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)

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

Calculate Pytorch model FLOPs

Deep Learning Parameter Calculator & AI Architecture Advisor — Calculate parameters, FLOPs, and memory for 14+ layer types. AI-powered architecture generation via RAG pipeline. PyTorch/TensorFlow/JAX code gen.

Transfer learning or fetching pretrained weights is one of the most utilizing way to train your model and classify the classes based on it. In this repository , transfer learning has been perfomed on different models in pytorch and calculated flops, parameters ,training time, best accuracy as well as testing accurcy.

A Python library to calculate FLOPs for PyTorch models.