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

Release newer version #14

Open david-andrew opened 2 years ago

david-andrew commented 2 years ago

Hi,

I was hoping to to find out if it would be possible to release a newer version. I'm mainly interested in this fix right here: https://github.com/1adrianb/pytorch-estimate-flops/commit/0ea1746177e42e1fbbe5c881bebd70cb849aef95. Using the current version, it is impossible to use this tool with completely silent output.

Thanks!

1adrianb commented 2 years ago

Hi, apologies for this, I will push a new version tomorrow, in the meantime it could be pip installed from git directly.

david-andrew commented 2 years ago

Awesome, thanks for the quick response!