Hello experts, I am trying to write BERT model with my own custom model (adding layer end of BERT).It goes well and I would like to save model to avoid future training. I am using trainer API and so, there are two ways to save model, right? training_arugments and trainer (save the model under a…

How to save custom model to get config.json file

John6666 October 20, 2024, 1:04pm 3

It is most reliable if save_pretrained() can be used.

Topic		Replies	Views
How to save and load the custom Hugging face model including config.json file using pytorch Models	2	7253	February 16, 2023
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 🤗Transformers	21	40744	October 30, 2024
Saving local bert/roberta model not working using save_pretrained Beginners	0	1072	November 18, 2022
Saving custom and/or finetuned models without the HUB Beginners	3	1056	March 2, 2022