search

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework
MIT License
2.82k stars 307 forks source link

Add support of MultiheadAttention #68

Closed sovrasov closed 3 years ago

sovrasov commented 3 years ago

resolves #65