Exporting a FastAI model
Learn how to export a FastAI model
Last updated 18th August, 2020.
Objective
FastAI is a popular machine learning library supported by OVHcloud ML Serving. This tutorial will cover how to export a FastAI trained model.
Requirements
- A python environment with FastAI installed
Save a simple model
Let's take a simple example of a FastAI model to illustrate:
import os
from fastai.text import *
# Load dataset
path = untar_data(URLs.IMDB_SAMPLE)
df = pd.read_csv(path / 'texts.csv')
data_lm = TextLMDataBunch.from_csv(path, 'texts.csv')
# Initialize classifier
data_clas = TextClasDataBunch.from_csv(path, 'texts.csv', vocab=data_lm.train_ds.vocab, bs=32)
learn = language_model_learner(data_lm, AWD_LSTM, drop_mult=0.5)
# Train ...
# Export trainer
learn.export(os.path.abspath('./my_export.pkl'))
Your model is now serialized on your local file system in the my_export.pkl file.
Going further
- You can check the OVHcloud documentation on how to deploy custom models.
- You can check the supported compatibilities for Tensorflow SavedModel
Did you find this guide useful?
Please feel free to give any suggestions in order to improve this documentation.
Whether your feedback is about images, content, or structure, please share it, so that we can improve it together.
Your support requests will not be processed via this form. To do this, please use the "Create a ticket" form.
Thank you. Your feedback has been received.