Saving trained number of epochs in config.json // Custom fields in config.json


I have a custom model that is a subclass from transformers.PreTrainedModel. I save the model with model.save_pretrained after every epoch.

Is it possible to save the number of trained epochs in config.json? Or is it in general possible to add custom fields to config.json?

Thanks for your help!

Up. Has somebody an answer to my question :innocent:?