public class TfModel
extends ai.djl.BaseModel
| Modifier and Type | Method and Description |
|---|---|
void |
cast(ai.djl.ndarray.types.DataType dataType) |
void |
close() |
java.lang.String[] |
getArtifactNames() |
ai.djl.nn.Block |
getBlock() |
ai.djl.ndarray.NDManager |
getNDManager() |
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) |
void |
save(java.nio.file.Path modelPath,
java.lang.String newModelName) |
void |
setBlock(ai.djl.nn.Block block) |
describeInput, describeOutput, finalize, getArtifact, getArtifact, getArtifactAsStream, getDataType, getModelPath, getName, getProperty, paramPathResolver, readParameters, setDataType, setModelDir, setPropertypublic void load(java.nio.file.Path modelPath,
java.lang.String prefix,
java.util.Map<java.lang.String,java.lang.Object> options)
throws java.io.FileNotFoundException
java.io.FileNotFoundExceptionpublic void save(java.nio.file.Path modelPath,
java.lang.String newModelName)
save in interface ai.djl.Modelsave in class ai.djl.BaseModelpublic ai.djl.nn.Block getBlock()
getBlock in interface ai.djl.ModelgetBlock in class ai.djl.BaseModelpublic void setBlock(ai.djl.nn.Block block)
setBlock in interface ai.djl.ModelsetBlock in class ai.djl.BaseModelpublic ai.djl.training.Trainer newTrainer(ai.djl.training.TrainingConfig trainingConfig)
public <I,O> ai.djl.inference.Predictor<I,O> newPredictor(ai.djl.translate.Translator<I,O> translator)
public ai.djl.ndarray.NDManager getNDManager()
getNDManager in interface ai.djl.ModelgetNDManager in class ai.djl.BaseModelpublic java.lang.String[] getArtifactNames()
public void cast(ai.djl.ndarray.types.DataType dataType)
public void close()