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
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'
)