keraflow.optimizers.SGD Class Reference

Stochastic gradient descent, with support for momentum, learning rate decay, and Nesterov momentum. More...

Inheritance diagram for keraflow.optimizers.SGD:

Public Member Functions
def	__init__
Public Member Functions inherited from keraflow.optimizers.Optimizer
def	__init__

Detailed Description

Stochastic gradient descent, with support for momentum, learning rate decay, and Nesterov momentum.

Constructor & Destructor Documentation

def keraflow.optimizers.SGD.__init__	(		self,
			lr = 0.01,
			momentum = 0.,
			decay = 0.,
			nesterov = False,
			kwargs
	)

Parameters

lr	float >= 0. Learning rate.
momentum	float >= 0. Parameter updates momentum.
decay	float >= 0. Learning rate decay over each update.
nesterov	boolean. Whether to apply Nesterov momentum.
kwargs	see Optimizer.__init__