PyPI page
Home page
Author:
None
License:
MIT License Copyright (c) [2023] [MrYXJ] Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in...
Summary:
This package is designed to compute the theoretical amount of FLOPs(floating-point operations)、MACs(multiply-add operations) and Parameters in all various neural networks, such as Linear、 CNN、 RNN、 GCN、Transformer(Bert、LlaMA etc Large Language Model)，including any custom models via torch.nn.function.* as long as based on the Pytorch implementation.
Latest version:
0.3.2
Required dependencies:
accelerate
|
huggingface-hub
|
torch
Downloads last day:
960
Downloads last week:
4,841
Downloads last month:
31,717