search

CrystalSixone / VLN-GOAT

Repository for Vision-and-Language Navigation via Causal Learning (Accepted by CVPR 2024)
Apache License 2.0
23 stars 4 forks source link

GFLOPs calculation #5

Closed eazook10003 closed 1 month ago

eazook10003 commented 1 month ago

I want to say that your work is amazing and has been incredibly helpful! I’m trying to calculate GFLOPs for HAMT(VLN) but it seems to be not straight forward where it consists with multiple transformers(Should I sum the GFLOPs for each transformer to get the total GFLOPs?). It would be helpful if you could provide the details about FLOPs calculation.

CrystalSixone commented 1 month ago

Hi! First of all, thank you very much for acknowledging our work.

I have updated the code for computing model efficiency, including the number of parameters and GFLOPs. You can calculate these metrics using the following command:

cd map_nav_src
bash scripts/model_efficiency_count.sh

Please note that we computed only a single step for each forward pass. I hope this helps you in computing GFLOPs for other similar methods.

eazook10003 commented 1 month ago

Thank you so much!