仅作为记录,大佬请跳过。
当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的:
参考