nikitadurasov / masksembles

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

Fixed mask generation wrapper with binary search #13

Closed ymetz closed 9 months ago

ymetz commented 9 months ago

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

nikitadurasov commented 9 months ago

Looks good, thanks for fixes!