Open sei-cabidi opened 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)
max_iter
temperature_scaling.py
optimizer = optim.LBFGS([self.temperature], lr=0.01, max_iter=50)
optimizer = optim.LBFGS([self.temperature], lr=0.01, max_iter=100)
It does not work for me, whatever the max_iter is.
If your ECE is going up, try simply increasing
max_iter
on line 60 oftemperature_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)