org.nd4j.parameterserver

Class ParameterServerListener

java.lang.Object

org.nd4j.parameterserver.ParameterServerListener

All Implemented Interfaces:

NDArrayCallback

public class ParameterServerListener
extends Object
implements NDArrayCallback

Constructor Summary

Constructors

Constructor and Description
ParameterServerListener(int[] shape) Shape of the ndarray
ParameterServerListener(int[] shape, int updatesPerEpoch) Shape of the ndarray
ParameterServerListener(int[] shape, ParameterServerUpdater updater)

Method Summary

Modifier and Type	Method and Description
void	finish() Do a final divide for averaging
void	onNDArray(org.nd4j.linalg.api.ndarray.INDArray arr) Setup an ndarray
void	onNDArrayMessage(NDArrayMessage message) A listener for ndarray message
void	onNDArrayPartial(org.nd4j.linalg.api.ndarray.INDArray arr, long idx, int... dimensions) Used for partial updates using tensor along dimension

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ParameterServerListener

public ParameterServerListener(int[] shape,
                               int updatesPerEpoch)

Shape of the ndarray

Parameters:

shape - the shape of the array

updatesPerEpoch - the number of updates per epoch for synchronization

ParameterServerListener

public ParameterServerListener(int[] shape)

Shape of the ndarray

Parameters:

shape - the shape of the array

ParameterServerListener

public ParameterServerListener(int[] shape,
                               ParameterServerUpdater updater)

Parameters:

shape - the shape of the array

updater - the updater to use for this server

Method Detail

onNDArrayMessage

public void onNDArrayMessage(NDArrayMessage message)

A listener for ndarray message

Specified by:

onNDArrayMessage in interface NDArrayCallback

Parameters:

message - the message for the callback

onNDArrayPartial

public void onNDArrayPartial(org.nd4j.linalg.api.ndarray.INDArray arr,
                             long idx,
                             int... dimensions)

Used for partial updates using tensor along dimension

Specified by:

onNDArrayPartial in interface NDArrayCallback

Parameters:

arr - the array to count as an update

idx - the index for the tensor along dimension

dimensions - the dimensions to act on for the tensor along dimension

onNDArray

public void onNDArray(org.nd4j.linalg.api.ndarray.INDArray arr)

Setup an ndarray

Specified by:

onNDArray in interface NDArrayCallback

Parameters:

arr -

finish

public void finish()

Do a final divide for averaging