Package ai.djl.training

Class LocalParameterServer

  • All Implemented Interfaces:
    ParameterServer, java.lang.AutoCloseable
    public class LocalParameterServer
extends java.lang.Object
implements ParameterServer
    LocalParameterServer is an implementation of the ParameterServer interface.

    • Constructor Summary

      Constructors 
      Constructor Description
      LocalParameterServer​(Optimizer optimizer)
      Create a new instance of LocalParameterServer for the given optimizer.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void close()
      void init​(java.lang.String parameterId, NDArray[] value)
      Initializes the ParameterStore for the given parameter.
      void update​(java.lang.String parameterId, NDArray[] grads, NDArray[] params)
      Updates the parameter of a key from Parameter Server.

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Constructor Detail

      • LocalParameterServer

        public LocalParameterServer​(Optimizer optimizer)
        Create a new instance of LocalParameterServer for the given optimizer.
        Parameters:
        optimizer - an optimizer

    • Method Detail

      • init

        public void init​(java.lang.String parameterId,
                 NDArray[] value)
        Initializes the ParameterStore for the given parameter.
        Specified by:
        init in interface ParameterServer
        Parameters:
        parameterId - the parameter ID
        value - the values to be set for the given parameter

      • update

        public void update​(java.lang.String parameterId,
                   NDArray[] grads,
                   NDArray[] params)
        Updates the parameter of a key from Parameter Server.
        Specified by:
        update in interface ParameterServer
        Parameters:
        parameterId - the key to identify the parameter
        grads - the gradient NDArrays in different devices to apply the update.
        params - the parameter NDArrays in different devices to be updated.

      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable
        Specified by:
        close in interface ParameterServer