SGD

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

Stochastic gradient descent with momentum

setup(policy: maze.core.agent.torch_policy.TorchPolicy) → None

(overrides Optimizer)

prepare optimizer for training