Package ai.djl.mxnet.engine

Class MxEngine

java.lang.Object
ai.djl.engine.Engine
ai.djl.mxnet.engine.MxEngine
public final class MxEngine extends ai.djl.engine.Engine
The MxEngine is an implementation of the Engine based on the Apache MXNet Deep Learning Framework.

To get an instance of the MxEngine when it is not the default Engine, call Engine.getEngine(String) with the Engine name "MXNet".

  • Field Details

  • Method Details

    • getAlternativeEngine

      public ai.djl.engine.Engine getAlternativeEngine()
      Specified by:
      getAlternativeEngine in class ai.djl.engine.Engine

    • getEngineName

      public String getEngineName()
      Specified by:
      getEngineName in class ai.djl.engine.Engine

    • getRank

      public int getRank()
      Specified by:
      getRank in class ai.djl.engine.Engine

    • getVersion

      public String getVersion()
      Specified by:
      getVersion in class ai.djl.engine.Engine

    • hasCapability

      public boolean hasCapability(String capability)
      Specified by:
      hasCapability in class ai.djl.engine.Engine

    • newSymbolBlock

      public ai.djl.nn.SymbolBlock newSymbolBlock(ai.djl.ndarray.NDManager manager)
      Overrides:
      newSymbolBlock in class ai.djl.engine.Engine

    • newModel

      public ai.djl.Model newModel(String name, ai.djl.Device device)
      Specified by:
      newModel in class ai.djl.engine.Engine

    • newBaseManager

      public ai.djl.ndarray.NDManager newBaseManager()
      Specified by:
      newBaseManager in class ai.djl.engine.Engine

    • newBaseManager

      public ai.djl.ndarray.NDManager newBaseManager(ai.djl.Device device)
      Specified by:
      newBaseManager in class ai.djl.engine.Engine

    • newGradientCollector

      public ai.djl.training.GradientCollector newGradientCollector()
      Overrides:
      newGradientCollector in class ai.djl.engine.Engine

    • newParameterServer

      public ai.djl.training.ParameterServer newParameterServer(ai.djl.training.optimizer.Optimizer optimizer)
      Overrides:
      newParameterServer in class ai.djl.engine.Engine

    • setRandomSeed

      public void setRandomSeed(int seed)
      Overrides:
      setRandomSeed in class ai.djl.engine.Engine

    • toString

      public String toString()
      Overrides:
      toString in class ai.djl.engine.Engine