search

1adrianb / pytorch-estimate-flops

Estimate/count FLOPS for a given neural network using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
305 stars 22 forks source link

mistake fixed #12

Closed MichaelMonashev closed 2 years ago

1adrianb commented 2 years ago

Thanks