how to calculata FLOPS with LSTM? - Githubissues

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework

MIT License

2.83k stars 306 forks source link

how to calculata FLOPS with LSTM? #100

Closed sisongsongxi closed 4 months ago

sovrasov commented 2 years ago

Hi. See https://github.com/sovrasov/flops-counter.pytorch/blob/master/ptflops/pytorch_ops.py#L84-L106 and https://github.com/sovrasov/flops-counter.pytorch/blob/master/ptflops/pytorch_ops.py#L109