Package ai.djl.paddlepaddle.engine

Class PpModel

  • All Implemented Interfaces:
    ai.djl.Model, java.lang.AutoCloseable
    public class PpModel
extends ai.djl.BaseModel
    PpModel is the PaddlePaddle implementation of Model.

    • Field Summary

      • Fields inherited from class ai.djl.BaseModel

        artifacts, block, dataType, inputData, manager, modelDir, modelName, properties

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void close()
      void load​(java.nio.file.Path modelPath, java.lang.String prefix, java.util.Map<java.lang.String,​?> options)
      Loads the PaddlePaddle model from a specified location.
      <I,​O>
      ai.djl.inference.Predictor<I,​O>      		 newPredictor​(ai.djl.translate.Translator<I,​O> translator, ai.djl.Device device)

      • Methods inherited from class ai.djl.BaseModel

        describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getArtifactNames, getBlock, getDataType, getModelPath, getName, getNDManager, getProperty, load, newTrainer, paramPathResolver, readParameters, save, setBlock, setDataType, setModelDir, setProperty, toString

      • Methods inherited from class java.lang.Object

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

      • Methods inherited from interface ai.djl.Model

        cast, load, load, load, newPredictor, quantize

    • Method Detail

      • load

        public void load​(java.nio.file.Path modelPath,
                 java.lang.String prefix,
                 java.util.Map<java.lang.String,​?> options)
          throws java.io.IOException
        Loads the PaddlePaddle model from a specified location. 
         Map<String, String> options = new HashMap<>()
 options.put("epoch", "3");
 model.load(modelPath, "squeezenet", options);
        Parameters:
        modelPath - the directory of the model
        prefix - the model file name or path prefix
        options - load model options, see documentation for the specific engine
        Throws:
        java.io.IOException - Exception for file loading

      • newPredictor

        public <I,​O> ai.djl.inference.Predictor<I,​O> newPredictor​(ai.djl.translate.Translator<I,​O> translator,
                                                                      ai.djl.Device device)
        Specified by:
        newPredictor in interface ai.djl.Model
        Overrides:
        newPredictor in class ai.djl.BaseModel

      • close

        public void close()
        Specified by:
        close in interface java.lang.AutoCloseable
        Specified by:
        close in interface ai.djl.Model
        Overrides:
        close in class ai.djl.BaseModel