normal-computing / posteriors

Uncertainty quantification with PyTorch
https://normal-computing.github.io/posteriors/
Apache License 2.0
314 stars 12 forks source link

Add support for `Callable` learning rates and temperature #69

Open SamDuffield opened 5 months ago

SamDuffield commented 5 months ago

Would be very useful to be able to support lr = lambda i: 1e-3 * i ** -0.5 to support Robbins-Monro style scheduling (and others)

Up for debate but could be useful to add this support for the temperature parameter in the spirit of simulated annealing