jax-js-nonconsuming (Fork)
System
Light
Dark
Search…
Preparing search index...
@hamk-uas/jax-js-nonconsuming-optax
sgd
Function sgd
sgd
(
learningRate
:
number
|
(
(
count
:
Array
)
=>
Array
)
,
opts
?:
SgdOptions
,
)
:
GradientTransformation
Stochastic gradient descent.
Parameters
learningRate
:
number
|
(
(
count
:
Array
)
=>
Array
)
opts
:
SgdOptions
= {}
Returns
GradientTransformation
jax-js-nonconsuming (Fork)
Loading...
Stochastic gradient descent.