Skip to content

Latest commit

 

History

History
24 lines (16 loc) · 1.36 KB

model-zoo.md

File metadata and controls

24 lines (16 loc) · 1.36 KB

DJL - Model Zoo

Deep Java Library's (DJL) Model Zoo is more than a collection of pre-trained models. It's a bridge between a model vendor and a consumer. It provides a framework for developers to create and publish their own models.

A ZooModel has the following characteristics:

  • Globally unique: similar to Java maven packages, a model has its own group ID and artifact ID that uniquely identify it.
  • Versioned: the model version scheme allows developers to continuously update their model without causing a backward compatibility issue.
  • Ready to use out of box: the model contains predefined pre-process and post-process functionality, which allows the user to run inference with a plain java object.
  • Can be published anywhere: models can be published to an S3 bucket, a web server, or a local folder.

We provide framework agnostic ZooModels in our model zoo. They can be used on any DJL backend engine.

MXNet has a large number of existing pre-trained models. We created an MXNet model zoo to make it easy for users to consume them.

Publish your own model to the model zoo

You can create your own model in the model zoo so customers can easily consume it. For more information, see Add a new Model to the model zoo .