pypomp.core.learning_rate.LearningRate.geometric_decay

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

Apply a geometric decay schedule: eta_t = eta_0 * (decay_rate ^ t).

Parameters:

  • decay_rate (float) – The decay rate per iteration (between 0 and 1).

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

Returns:

A new LearningRate object with geometric decay applied.

Return type:

LearningRate