From 7c0e1544157ad902e2f6e8fa63e7e9f1262d4fc2 Mon Sep 17 00:00:00 2001
From: "Julien B." <xm9q8f80@jlnbrtn.me>
Date: Sat, 24 Aug 2024 22:56:29 +0200
Subject: [PATCH] fix(trainer): change trainer function
---
microservices/trainer/trainer.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py
index 0615bec..b9d00d7 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
--
GitLab