diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py index ad423312cac9e92ac7083418fad68eed9d828b1b..0615bec8b7c329bacbcec997e6e85b1d9d9f79e3 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,