Saving and Loading Models

You can use the BaseModel.save() and load() methods to serialize and deserialize trained models. Note that if you use adapters or only train a subset of layers using the num_layers_trained configuration flag, your save files will be much smaller. This is because Finetune only saves out weights that are different than the default weights of the base model. If you are trying to optimize for save file size, check out Using Adapters and the DeploymentModel class.

model = Classifier()
model.fit(train_data, train_labels)
model.save(filepath)

model = Classifier.load(filepath)