ai.djl.repository.zoo

Class DefaultModelLoader

java.lang.Object

ai.djl.repository.zoo.BaseModelLoader<NDList,NDList>

ai.djl.repository.zoo.DefaultModelLoader

All Implemented Interfaces:

ModelLoader<NDList,NDList>

public class DefaultModelLoader
extends BaseModelLoader<NDList,NDList>

A ModelLoader loads a particular ZooModel from a local folder.

Field Summary

Fields inherited from class ai.djl.repository.zoo.BaseModelLoader

factories, modelZoo, mrl, repository, version

Constructor Summary

Constructors

Constructor and Description
DefaultModelLoader(Repository repository, MRL mrl) Creates the model loader from the given repository.

Method Summary

Modifier and Type	Method and Description
Application	getApplication() Returns the application of the ModelLoader.
ZooModel<NDList,NDList>	loadModel(java.util.Map<java.lang.String,java.lang.String> filters, Device device, ai.djl.util.Progress progress) Loads the model with the given search filters.

Methods inherited from class ai.djl.repository.zoo.BaseModelLoader

createModel, getArtifactId, listModels, loadModel, match, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface ai.djl.repository.zoo.ModelLoader

loadModel, loadModel, loadModel, loadModel

Constructor Detail

DefaultModelLoader

public DefaultModelLoader(Repository repository,
                          MRL mrl)

Creates the model loader from the given repository.

Parameters:

repository - the repository to load the model from

mrl - the mrl of the model to load

Method Detail

getApplication

public Application getApplication()

Returns the application of the ModelLoader.

Returns:

the application of the ModelLoader

loadModel

public ZooModel<NDList,NDList> loadModel(java.util.Map<java.lang.String,java.lang.String> filters,
                                         Device device,
                                         ai.djl.util.Progress progress)
                                  throws java.io.IOException,
                                         ModelNotFoundException,
                                         MalformedModelException

Loads the model with the given search filters.

Parameters:

filters - the search filters to match against the loaded model

device - the device the loaded model should use

progress - the progress tracker to update while loading the model

Returns:

the loaded model

Throws:

java.io.IOException - for various exceptions loading data from the repository

ModelNotFoundException - if no model with the specified criteria is found

MalformedModelException - if the model data is malformed