About FLOPs

[ncTimTang]

Thanks for your impressive work. I want to follow your work, but I encounter a problem. How can I calculate the FLOPs of Vim and VideoMamba? Can you share the script with us? Thank you very much!

[Andy1621]

Please check the __main__ in model files.

[alpacaduby]

Thank you for sharing your work.

I also met the problem, as when I was using the float counting script in the __main__ function, and I tried to print out the unsupported operations, I found the PythonOp.MambaInnerFnNoOutProj in it. Is this normal? Would that mean that when counting flops, the mamba module has not been included?

Thanks.

Where I modified:

The output

[Andy1621]

Hi! I'm not sure whether it's normal. And I suggest you follow VMamba for some insights about calculating the GFLOPs manually.