How to get the model's FLOPs and params?

MzeroMiko / VMamba

VMamba: Visual State Space Models，code is based on mamba

MIT License

2.01k stars 115 forks source link

How to get the model's FLOPs and params? #260

Open Seeingsunshion opened 1 month ago

Seeingsunshion commented 1 month ago

Hello, after replacing the other dataset, how can I get and print out the current model's FLOPs and params values after having trained and tested target detection on the model according to configure model according to configs/vssm/mask_rcnn_vssm_fpn_coco_base.py?

MzeroMiko commented 1 month ago

You can refer to the code in flops.py in analyze or mmdetection/tools for details.