Does it support non-image models?
SingCheng opened this issue · comments
PoSingCheng commented
Hi everyone,
I just wondering is it able to calculate the numbers of parameters or Flops for NLP tasks model.
One of the examples:
https://github.com/hangyang-nlp/de-ppn
Ligeng Zhu commented
It would be able to do it. Let me prepare a patch.