Deep Java Library 0.31.1 - 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.

Packages

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.