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

fix(trainer): change trainer function

parent 24bb7315
Branches
No related tags found
No related merge requests found
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment