mlpc-ucsd/CoaT

flops and params for detection tools

Opened this issue · 1 comments

Hi, Thanks for your great work. I wonder would you share the tools for calculating flops and parameters for detection?

Hi @jiangzhengkai, thank you for your interest! We basically used a modified version of the FLOPs calculation script from mmcv, following whai362/PVT#27 (the calculation for attention part is modified accordingly). Parameters are simply counted from the items model.parameters(). Due to some other on-going affairs we have not updated the tools for flops and parameters yet. Stay tuned or you can try to calculating by yourself!