search

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

Closed ulriks9 closed 9 months ago

ulriks9 commented 11 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 :)