Configurar o Trainer

Seu modelo vai substituir expressões complexas por paráfrases mais simples e acessíveis para melhorar a legibilidade de traduções. Por exemplo, ele pode simplificar a frase original "The inclement weather conditions precipitated the postponement of the outdoor event" para "The bad weather caused the outdoor event to be delayed." Construa o Trainer para preparar o treinamento do seu serviço de tradução de linguagem! O exercício levará algum tempo para rodar por causa da chamada trainer.train().

Alguns dados já foram pré-carregados:

model é um modelo Transformer
dataset contém o conjunto de dados MRPC de paráfrases de sentenças
A função compute_metrics retorna acurácia e F1-score
Você definiu training_args em um exercício anterior

Este exercício faz parte do curso

Treinamento Eficiente de Modelos de IA com PyTorch

Ver curso

Instruções do exercício

Passe o model para a classe Trainer().
Insira os argumentos de treinamento na classe Trainer().
Passe uma função para calcular as métricas para a classe Trainer().
Imprima o dispositivo que o trainer escolhe.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

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 e executar o código