記錄一下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刪除。