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

 

posted @ 2021-11-18 19:03  weller  阅读(1336)  评论(0)  编辑  收藏  举报