Hi there, I wanted to create a custom model that includes a transformer and save it using the save_pretrained function after training for a few epochs. I would then want to load it in a different notebook using the from_pretrained function for inference. Suppose I follow this guide and created a c…

Yes you can inherit from PreTrainedModel to inherit methods like from_pretrained, save_pretrained and push_to_hub. Alternatively, you can leverage the PyTorchModelHubMixin class available in the huggingface_hub library. This allows you to get the same functionality: from torch import nn from huggi…

How to load a pretrained custom model using `from_pretrained`

Beginners

luozi March 16, 2023, 7:32am 4

nice，thank

Topic		Replies	Views
Load Custom Model Intermediate	8	1436	November 21, 2022
Saving/Loading custom model build from varying HF models Intermediate	1	1385	March 20, 2023
Custom GPT2 Model won't load after training Intermediate	1	1176	September 15, 2021
Elegant way to load and save a pretrained model as part of other model? 🤗Transformers	0	860	June 9, 2022
Custom PretrainedModel does not loaded correctly Beginners	0	279	November 17, 2022

How to load a pretrained custom model using `from_pretrained`

Related topics