How to find flop count of a nemo model?

[afsara-ben]

The NEMO models call a function such as transcribe() to perform ASR. With the existent flop count libraries like thop, fvcore, I cannot obtain the flop count as it requires the input to be passed through the model's forward() function. The following does not work.

from fvcore.nn import FlopCountAnalysis
import nemo.collections.asr as nemo_asr
asr_model = nemo_asr.models.EncDecCTCModelBPE.from_pretrained(model_name="stt_en_conformer_ctc_small")
flops = FlopCountAnalysis(asr_model.transcribe, [audio_path])