Skip to content
Snippets Groups Projects
Commit 7c0e1544 authored by Julien B.'s avatar Julien B.
Browse files

fix(trainer): change trainer function

parent 316af8cd
No related branches found
No related tags found
No related merge requests found
...@@ -220,7 +220,6 @@ def training_process(training_data, fondation_model_id, finetuned_repo_name, hug ...@@ -220,7 +220,6 @@ def training_process(training_data, fondation_model_id, finetuned_repo_name, hug
logging_steps=100, logging_steps=100,
save_strategy='epoch', save_strategy='epoch',
save_total_limit=2, save_total_limit=2,
load_best_model_at_end=True,
metric_for_best_model='macro_f1', metric_for_best_model='macro_f1',
log_level='critical', log_level='critical',
seed=12345 seed=12345
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment