mlx.optimizers.linear_schedule

linear_schedule(init: float, end: float, steps: int) → Callable

创建一个线性调度器。

参数:

• init (float) – 初始值。

• end (float) – 最终值。

• steps (int) – 应用调度器的步数。对于超过 steps 的任何步数，值都为 end。

示例

>>> lr_schedule = optim.linear_schedule(0, 1e-1, 100)
>>> optimizer = optim.Adam(learning_rate=lr_schedule)
>>> optimizer.learning_rate
array(0.0, dtype=float32)
>>> for _ in range(101): optimizer.update({}, {})
...
>>> optimizer.learning_rate
array(0.1, dtype=float32)