FLOPs Calculation of espnet model

[bobbiaditya]

Hi, is there any function that available in espnet to see the FLOPs calculation of a model?

Thanks!

[sw005320]

@pyf98, can you answer it?
You seem to have some tools https://github.com/pyf98/espnet_utils

[pyf98]

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.

[bobbiaditya]

thanks for the answer, I will try to use the tools provided