Package ai.djl.training

Interface ParameterServer

All Superinterfaces:

AutoCloseable

All Known Implementing Classes:

LocalParameterServer

public interface ParameterServer
extends AutoCloseable

An interface for a key-value store to store parameters, and their corresponding gradients.

Method Summary

Modifier and Type	Method	Description
void	close()
void	init(String parameterId, NDArray[] value)	Initializes the ParameterStore for the given parameter.
default void	update(String parameterId, NDArray[] params)	Updates the parameter of a key from Parameter Server.
void	update(String parameterId, NDArray[] grads, NDArray[] params)	Updates the parameter of a key from Parameter Server.

Method Details

init

void init(String parameterId,
 NDArray[] value)

Initializes the ParameterStore for the given parameter.

Parameters:

parameterId - the parameter ID

value - the values to be set for the given parameter

update

default void update(String parameterId,
 NDArray[] params)

Updates the parameter of a key from Parameter Server.

Parameters:

parameterId - the key to identify the parameter

params - the parameter NDArrays in different devices to be updated.

update

void update(String parameterId,
 NDArray[] grads,
 NDArray[] params)

Updates the parameter of a key from Parameter Server.

Parameters:

parameterId - the key to identify the parameter

grads - the gradient NDArrays in different devices to apply the update.

params - the parameter NDArrays in different devices to be updated.

close

void close()

Specified by:

close in interface AutoCloseable