public class MxParameterServer extends NativeResource implements ai.djl.training.ParameterServer
MxParameterServer is the MXNet implementation of ParameterServer.handle| Constructor and Description |
|---|
MxParameterServer(ai.djl.training.optimizer.Optimizer optimizer)
Constructs a new
MxParameterServer. |
| Modifier and Type | Method and Description |
|---|---|
void |
close() |
void |
init(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] values) |
void |
pull(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] weights,
int priority) |
void |
push(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] grads,
int priority) |
finalize, getHandle, getUid, isReleasedpublic MxParameterServer(ai.djl.training.optimizer.Optimizer optimizer)
MxParameterServer.optimizer - the optimizer to use for the parameter server updatespublic void init(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] values)
init in interface ai.djl.training.ParameterServerpublic void push(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] grads,
int priority)
push in interface ai.djl.training.ParameterServerpublic void pull(java.lang.String parameterId,
ai.djl.ndarray.NDArray[] weights,
int priority)
pull in interface ai.djl.training.ParameterServerpublic void close()
close in interface ai.djl.training.ParameterServerclose in interface java.lang.AutoCloseableclose in class NativeResource