EXTD_Pytorch icon indicating copy to clipboard operation
EXTD_Pytorch copied to clipboard

the size of Params

Open likesundayl opened this issue 6 years ago • 1 comments

Thank you for your codes. But I have a question. I have trained EXTD-64, I got the params of 2.71M, much larger than 0.16M in the paper, I want to know if you use some trick, thank you!

likesundayl avatar Sep 25 '19 09:09 likesundayl

@likesundayl Hi! I counted the parameter size with a code line in train.py line 159, and I think this is the typical way to measure the param size.

ps: If the num param is 2.71M then the model size will be over 10Mb ~= 2.71M*32bit. But, as you can see that our pth file for EXTD 64 is about 700Kb.

yjyoo3312 avatar Oct 26 '19 06:10 yjyoo3312