Package ai.djl.pytorch.zoo

Class 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 java.lang.String GROUP_ID  

    • Constructor Summary

      Constructors 
      Constructor Description
      PtModelZoo()  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      java.lang.String getGroupId()
      java.util.List<ai.djl.repository.zoo.ModelLoader> getModelLoaders()
      java.util.Set<java.lang.String> getSupportedEngines()

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

        getModelLoader, getModelZoo, hasModelZoo, listModels, listModels, listModelZoo, loadModel, registerModelZoo

      • Methods inherited from class java.lang.Object

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

    • Constructor Detail

      • PtModelZoo

        public PtModelZoo()

    • Method Detail

      • getModelLoaders

        public java.util.List<ai.djl.repository.zoo.ModelLoader> getModelLoaders()
        Specified by:
        getModelLoaders in class ai.djl.repository.zoo.ModelZoo

      • getGroupId

        public java.lang.String getGroupId()
        Specified by:
        getGroupId in class ai.djl.repository.zoo.ModelZoo

      • getSupportedEngines

        public java.util.Set<java.lang.String> getSupportedEngines()
        Specified by:
        getSupportedEngines in class ai.djl.repository.zoo.ModelZoo