jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/optax
adamw
Function adamw
adamw
(
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
,
opts
?:
AdamWOptions
,
)
:
GradientTransformation
Adam with weight decay regularization.
Parameters
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
opts
:
AdamWOptions
= {}
Returns
GradientTransformation
jax-js
Loading...
Adam with weight decay regularization.