Package ai.djl.training
Interface ParameterServer
-
- All Superinterfaces:
java.lang.AutoCloseable
- All Known Implementing Classes:
LocalParameterServer
public interface ParameterServer extends java.lang.AutoCloseableAn interface for a key-value store to store parameters, and their corresponding gradients.
-
-
Method Summary
All Methods Instance Methods Abstract Methods Default Methods Modifier and Type Method Description voidclose()voidinit(java.lang.String parameterId, NDArray[] value)Initializes theParameterStorefor the given parameter.default voidupdate(java.lang.String parameterId, NDArray[] params)Updates the parameter of a key from Parameter Server.voidupdate(java.lang.String parameterId, NDArray[] grads, NDArray[] params)Updates the parameter of a key from Parameter Server.
-
-
-
Method Detail
-
init
void init(java.lang.String parameterId, NDArray[] value)Initializes theParameterStorefor the given parameter.- Parameters:
parameterId- the parameter IDvalue- the values to be set for the given parameter
-
update
default void update(java.lang.String parameterId, NDArray[] params)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
void update(java.lang.String parameterId, NDArray[] grads, NDArray[] params)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:
closein interfacejava.lang.AutoCloseable
-
-