Deep Java Library 0.25.0 - pytorch-model-zoo API
This document is the API specification for the Deep Java Library (DJL) PyTorch Model Zoo.
The PyTorch Model Zoo module contains PyTorch pretrained traced modules that can be used for inference. See here for more details.
Package | Description |
---|---|
ai.djl.pytorch.zoo |
Contains the built-in
PtModelZoo . |
ai.djl.pytorch.zoo.cv.objectdetection |
Contains classes for the
Application.CV.OBJECT_DETECTION models in the PtModelZoo . |
ai.djl.pytorch.zoo.nlp.qa |
Contains classes for the
Application.NLP.QUESTION_ANSWER models in the PtModelZoo . |
ai.djl.pytorch.zoo.nlp.sentimentanalysis |
Contains classes for the
Application.NLP.SENTIMENT_ANALYSIS models in the PtModelZoo . |
ai.djl.pytorch.zoo.nlp.textgeneration |
Contains classes for the
Application.NLP.TEXT_GENERATION models. |