Package 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, 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
    Description
    MxParameterServer(ai.djl.training.optimizer.Optimizer optimizer)
    Constructs a new MxParameterServer.

  • Method Summary

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

    Methods inherited from class ai.djl.util.NativeResource

    getHandle, getUid, isReleased, onClose

    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

    • 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 Details

    • init

      public void init(String parameterId, ai.djl.ndarray.NDArray[] values)
      Specified by:
      init in interface ai.djl.training.ParameterServer

    • update

      public void update(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 AutoCloseable
      Specified by:
      close in interface ai.djl.training.ParameterServer
      Overrides:
      close in class ai.djl.util.NativeResource<com.sun.jna.Pointer>