Configurar o Trainer

Seu modelo substituirá frases complexas por paráfrases mais simples e acessíveis para melhorar a legibilidade das traduções. Por exemplo, você pode simplificar a frase original "As condições climáticas inclementes precipitaram o adiamento do evento ao ar livre" com "O mau tempo fez com que o evento ao ar livre fosse adiado". Crie o site Trainer para preparar o treinamento do seu serviço de tradução de idiomas! O exercício levará algum tempo para ser executado com a chamada para trainer.train().

Alguns dados foram pré-carregados:

model é um modelo de transformador
dataset contém o conjunto de dados MRPC de paráfrases de frases
compute_metrics A função retorna a precisão e a pontuação F1
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 endereço model para a classe Trainer().
Insira os argumentos de treinamento na classe Trainer().
Passe uma função para computar métricas para a classe Trainer().
Imprima o dispositivo que o trainer escolher.

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