Open dazz-debug56 opened 1 month ago
Hi dazz-debug56,
Thank you for your interest. We used the thop library to calculate the FLOPs. Below is the code snippet we used:
from thop import profile
flops, param = profile(generator, inputs=(noisy_mag, noisy_pha))
print('FLOPs = {:.3f} GFLOPs'.format(flops / 1e9))
This calculation was performed with an input audio length of 2 seconds and a batch size of 1 on a single GPU.
Best regards, Roy Chao
Thank you for the great work.
I would like to ask how did you compute the FLOPs of the model and whether you can share the method?
fvcore doesn't seem to work here due to the Mamba blocks.