Optimizers

class tinyml.optims.SGDOptimizer(lr, momentum=None)

In this class, we implement the stochastic gradient descent algorithm, which is used to update the parameters in a neural network. The algorithm is simple:

\[w^{new} = w^{old}-\lambda \nabla\]

where \(\lambda\) is the preset learning rate, and \(\nabla\) is the gradient.