Package ai.djl.pytorch.zoo

Class PtModelZoo

java.lang.Object

ai.djl.repository.zoo.ModelZoo

ai.djl.pytorch.zoo.PtModelZoo

public class PtModelZoo
extends ai.djl.repository.zoo.ModelZoo

PtModelZoo is a repository that contains all PyTorch models in PtSymbolBlock for DJL.

Field Summary

Fields

Modifier and Type	Field	Description
static final String	GROUP_ID

Method Summary

Modifier and Type	Method	Description
String	getGroupId()
Set<String>	getSupportedEngines()

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

addModel, addModel, getModelLoader, getModelLoaders, getModelZoo, hasModelZoo, listModels, listModels, listModels, listModelZoo, loadModel, registerModelZoo, setModelZooResolver

Methods inherited from class java.lang.Object

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

Field Details

GROUP_ID

public static final String GROUP_ID

See Also:

• Constant Field Values

Method Details

getGroupId

public String getGroupId()

Specified by:

getGroupId in class ai.djl.repository.zoo.ModelZoo

getSupportedEngines

public Set<String> getSupportedEngines()

Specified by:

getSupportedEngines in class ai.djl.repository.zoo.ModelZoo