ai.djl.pytorch.engine

Class PtModel

java.lang.Object

ai.djl.BaseModel

ai.djl.pytorch.engine.PtModel

All Implemented Interfaces:

ai.djl.Model, java.lang.AutoCloseable

public class PtModel
extends ai.djl.BaseModel

PtModel is the PyTorch implementation of Model.

PtModel contains all the methods in Model to load and process a model. In addition, it provides PyTorch Specific functionality

Field Summary

Fields inherited from class ai.djl.BaseModel

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

Method Summary

Modifier and Type	Method and Description
void	cast(ai.djl.ndarray.types.DataType dataType)
void	close()
java.lang.String[]	getArtifactNames()
void	load(java.nio.file.Path modelPath, java.lang.String prefix, java.util.Map<java.lang.String,java.lang.Object> options)
<I,O> ai.djl.inference.Predictor<I,O>	newPredictor(ai.djl.translate.Translator<I,O> translator)
ai.djl.training.Trainer	newTrainer(ai.djl.training.TrainingConfig trainingConfig)

Methods inherited from class ai.djl.BaseModel

describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getBlock, getDataType, getName, getNDManager, getProperty, paramPathResolver, readParameters, save, setBlock, setDataType, setModelDir, setProperty

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.Model

load, load, newInstance, newInstance, newInstance, newInstance, quantize

Method Detail

load

public void load(java.nio.file.Path modelPath,
                 java.lang.String prefix,
                 java.util.Map<java.lang.String,java.lang.Object> options)
          throws java.io.IOException,
                 ai.djl.MalformedModelException

Throws:

java.io.IOException

ai.djl.MalformedModelException

newTrainer

public ai.djl.training.Trainer newTrainer(ai.djl.training.TrainingConfig trainingConfig)

newPredictor

public <I,O> ai.djl.inference.Predictor<I,O> newPredictor(ai.djl.translate.Translator<I,O> translator)

getArtifactNames

public java.lang.String[] getArtifactNames()

cast

public void cast(ai.djl.ndarray.types.DataType dataType)

close

public void close()