一、 含义
FLOPs
(计算量):注意s
小写,是floating point operations
的缩写(这里的小s
则表示复数),表示浮点运算数,理解为计算量。可以用来衡量算法/模型的复杂度。
Params
(参数量):表示模型的参数量,也是用来衡量算法/模型的复杂度。
二、 thop安装
pip install thop
三、代码
import torch
#--------------------#
# 计算FLOPs,Parmas
#--------------------#
from thop import profile
from net import YourModel
import config
your_model = YourModel()
input1 = torch.randn(1, 3, 320, 320)
flops, params = profile(your_model, inputs=(input1, ))
#--------------------#
# 计算量
#--------------------#
print('FLOPs = ' + str(flops/1000**3) + 'G')
#--------------------#
# 参数量
#--------------------#
print('Params = ' + str(params/1000**2) + 'M')