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