org.deeplearning4j.optimize.solvers.accumulation

Interface GradientsAccumulator

All Superinterfaces:

Serializable

All Known Implementing Classes:

BasicGradientsAccumulator, EncodedGradientsAccumulator

public interface GradientsAccumulator
extends Serializable

Author:

[email protected]

Method Summary

Modifier and Type	Method and Description
void	applyUpdate(StepFunction function, org.nd4j.linalg.api.ndarray.INDArray params, org.nd4j.linalg.api.ndarray.INDArray updates) This method applies accumulated updates via given StepFunction
void	applyUpdate(StepFunction function, org.nd4j.linalg.api.ndarray.INDArray params, org.nd4j.linalg.api.ndarray.INDArray updates, double alpha) This method applies accumulated updates via given StepFunction
void	receiveUpdate(org.nd4j.linalg.api.ndarray.INDArray array) This method accepts updates suitable for StepFunction and puts them to the queue, which is used in backpropagation loop PLEASE NOTE: array is expected to be ready for use and match params dimensionality
void	reset() This method resets all accumulated updates (if any)
void	setExternalSource(Queue<org.nd4j.linalg.api.ndarray.INDArray> source) This method allows to pass external updates to accumulator, they will be populated across all workers using this GradientsAccumulator instance
void	storeUpdate(org.nd4j.linalg.api.ndarray.INDArray array) This method accepts updates suitable for StepFunction, and accumulates/propagates it across all workers
void	touch() This method does initialization of given worker wrt Thread-Device Affinity

Method Detail

setExternalSource

void setExternalSource(Queue<org.nd4j.linalg.api.ndarray.INDArray> source)

This method allows to pass external updates to accumulator, they will be populated across all workers using this GradientsAccumulator instance

Parameters:

source -

applyUpdate

void applyUpdate(StepFunction function,
                 org.nd4j.linalg.api.ndarray.INDArray params,
                 org.nd4j.linalg.api.ndarray.INDArray updates)

This method applies accumulated updates via given StepFunction

Parameters:

function -

params -

applyUpdate

void applyUpdate(StepFunction function,
                 org.nd4j.linalg.api.ndarray.INDArray params,
                 org.nd4j.linalg.api.ndarray.INDArray updates,
                 double alpha)

This method applies accumulated updates via given StepFunction

Parameters:

function -

params -

storeUpdate

void storeUpdate(org.nd4j.linalg.api.ndarray.INDArray array)

This method accepts updates suitable for StepFunction, and accumulates/propagates it across all workers

Parameters:

array -

receiveUpdate

void receiveUpdate(org.nd4j.linalg.api.ndarray.INDArray array)

This method accepts updates suitable for StepFunction and puts them to the queue, which is used in backpropagation loop PLEASE NOTE: array is expected to be ready for use and match params dimensionality

Parameters:

array -

reset

void reset()

This method resets all accumulated updates (if any)

touch

void touch()

This method does initialization of given worker wrt Thread-Device Affinity