Sgd

SGD optimizer implementation.

nabla.nn.optim.sgd.sgd_step(*args)

nabla.nn.optim.sgd.init_sgd_state(params)

Initialize SGD momentum states.

Parameters:

params (list[Array]) – List of parameter arrays

Returns:

List of zero-initialized momentum states

Return type:

list[Array]