org.deeplearning4j.nn.api

Interface Updater

All Superinterfaces:

Serializable

All Known Implementing Classes:

BaseMultiLayerUpdater, ComputationGraphUpdater, LayerUpdater, MultiLayerUpdater

public interface Updater
extends Serializable

Update the model

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	getStateViewArray()
void	setStateViewArray(Layer layer, org.nd4j.linalg.api.ndarray.INDArray viewArray, boolean initialize) Set the internal (historical) state view array for this updater
void	update(Layer layer, Gradient gradient, int iteration, int miniBatchSize) Updater: updates the model

Method Detail

setStateViewArray

void setStateViewArray(Layer layer,
                       org.nd4j.linalg.api.ndarray.INDArray viewArray,
                       boolean initialize)

Set the internal (historical) state view array for this updater

Parameters:

layer - Layer that this updater belongs to

viewArray - View array

initialize - Whether to initialize the array or not

getStateViewArray

org.nd4j.linalg.api.ndarray.INDArray getStateViewArray()

Returns:

the view array for this updater

update

void update(Layer layer,
            Gradient gradient,
            int iteration,
            int miniBatchSize)

Updater: updates the model

Parameters:

layer -

gradient -

iteration -