Skip to main content

Module sgd

teeny_kernels::nn::optim

Module sgd

Structs§

SgdMomentumStep: SGD step with momentum (non-Nesterov).
SgdNesterovStep: SGD step with Nesterov momentum.
SgdStep: SGD step (no momentum).

Functions§

sgd_momentum_step: SGD step with momentum (non-Nesterov).
sgd_nesterov_step: SGD step with Nesterov momentum.
sgd_step: SGD step (no momentum).