ai.djl.mxnet.engine

Class MxParameterServer

java.lang.Object

ai.djl.util.NativeResource<com.sun.jna.Pointer>

ai.djl.mxnet.engine.MxParameterServer

All Implemented Interfaces:

ai.djl.training.ParameterServer, java.lang.AutoCloseable

public class MxParameterServer
extends ai.djl.util.NativeResource<com.sun.jna.Pointer>
implements ai.djl.training.ParameterServer

MxParameterServer is the MXNet implementation of ParameterServer.

Field Summary

Fields inherited from class ai.djl.util.NativeResource

handle

Constructor Summary

Constructors

Constructor and Description
MxParameterServer(ai.djl.training.optimizer.Optimizer optimizer) Constructs a new MxParameterServer.

Method Summary

Modifier and Type	Method and Description
void	close()
void	init(java.lang.String parameterId, ai.djl.ndarray.NDArray[] values)
void	update(java.lang.String parameterId, ai.djl.ndarray.NDArray[] grads, ai.djl.ndarray.NDArray[] params)

Methods inherited from class ai.djl.util.NativeResource

getHandle, getUid, isReleased

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

MxParameterServer

public MxParameterServer(ai.djl.training.optimizer.Optimizer optimizer)

Constructs a new MxParameterServer.

Parameters:

optimizer - the optimizer to use for the parameter server updates

Method Detail

init

public void init(java.lang.String parameterId,
                 ai.djl.ndarray.NDArray[] values)

Specified by:

init in interface ai.djl.training.ParameterServer

update

public void update(java.lang.String parameterId,
                   ai.djl.ndarray.NDArray[] grads,
                   ai.djl.ndarray.NDArray[] params)

Specified by:

update in interface ai.djl.training.ParameterServer

close

public void close()

Specified by:

close in interface ai.djl.training.ParameterServer

Specified by:

close in interface java.lang.AutoCloseable

Overrides:

close in class ai.djl.util.NativeResource<com.sun.jna.Pointer>