Configurar o Trainer

Seu modelo vai substituir frases complexas por paráfrases mais simples e acessíveis para melhorar a legibilidade das 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 com a 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 exercicio faz parte do curso

Treinamento Eficiente de Modelos de IA com PyTorch

Ver curso

Instruções do exercicio

Passe o model para a classe Trainer().
Forneça os argumentos de treino na classe Trainer().
Passe uma função para calcular métricas para a classe Trainer().
Imprima o dispositivo que o trainer escolher.

exercicio interativo prático

Tente este exercicio completando este código de exemplo.

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 e Executar Código