jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/optax
scaleBySchedule
Function scaleBySchedule
scaleBySchedule
(
stepSizeFn
:
(
count
:
number
)
=>
number
)
:
GradientTransformation
Scale updates using a custom schedule for the step size.
Parameters
stepSizeFn
:
(
count
:
number
)
=>
number
Returns
GradientTransformation
jax-js
Loading...
Scale updates using a custom schedule for the step size.