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.