Kom igångKom igång gratis

Definiera anpassade recept

Du finjusterar en förtränad Llama-modell åt en kund som har specifika konfigurationskrav. Du planerar att använda TorchTune för finjusteringen och behöver därför förbereda en Python-ordbok där du kan lagra kraven för det anpassade recept du ska använda för att köra finjusteringsjobbet.

Den här övningen är en del av kursen

Finjustering med Llama 3

Visa kurs

Övningsinstruktioner

  • Ange kundens krav i din ordbok: lägg först till modellen torchtune.models.llama3_2.llama3_2_1b.
  • Lägg till en batchstorlek på 8 och en GPU-enhet.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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"
}
Redigera och kör kod