optax.bias_correction

optax.bias_correction#

optax.bias_correction(
moment,
decay,
count,
)[source]#

Performs bias correction. It becomes a no-op as count goes to infinity.