Hi, I am trying to convert my model to onnx format with the help of this notebook I got error , since config.json does not exist. My model is a custom model with extra layers, similar to this, Now how can I create a config.json file for this?

You need to subclass it to have the save_pretrained methods available. So instead of class Mean_Pooling_Model(nn.Module): use from transformers.modeling_utils import PreTrainedModel class Mean_Pooling_Model(PreTrainedModel): It will add extra functionality on top of nn.Module. huggingfa…

How to create a config.json after saving a model

🤗Transformers

seanbenhur October 4, 2021, 9:46am 3

Yes, but this is a custom model that I have saved in pytorch style, since it consists of additional layers, is there anyway to generate confg.json file?

Topic		Replies	Views
Convert Pytorch Model to Huggingface Transformer? 🤗Transformers	2	11013	May 2, 2024
How to get config.json from .pt loaded model? Beginners	1	1350	April 11, 2022
Save custom transformer as PreTrainedModel Intermediate	1	934	September 7, 2021
How to save custom model to get config.json file Beginners	2	1489	October 20, 2024
Saving trained number of epochs in config.json // Custom fields in config.json 🤗Transformers	1	278	November 7, 2022

How to create a config.json after saving a model

Related topics