search

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.
MIT License
4.9k stars 528 forks source link

Does it support non-image models? #195

Open SingCheng opened 1 year ago

SingCheng commented 1 year ago

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

Lyken17 commented 1 year ago

It would be able to do it. Let me prepare a patch.