pypomp.LearningRate.geometric_decay

LearningRate.geometric_decay(decay_rate: float, M: int) → LearningRate

Apply a geometric decay schedule.

The rate at step t is eta_t = eta_0 * (decay_rate ^ t).

Parameters:

• decay_rate (float) – Decay factor per iteration. Must be in the interval \([0, 1]\).

• M (int) – Number of iterations for the schedule.

Returns:

A new learning rate object with geometric decay applied.

Return type:

LearningRate