Hi
I’m using TFDistilBertForSequenceClassification.from_pretrained and DistilBertTokenizer to create a model and tokenizer. Later I’m using model.save_pretrained and tokenizer.save_pretrained to save the model and tokenizer.
However, I want to try and train 3 models at once (say a, b and c) so I want to save the model and tokenizer with custom names (e.g. a_model, b_model, c_model, etc).
I can’t find any reference to being able to do this. All I can specify is the save directory.
I guess I could create 3 directories (a, b, c) and save the models into theit respective ones, but that feels a bit messy.
Is there any way to change the default output file name?
Many thanks in advance