Comece agoraComece grátis

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

Ver curso

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