Export trained PyTorch classification model to TorchScript model
This Python script allows you to export a trained PyTorch classification model to a TorchScript model that can be easily deployed in various runtime environment, with an optimized latency and throughput.
The source code of this sample can be found in
when installing Metavision SDK from installer or packages. For other deployment methods, check the page
Path of Samples.
Compiled TorchScript Model that can be easily deployed during runtime.
Specifically, it will output:
model.ptjit (the model)
info_ssd_jit.json (the hyperparameters used during training)
Setup & requirements
You will need to provide the following input:
path to the checkpoint. You can use``chifoumi_rnn.ckpt`` from our pre-trained models
path to the output directory
How to start
To run the script:
python3 export_classifier.py classifier_model.ckpt /path/to/output
python export_classifier.py classifier_model.ckpt /path/to/output
To find the full list of options, run:
python3 export_classifier.py -h
python export_classifier.py -h