2021年11月18日
pytorch中计算MACs/FLOPs和参数量
摘要： 记录一下pytorch中一个计算MACs/FLOPs的包：flops_counter https://github.com/sovrasov/flops-counter.pytorch 还有两个感觉不太好用： torchstat https://github.com/Swall0w/torchsta 阅读全文
posted @ 2021-11-18 19:03 weller 阅读(1530) 评论(0) 推荐(0) 编辑