深度学习中的FLOPs和Params如何计算

仅作为记录，大佬请跳过。

当model读取后，使用：

from thop import profile

dummy_input = torch.randn(1, 3, 224, 224)
flops, params = profile(model, (dummy_input,))
print('flops: ', flops, 'params: ', params)
print('flops: %.2f M, params: %.2f M' % (flops / 1000000.0, params / 1000000.0))

即可。

如resnet-101的：

参考

传送门