ComenzarEmpieza gratis

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

Ver curso

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,
)
Editar y ejecutar código