search

gpleiss / temperature_scaling

A simple way to calibrate your neural network.
MIT License
1.09k stars 159 forks source link

Tip: increase your max_iter #34

Open sei-cabidi opened 1 year ago

sei-cabidi commented 1 year ago

If your ECE is going up, try simply increasing max_iter on line 60 of temperature_scaling.py before: optimizer = optim.LBFGS([self.temperature], lr=0.01, max_iter=50) after: optimizer = optim.LBFGS([self.temperature], lr=0.01, max_iter=100)

MengyuanChen21 commented 11 months ago

It does not work for me, whatever the max_iter is.