Model Preparation

In this section we will discuss the steps required to prepare a model for deployment.


Serialization of Trained PyTorch Models

We have prepared a guide on how to serialize trained PyTorch models here.

At the moment, we only support PyTorch models. We are working on adding support for ONNX models as well.

Add a Serialized Model to the Plugin Project

Once you have serialized your model, you can add it to the plugin project using the guide here.


Table of contents