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