Flop/Memory Evaluation Code

uncbiag / SimpleClick

SimpleClick: Interactive Image Segmentation with Simple Vision Transformers (ICCV 2023)

MIT License

209 stars 32 forks source link

Flop/Memory Evaluation Code #14

Closed Mingyuan1997 closed 1 year ago

Mingyuan1997 commented 1 year ago

Thanks for your contribution!

Could you also share FLOP/mem evaluation code about the transformer based model? It seems that there is not a common and standardlized measurement library for that.

Best

qinliuliuqin commented 1 year ago

Hi, thanks for your questions.

FLOPs: https://github.com/uncbiag/SimpleClick/blob/v1.0/notebooks/analyze_effiency.ipynb

Mem: https://github.com/uncbiag/SimpleClick/blob/4099621fcfa99f2cc8a4409ea61420e016ec7920/scripts/evaluate_model.py#L142