flops-counter
Flops counter for neural networks in pytorch framework
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)
Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
FLOPs and other statistics COunter for tf.keras neural networks
FLOPs & Complexity Calculator for PyTorch Deep Learning Model