Configura el 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" por "The bad weather caused the outdoor event to be delayed." ¡Construye el Trainer para preparar el entrenamiento de tu servicio de traducción automática! El 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 la exactitud y la puntuación F1
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 el trainer.

ejercicio interactivo práctico

Prueba este ejercicio completando este código de ejemplo.

trainer = Trainer(
    # Pass in the model
    model=____,
    # Input the training arguments
    args=____,
    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(trainer.args.____)

Editar y ejecutar código