search

ChenhongyiYang / PlainMamba

[BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition
Apache License 2.0
68 stars 7 forks source link

Inquiry about the FLOPs computation #8

Closed xinyuliu-jeffrey closed 1 month ago

xinyuliu-jeffrey commented 1 month ago

Dear authors,

Thank you for your work! I have some questions regarding the FLOPs computation.

  1. I am wondering why the d_inner is also multiplied in the flops computation here: https://github.com/ChenhongyiYang/PlainMamba/blob/bac7a7bca3536db1e931ad772f78faec618a98d0/tools/analysis_tools/get_flops.py#L35 As the original mamba authors stated that the computation for a single scan is just 9LN.

  2. I wonder what is the additional computation here: https://github.com/ChenhongyiYang/PlainMamba/blob/bac7a7bca3536db1e931ad772f78faec618a98d0/tools/analysis_tools/get_flops.py#L36

Thank you for you time!

xinyuliu-jeffrey commented 1 month ago

I've figure it out by myself. Never mind. Thanks again for your work!