optax.ApplyEvery

Contents

optax.ApplyEvery#

class optax.ApplyEvery(count: jax.typing.ArrayLike, grad_acc: optax.Updates)[source]#

Contains a counter and a gradient accumulator.