用huggingface的transformer库训练模型时，怎么在TrainingArguments设置保存结果最好的模型

Question

现在只发现有保存模型个数，和保存模型的方法的设置，还有一个是load_best_model，但这不是保存最好的模型吧。

Answer 1

huggingface的参数并不能实现只保存最佳的模型，但是它保存的模型里面肯定包含了最佳的模型，但是evaluation_strategy，save_strategy，load_best_model_at_end这几个参数设置好的话可以让你在最后加载的是最佳的模型，你可以尝试在训练完后单独保存这个最佳的模型，反正都是Pytorch的模型嘛，可以试试torch.save()或者huggingface里面的model.save_pretrained之类的方法吧。实在不行就只用huggingface提供的模型吧，然后自己去定义训练和评估以及模型的保存策略。