search

sovrasov / flops-counter.pytorch

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

Fixes, version update #61

Closed sovrasov closed 3 years ago

sovrasov commented 3 years ago

Fix a hook Linear layer

Resolves #60

sovrasov commented 3 years ago

Resolves #60