flops, params = profile(model, inputs=(x,))计算

 

计算量:
FLOPs,FLOP时指浮点运算次数,s是指秒,即每秒浮点运算次数的意思,考量一个网络模型的计算量的标准。
参数量:
Params,是指网络模型中需要训练的参数总数。

 

flops(G) = flops / 1000**3

params(M) = params /1000**2

posted @ 2021-08-20 19:44  九叶草  阅读(747)  评论(0编辑  收藏  举报