If I use a VSSM block in a model, how can I quickly calculate the FLOPs?

[batman47steam]

I modified the code in flops.py and commented out the patch_embed in backbone_vssm in vmamba.py to calculate the FLOPs for a single VSSM block.

Will this give an accurate result for the FLOPs of a single VSSM block?

the output

[MzeroMiko]

I think its better using the functions FLOPs.fvcore_flop_count to count the whole new model you build，with unsupported operations printed.

FLOPs.fvcore_flop_count can be seen as an extended version of fvcore.flops_count which supports s6 operations in ss2d blocks.

[batman47steam]

Thank you for your reply; that was very helpful. I'd like to ask further about the impact of those unsupported operations on the final results. Can they be ignored?

[MzeroMiko]

For vmamba specific，the selective_scan operation is supported by our own implemented flops_count (so it should not appear in the unsupported list), the crossscantritonF is just a copy operation， so it has no effect on flops. For other operations，you can refer to networks without ss2d.