diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py index 0615bec8b7c329bacbcec997e6e85b1d9d9f79e3..b9d00d7dbe6bc0f5ea8d21b1628a5dcd16a0c844 100644 --- a/microservices/trainer/trainer.py +++ b/microservices/trainer/trainer.py @@ -220,7 +220,6 @@ def training_process(training_data, fondation_model_id, finetuned_repo_name, hug logging_steps=100, save_strategy='epoch', save_total_limit=2, - load_best_model_at_end=True, metric_for_best_model='macro_f1', log_level='critical', seed=12345