Flop/Memory Evaluation Code
Mingyuan1997 opened this issue · 1 comments
Mingyuan1997 commented
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
Hi, thanks for your questions.
FLOPs:
https://github.com/uncbiag/SimpleClick/blob/v1.0/notebooks/analyze_effiency.ipynb
Mem:
SimpleClick/scripts/evaluate_model.py
Line 142 in 4099621