weidel-p / calculate-flops.pytorch Goto Github PK
View Code? Open in Web Editor NEWThis project forked from mryxj/calculate-flops.pytorch
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)
Home Page: https://pypi.org/project/calflops/
License: MIT License