ai.djl.training

Class LocalParameterServer

java.lang.Object

ai.djl.training.LocalParameterServer

All Implemented Interfaces:

ParameterServer, java.lang.AutoCloseable

public class LocalParameterServer
extends java.lang.Object
implements ParameterServer

LocalParameterServer is an implementation of the ParameterServer interface.

Constructor Summary

Constructors

Constructor and Description
LocalParameterServer(Optimizer optimizer) Create a new instance of LocalParameterServer for the given optimizer.

Method Summary

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

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.training.ParameterServer

update

Constructor Detail

LocalParameterServer

public LocalParameterServer(Optimizer optimizer)

Create a new instance of LocalParameterServer for the given optimizer.

Parameters:

optimizer - an optimizer

Method Detail

init

public void init(java.lang.String parameterId,
                 NDArray[] value)

Initializes the ParameterStore for the given parameter.

Specified by:

init in interface ParameterServer

Parameters:

parameterId - the parameter ID

value - the values to be set for the given parameter

update

public void update(java.lang.String parameterId,
                   NDArray[] grads,
                   NDArray[] params)

Updates the parameter of a key from Parameter Server.

Specified by:

update in interface ParameterServer

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

public void close()

Specified by:

close in interface ParameterServer

Specified by:

close in interface java.lang.AutoCloseable