[人工智能]MACs 和 FLOPs 有什么区别？如何计算？

一、MACs 和 FLOPs 有什么区别？

FLOPs is abbreviation of floating operations which includes mul / add / div … etc.

MACs stands for multiply–accumulate operation that performs a <- a + (b x c).¹²

二、如何计算？

安装thop

pip install thop

pytorch使用thop：³

from torchvision.models import resnet50
from thop import profile
model = resnet50()
input = torch.randn(1, 3, 224, 224)
macs, params = profile(model, inputs=(input, ))

三、MACs 和 FLOPs的关系

粗略的：⁴,¹,⁵

MACs = 2 * FLOPs