nikitadurasov / masksembles

Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).
https://www.norange.io/projects/masksembles/
MIT License
99 stars 15 forks source link

Fixed mask generation wrapper with binary search #13

Closed ymetz closed 1 year ago

ymetz commented 1 year ago

Look for a scale factor with binary search, works way more reliable for many scale factors and layer sizes

nikitadurasov commented 1 year ago

Looks good, thanks for fixes!