pypomp.core.learning_rate.LearningRate.linear_decay

LearningRate.linear_decay(final_factor: float, M: int) → LearningRate

Apply a linear decay schedule from 1.0 down to final_factor.

Parameters:

• final_factor (float) – The factor to reach at the end of the schedule (between 0 and 1).

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

Returns:

A new LearningRate object with linear decay applied.

Return type:

LearningRate