记录一下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
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。