From 316af8cd53f25e4a5d7de1b29d5331419171205e Mon Sep 17 00:00:00 2001 From: "Julien B." <xm9q8f80@jlnbrtn.me> Date: Sat, 24 Aug 2024 22:48:20 +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 ad42331..0615bec 100644 --- a/microservices/trainer/trainer.py +++ b/microservices/trainer/trainer.py @@ -212,7 +212,6 @@ def training_process(training_data, fondation_model_id, finetuned_repo_name, hug training_args = TrainingArguments( output_dir="./models/fine_tune_bert_output_span_cat", - evaluation_strategy="epoch", learning_rate=2.5e-4, per_device_train_batch_size=16, per_device_eval_batch_size=16, -- GitLab