How to use fine-tuned model

Beginners
1

hi,
I use trainer.train() to fine-tune my custome NER dataset with bert model,
and train.save_model() to save the models that having different hyperparameters.

how should I do to reload my different model?

2

You can just do

from transformers import AutoModel

model = AutoModel.from_pretrained(ouput_dir_of_your_trainer)