Package ai.djl.training
Interface ParameterServer
- All Superinterfaces:
AutoCloseable
- All Known Implementing Classes:
LocalParameterServer
An interface for a key-value store to store parameters, and their corresponding gradients.
-
Method Summary
Modifier and TypeMethodDescriptionvoid
close()
void
Initializes theParameterStore
for the given parameter.default void
Updates the parameter of a key from Parameter Server.void
Updates the parameter of a key from Parameter Server.
-
Method Details
-
init
Initializes theParameterStore
for the given parameter.- Parameters:
parameterId
- the parameter IDvalue
- the values to be set for the given parameter
-
update
Updates the parameter of a key from Parameter Server.- Parameters:
parameterId
- the key to identify the parameterparams
- the parameter NDArrays in different devices to be updated.
-
update
Updates the parameter of a key from Parameter Server.- 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
void close()- Specified by:
close
in interfaceAutoCloseable
-