Comece agoraComece grátis

Configurando os argumentos de treinamento do Llama

Você vai trabalhar com o modelo Llama usado em um chatbot de atendimento ao cliente, fazendo o fine-tuning com dados de suporte ao cliente criados para perguntas e respostas. Para garantir o melhor desempenho desses modelos, sua equipe fará o fine-tuning de um modelo Llama para essa tarefa usando o conjunto de dados bitext.

Você quer fazer um teste rápido do loop de treinamento para verificar se o script funciona. Então, comece definindo uma taxa de aprendizado pequena e limitando o treinamento a poucas etapas nos seus argumentos de treinamento.

Este exercicio faz parte do curso

Ajuste Fino com Llama 3

Ver curso

Instruções do exercicio

  • Importe e instancie a classe auxiliar para armazenar seus argumentos de treinamento.
  • Defina o argumento de treinamento para a taxa de aprendizado com o valor 2e-3.

exercicio interativo prático

Tente este exercicio completando este código de exemplo.

# Load helper class for the training arguments from the correct library
from ____ import ____ 
training_arguments = ____(
  	# Set learning rate
    ____=____, 
    warmup_ratio=0.03,
  	num_train_epochs=3,
  	output_dir='/tmp',
    per_device_train_batch_size=1,
    gradient_accumulation_steps=1,
    save_steps=10,
    logging_steps=2,
    lr_scheduler_type='constant',
    report_to='none'
)
Editar e Executar Código