Configura Trainer

Tu modelo reemplazará frases complejas por paráfrasis más sencillas y accesibles para mejorar la legibilidad de las traducciones. Por ejemplo, puede simplificar la frase original «The inclement weather conditions precipitated the postponement of the outdoor event» a «The bad weather caused the outdoor event to be delayed.» ¡Construye el Trainer para dejar listo el entrenamiento de tu servicio de traducción de lenguaje! Este ejercicio tardará un poco en ejecutarse con la llamada a trainer.train().

Se han precargado algunos datos:

model es un modelo Transformer
dataset contiene el conjunto de datos MRPC de paráfrasis de oraciones
La función compute_metrics devuelve accuracy y F1 score
Has definido training_args en un ejercicio anterior

Este ejercicio forma parte del curso

Entrenamiento eficiente de modelos de IA con PyTorch

Ver curso

Instrucciones del ejercicio

Pasa el model a la clase Trainer().
Introduce los argumentos de entrenamiento en la clase Trainer().
Pasa una función para calcular métricas a la clase Trainer().
Imprime el dispositivo que elige trainer.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

trainer = Trainer(
    # Pass in the model
    ____=____,
    # Input the training arguments
    ____=____,
    train_dataset=dataset["train"],
    eval_dataset=dataset["validation"],
    # Pass in a function to compute metrics
    compute_metrics=____,
)

trainer.train()

# Print the device that the trainer chooses
print(____.args.____)

Editar y ejecutar código