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
     

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

    DJL_REPO_URL

  • 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

  • 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