Package 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 theParameterServer
interface.
-
-
Constructor Summary
Constructors Constructor Description LocalParameterServer(Optimizer optimizer)
Create a new instance ofLocalParameterServer
for the given optimizer.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description void
close()
void
init(java.lang.String parameterId, NDArray[] value)
Initializes theParameterStore
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 ofLocalParameterServer
for the given optimizer.- Parameters:
optimizer
- an optimizer
-
-
Method Detail
-
init
public void init(java.lang.String parameterId, NDArray[] value)
Initializes theParameterStore
for the given parameter.- Specified by:
init
in interfaceParameterServer
- Parameters:
parameterId
- the parameter IDvalue
- 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 interfaceParameterServer
- Parameters:
parameterId
- the key to identify the parametergrads
- 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 interfacejava.lang.AutoCloseable
- Specified by:
close
in interfaceParameterServer
-
-