Package ai.djl.training
Class LocalParameterServer
java.lang.Object
ai.djl.training.LocalParameterServer
- All Implemented Interfaces:
ParameterServer
,AutoCloseable
LocalParameterServer
is an implementation of the ParameterServer
interface.-
Constructor Summary
ConstructorDescriptionLocalParameterServer
(Optimizer optimizer) Create a new instance ofLocalParameterServer
for the given optimizer. -
Method Summary
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 Details
-
LocalParameterServer
Create a new instance ofLocalParameterServer
for the given optimizer.- Parameters:
optimizer
- an optimizer
-
-
Method Details
-
init
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
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 interfaceAutoCloseable
- Specified by:
close
in interfaceParameterServer
-