Add learning rate schedules

jeremiecoullon / SGMCMCJax

Lightweight library of stochastic gradient MCMC algorithms written in JAX.

Apache License 2.0

95 stars 8 forks source link

Closed jeremiecoullon closed 3 years ago

jeremiecoullon commented 3 years ago

Add learning rate schedules to the diffusions. Follow how Jax does it for optimisers

jeremiecoullon commented 3 years ago

Added the constant and polynomial schedules.