FLOPs Calculation of espnet model #5499
-
Hi, is there any function that available in espnet to see the FLOPs calculation of a model? Thanks! |
Beta Was this translation helpful? Give feedback.
Replies: 3 comments
-
@pyf98, can you answer it? |
Beta Was this translation helpful? Give feedback.
-
As @sw005320 shared, I used DeepSpeed's profiler to calculate FLOPs. You can check that repo for an example usage. Note that if you have some special operations that are not defined in DeepSpeed, it will not be counted. |
Beta Was this translation helpful? Give feedback.
-
thanks for the answer, I will try to use the tools provided |
Beta Was this translation helpful? Give feedback.
@pyf98, can you answer it?
You seem to have some tools https://github.com/pyf98/espnet_utils