org.deeplearning4j.nn.updater

Class MultiLayerUpdater

java.lang.Object

org.deeplearning4j.nn.updater.BaseMultiLayerUpdater<MultiLayerNetwork>

org.deeplearning4j.nn.updater.MultiLayerUpdater

All Implemented Interfaces:

Serializable, Updater

public class MultiLayerUpdater
extends BaseMultiLayerUpdater<MultiLayerNetwork>

MultiLayerUpdater: Gradient updater for MultiLayerNetworks. Expects backprop gradients for all layers to be in single Gradient object, keyed by "0_b", "1_w" etc., as per MultiLayerNetwork.backward()

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater

layersByName, network, updaterBlocks, updaterStateViewArray

Constructor Summary

Constructors

Constructor and Description
MultiLayerUpdater(MultiLayerNetwork network)
MultiLayerUpdater(MultiLayerNetwork network, org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Summary

Modifier and Type	Method and Description
Updater	clone()
protected org.nd4j.linalg.api.ndarray.INDArray	getFlattenedGradientsView()
protected Layer[]	getOrderedLayers()
protected org.nd4j.linalg.api.ndarray.INDArray	getParams()
protected boolean	isMiniBatch()

Methods inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater

equals, getStateViewArray, hashCode, isSingleLayerUpdater, preApply, setStateViewArray, setStateViewArray, update, update

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

MultiLayerUpdater

public MultiLayerUpdater(MultiLayerNetwork network)

MultiLayerUpdater

public MultiLayerUpdater(MultiLayerNetwork network,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Detail

getOrderedLayers

protected Layer[] getOrderedLayers()

Specified by:

getOrderedLayers in class BaseMultiLayerUpdater<MultiLayerNetwork>

Returns:

Array of layers, in the correct order (i.e., same order as the parameter/gradient/updater flattening order - input to output for MultiLayerNetwork, or topological order for ComputationGraph)

getFlattenedGradientsView

protected org.nd4j.linalg.api.ndarray.INDArray getFlattenedGradientsView()

Specified by:

getFlattenedGradientsView in class BaseMultiLayerUpdater<MultiLayerNetwork>

Returns:

The flattened gradient view array for the model

getParams

protected org.nd4j.linalg.api.ndarray.INDArray getParams()

Specified by:

getParams in class BaseMultiLayerUpdater<MultiLayerNetwork>

Returns:

The flattened parameter array for the model

isMiniBatch

protected boolean isMiniBatch()

Specified by:

isMiniBatch in class BaseMultiLayerUpdater<MultiLayerNetwork>

Returns:

True if the configuration for the model is set to minibatch (divide by minibatch size), false otherwise

clone

public Updater clone()

Overrides:

clone in class Object