pytorch中计算MACs/FLOPs和参数量

记录一下pytorch中一个计算MACs/FLOPs的包：flops_counter

https://github.com/sovrasov/flops-counter.pytorch

 

还有两个感觉不太好用：

torchstat https://github.com/Swall0w/torchstat

pytorch-OpCounter https://github.com/Lyken17/pytorch-OpCounter