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
Ö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"
}