Package ai.djl.python.engine
Class PyModel
java.lang.Object
ai.djl.BaseModel
ai.djl.python.engine.PyModel
- All Implemented Interfaces:
ai.djl.Model
,AutoCloseable
public class PyModel
extends ai.djl.BaseModel
PyModel
is the Python engine implementation of Model
.-
Field Summary
Fields inherited from class ai.djl.BaseModel
artifacts, block, dataType, inputData, manager, modelDir, modelName, properties, wasLoaded
-
Method Summary
Methods inherited from class ai.djl.BaseModel
describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getArtifactNames, getBlock, getDataType, getModelPath, getName, getNDManager, getProperties, getProperty, load, newTrainer, paramPathResolver, readParameters, 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, getProperty, intProperty, load, load, load, longProperty, newPredictor, quantize
-
Method Details
-
load
- Throws:
IOException
-
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 interfaceai.djl.Model
- Overrides:
newPredictor
in classai.djl.BaseModel
-
close
public void close()- Specified by:
close
in interfaceAutoCloseable
- Specified by:
close
in interfaceai.djl.Model
- Overrides:
close
in classai.djl.BaseModel
-