Package org.deeplearning4j.nn.updater
Class LayerUpdater
- java.lang.Object
-
- org.deeplearning4j.nn.updater.BaseMultiLayerUpdater<Layer>
-
- org.deeplearning4j.nn.updater.LayerUpdater
-
- All Implemented Interfaces:
Serializable
,Updater
public class LayerUpdater extends BaseMultiLayerUpdater<Layer>
- See Also:
- Serialized Form
-
-
Field Summary
-
Fields inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater
gradientsForMinibatchDivision, initializedMinibatchDivision, layersByName, network, updaterBlocks, updaterStateViewArray
-
-
Constructor Summary
Constructors Constructor Description LayerUpdater(Layer layer)
LayerUpdater(Layer layer, INDArray updaterState)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
getFlattenedGradientsView()
protected Trainable[]
getOrderedLayers()
protected INDArray
getParams()
protected boolean
isMiniBatch()
protected boolean
isSingleLayerUpdater()
-
Methods inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater
divideByMinibatch, equals, getMinibatchDivisionSubsets, getStateViewArray, getStateViewArrayCopy, hashCode, preApply, setStateViewArray, setStateViewArray, update, update
-
-
-
-
Method Detail
-
getOrderedLayers
protected Trainable[] getOrderedLayers()
- Specified by:
getOrderedLayers
in classBaseMultiLayerUpdater<Layer>
- 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
public INDArray getFlattenedGradientsView()
- Specified by:
getFlattenedGradientsView
in classBaseMultiLayerUpdater<Layer>
- Returns:
- The flattened gradient view array for the model
-
getParams
protected INDArray getParams()
- Specified by:
getParams
in classBaseMultiLayerUpdater<Layer>
- Returns:
- The flattened parameter array for the model
-
isMiniBatch
protected boolean isMiniBatch()
- Specified by:
isMiniBatch
in classBaseMultiLayerUpdater<Layer>
- Returns:
- True if the configuration for the model is set to minibatch (divide by minibatch size), false otherwise
-
isSingleLayerUpdater
protected boolean isSingleLayerUpdater()
- Overrides:
isSingleLayerUpdater
in classBaseMultiLayerUpdater<Layer>
-
-