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.