Target detection model size calculation, model complexity (parameter conversion formula)

Target detection model size calculation , Model complexity （ Parameter conversion formula ）

Recently, when doing target detection, we need to compare the size of the model , Because some model output results do not include model size , Only output the parameter quantity as shown in the figure

Or this

, And every one saved at this time epoch Of pt The file is not just the size of the parameter weight , But the whole checkpoint, Including optimizer and so on .
Therefore, I was also distressed for a long time to find a method to estimate the size of the model through the parameter quantity , Or formula , After I compare the calculation results of many models, it is more accurate ,
Parameter conversion formula : Parameter quantity * 2 / 1024 /1024（ One float16 Occupy 2 Bytes , 1024 Bytes to 1k）