Definindo receitas personalizadas
Você está fazendo fine-tuning de um modelo Llama pré-treinado para um cliente que precisa de configurações específicas. Seu plano é usar o TorchTune para o fine-tuning e, por isso, você precisa preparar um dicionário em Python para armazenar os requisitos da receita personalizada que será usada para executar a tarefa de fine-tuning.
Este exercicio faz parte do curso
Ajuste Fino com Llama 3
Instruções do exercicio
- Especifique os requisitos do cliente no seu dicionário: primeiro, adicione o modelo
torchtune.models.llama3_2.llama3_2_1b. - Adicione um tamanho de lote de 8 e um dispositivo de GPU.
exercicio interativo prático
Tente este exercicio completando este código de exemplo.
config_dict = {
# Define the model
____,
# Define the batch size
____,
# Define the device type
____,
"epochs": 15,
"optimizer": {"_component_": "bitsandbytes.optim.PagedAdamW8bit", "lr": 3e-05},
"dataset": {"_component_": "custom_dataset"},
"output_dir": "/tmp/finetune_results"
}