Skip to content

SGD

Bases: Optimizer

Stochastic Gradient Descent optimizer.

Updates parameters using gradients and optional momentum.

Parameters:

Name Type Description Default
parameters

Trainable model parameters.

required
learning_rate

Step size used for parameter updates.

0.001
momentum_rate

Momentum factor used to smooth updates.

0.9