public interface GradientsAccumulator extends Serializable
Modifier and Type | Method and Description |
---|---|
void |
applyUpdate(StepFunction function,
INDArray params,
INDArray updates)
This method applies accumulated updates via given StepFunction
|
void |
applyUpdate(StepFunction function,
INDArray params,
INDArray updates,
double alpha)
This method applies accumulated updates via given StepFunction
|
void |
receiveUpdate(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<INDArray> source)
This method allows to pass external updates to accumulator, they will be populated across all workers using this GradientsAccumulator instance
|
void |
storeUpdate(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
|
void setExternalSource(Queue<INDArray> source)
source
- void applyUpdate(StepFunction function, INDArray params, INDArray updates)
function
- params
- void applyUpdate(StepFunction function, INDArray params, INDArray updates, double alpha)
function
- params
- void storeUpdate(INDArray array)
array
- void receiveUpdate(INDArray array)
array
- void reset()
void touch()
Copyright © 2018. All rights reserved.