search

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 a bug where max() was used and overwritten #14

Closed ulriks9 closed 6 months ago

ulriks9 commented 8 months ago

I tried to run the code for myself by just importing generation_wrapper() directly. With this I got the following error: "UnboundLocalError: local variable 'max' referenced before assignment".

I fixed it using the code provided. I just thought I would share the fix and you can implement it if you want :)