Ajuste del CV: configuración del entrenador
Ahora que ya has preparado el conjunto de datos y adaptado un modelo preentrenado a las nuevas clases, es el momento de configurar tu entrenador.
Se han cargado las bibliotecas TrainingArguments y Trainer desde la biblioteca transformers. El modelo (model) y el conjunto de datos (dataset) se han cargado tal y como los configuraste anteriormente.
Este ejercicio forma parte del curso
Modelos multimodales con Hugging Face
Instrucciones del ejercicio
- Ajusta la tasa de aprendizaje a un
6e-5. - Proporciona el modelo, los datos de entrenamiento y los datos de prueba a la instancia de
Trainer.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
training_args = TrainingArguments(
output_dir="dataset_finetune",
# Adjust the learning rate
____,
gradient_accumulation_steps=4,
num_train_epochs=3,
push_to_hub=False
)
trainer = Trainer(
# Provide the model and datasets
model=____,
args=training_args,
data_collator=data_collator,
train_dataset=____,
eval_dataset=____,
processing_class=image_processor,
compute_metrics=compute_metrics,
)