Package 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	Description
INDArray	getStateViewArray()
void	setStateViewArray(Trainable layer, INDArray viewArray, boolean initialize)	Set the internal (historical) state view array for this updater
void	update(Trainable layer, Gradient gradient, int iteration, int epoch, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)	Updater: updates the model

Method Detail

setStateViewArray

void setStateViewArray(Trainable layer,
                       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

INDArray getStateViewArray()

Returns:

the view array for this updater

update

void update(Trainable layer,
            Gradient gradient,
            int iteration,
            int epoch,
            int miniBatchSize,
            LayerWorkspaceMgr workspaceMgr)

Updater: updates the model

Parameters:

layer -

gradient -

iteration -