SGD

class maze.train.trainers.es.optimizers.sgd.SGD(step_size: float, momentum: float = 0.9)

Stochastic gradient descent with momentum

setup(policy: TorchPolicy) → None

(overrides Optimizer)

prepare optimizer for training