Add simple inference FLOP counter to `calc_transformer_flops.py`#31

Merged

Quentin-Anthony merged 4 commits intomainfrom simple-inference-flops

Feb 20, 2024

Commits on Feb 19, 2024

add --infer arg to flops calculator
haileyschoelkopf
committed

Commits on Feb 20, 2024