jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/optax
scaleByLearningRate
Function scaleByLearningRate
scaleByLearningRate
(
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
,
flipSign
?:
boolean
,
)
:
GradientTransformation
Scale by the (negative) learning rate (either as scalar or as schedule).
Parameters
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
flipSign
:
boolean
= true
Returns
GradientTransformation
jax-js
Loading...
Scale by the (negative) learning rate (either as scalar or as schedule).