search

Lyken17 / pytorch-OpCounter

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

Add transformer counter to profile #149

Open HaoKang-Timmy opened 3 years ago

HaoKang-Timmy commented 3 years ago
  1. add transformer counter to rnn_hooks.py
  2. provide evaluate_transformer.py as an example
liminn commented 2 years ago

why not merge this pull? It is very necessary to calculate the flops and prams of transformer.

quancs commented 1 year ago

Is there any update now