File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -109,11 +109,14 @@ def compile(
109109
110110 - For backend JAX:
111111
112- - `linear_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.linear_schedule>`_:
112+ - `linear_schedule
113+ <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.linear_schedule>`_:
113114 ("linear", end_value, transition_steps)
114- - `cosine_decay_schedule <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.cosine_decay_schedule>`_:
115+ - `cosine_decay_schedule
116+ <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.cosine_decay_schedule>`_:
115117 ("cosine", decay_steps, alpha)
116- - `exponential_decay <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.exponential_decay>`_:
118+ - `exponential_decay
119+ <https://optax.readthedocs.io/en/latest/api/optimizer_schedules.html#optax.schedules.exponential_decay>`_:
117120 ("exponential", transition_steps, decay_rate)
118121
119122 loss_weights: A list specifying scalar coefficients (Python floats) to
You can’t perform that action at this time.
0 commit comments